mirror of
https://github.com/csd4ni3l/music-player.git
synced 2025-11-05 05:58:17 +01:00
138 lines
6.4 KiB
Python
138 lines
6.4 KiB
Python
from yt_dlp import YoutubeDL
|
|
from mutagen.easyid3 import EasyID3
|
|
from pydub import AudioSegment
|
|
|
|
import arcade, arcade.gui, os, json, yt_dlp, threading
|
|
|
|
from arcade.gui.experimental.focus import UIFocusGroup
|
|
|
|
from utils.utils import UIFocusTextureButton, BufferLogger
|
|
from utils.constants import button_style, dropdown_style, yt_dlp_parameters
|
|
from utils.preload import button_texture, button_hovered_texture
|
|
|
|
class Downloader(arcade.gui.UIView):
|
|
def __init__(self, pypresence_client, current_mode, current_music_name, current_length, current_music_player, queue, loaded_sounds, shuffle):
|
|
super().__init__()
|
|
|
|
self.current_mode = current_mode
|
|
self.current_music_name = current_music_name
|
|
self.current_length = current_length
|
|
self.current_music_player = current_music_player
|
|
self.queue = queue
|
|
self.loaded_sounds = loaded_sounds
|
|
self.shuffle = shuffle
|
|
|
|
self.pypresence_client = pypresence_client
|
|
self.pypresence_client.update(state="Downloading music", start=self.pypresence_client.start_time)
|
|
|
|
with open("settings.json", "r") as file:
|
|
self.settings_dict = json.load(file)
|
|
|
|
self.tab_options = self.settings_dict.get("tab_options", ["~/Music", "~/Downloads"])
|
|
|
|
self.yt_dl = YoutubeDL(params=yt_dlp_parameters)
|
|
self.yt_dl.params["logger"] = self.yt_dl_logger = BufferLogger()
|
|
|
|
def on_show_view(self):
|
|
super().on_show_view()
|
|
|
|
self.anchor = self.add_widget(UIFocusGroup(size_hint=(1, 1)))
|
|
|
|
self.download_box = self.anchor.add(arcade.gui.UIBoxLayout(space_between=10), anchor_x="center", anchor_y="center")
|
|
|
|
self.url_name_label = self.download_box.add(arcade.gui.UILabel(text="URL or Name:", font_name="Protest Strike", font_size=36))
|
|
self.url_name_input = self.download_box.add(arcade.gui.UIInputText(font_name="Protest Strike", width=self.window.width / 2, height=self.window.height / 15, font_size=36))
|
|
self.url_name_input.activate()
|
|
|
|
self.tab_label = self.download_box.add(arcade.gui.UILabel(text="Path:", font_name="Protest Strike", font_size=36))
|
|
self.tab_selector = self.download_box.add(arcade.gui.UIDropdown(default=self.tab_options[0], options=self.tab_options, width=self.window.width / 2, height=self.window.height / 15, primary_style=dropdown_style, dropdown_style=dropdown_style, active_style=dropdown_style))
|
|
|
|
self.status_label = self.download_box.add(arcade.gui.UILabel(text="No errors.", font_size=16, text_color=arcade.color.LIGHT_GREEN))
|
|
|
|
self.download_button = self.anchor.add(arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='Download', style=button_style, width=self.window.width / 2, height=self.window.height / 10), anchor_x="center", anchor_y="bottom", align_y=10)
|
|
self.download_button.on_click = lambda event: threading.Thread(target=self.download, daemon=True).start()
|
|
|
|
self.back_button = self.anchor.add(arcade.gui.UITextureButton(texture=button_texture, texture_hovered=button_hovered_texture, text='<--', style=button_style, width=100, height=50), anchor_x="left", anchor_y="top", align_x=5, align_y=-5)
|
|
self.back_button.on_click = lambda event: self.main_exit()
|
|
|
|
self.anchor.detect_focusable_widgets()
|
|
|
|
def on_update(self, delta_time: float) -> bool | None:
|
|
self.status_label.text = self.yt_dl_logger.buffer
|
|
|
|
if "WARNING" in self.yt_dl_logger.buffer:
|
|
self.status_label.update_font(font_color=arcade.color.YELLOW)
|
|
elif "ERROR" in self.yt_dl_logger.buffer:
|
|
self.status_label.update_font(font_color=arcade.color.RED)
|
|
else:
|
|
self.status_label.update_font(font_color=arcade.color.LIGHT_GREEN)
|
|
|
|
def download(self):
|
|
if not self.tab_selector.value:
|
|
return
|
|
|
|
url = self.url_name_input.text
|
|
|
|
if not "http" in url:
|
|
url = f"ytsearch1:{url}"
|
|
|
|
path = os.path.expanduser(self.tab_selector.value)
|
|
|
|
try:
|
|
info = self.yt_dl.extract_info(url, download=True)
|
|
except yt_dlp.DownloadError as e:
|
|
message = "".join(e.msg.strip().split("] ")[1:]) if e.msg else "Unknown yt-dlp error."
|
|
self.yt_dl_logger.buffer = f"ERROR: {message}"
|
|
return
|
|
|
|
if info:
|
|
entry = info['entries'][0] if 'entries' in info else info
|
|
title = entry.get('title', 'Unknown')
|
|
uploader = entry.get('uploader', 'Unknown')
|
|
|
|
if " - " in title:
|
|
artist, track_title = title.split(" - ", 1)
|
|
else:
|
|
artist = uploader
|
|
track_title = title
|
|
title = f"{artist} - {track_title}"
|
|
|
|
try:
|
|
audio = EasyID3("downloaded_music.mp3")
|
|
audio["artist"] = artist
|
|
audio["title"] = track_title
|
|
audio.save()
|
|
except Exception as meta_err:
|
|
self.yt_dl_logger.buffer = f"ERROR: Tried to override metadata based on title, but failed: {meta_err}"
|
|
return
|
|
|
|
if self.settings_dict.get("normalize_audio", True):
|
|
try:
|
|
audio = AudioSegment.from_file("downloaded_music.mp3")
|
|
|
|
if audio.dBFS < self.settings_dict.get("normalized_volume", -8):
|
|
change = self.settings_dict.get("normalized_volume", -8) - audio.dBFS
|
|
audio = audio.apply_gain(change)
|
|
|
|
audio.export("downloaded_music.mp3", format="mp3")
|
|
|
|
except Exception as e:
|
|
self.yt_dl_logger.buffer = f"ERROR: Could not normalize volume due to an error: {e}"
|
|
return
|
|
try:
|
|
output_filename = os.path.join(path, f"{title}.mp3")
|
|
os.replace("downloaded_music.mp3", output_filename)
|
|
|
|
except Exception as e:
|
|
self.yt_dl_logger.buffer = f"ERROR: Could not move file due to an error: {e}"
|
|
return
|
|
else:
|
|
self.yt_dl_logger.buffer = f"ERROR: Info unavailable. This maybe due to being unable to download it due to DRM or other issues"
|
|
return
|
|
|
|
self.yt_dl_logger.buffer = f"Successfully downloaded {title} to {path}"
|
|
|
|
def main_exit(self):
|
|
from menus.main import Main
|
|
self.window.show_view(Main(self.pypresence_client, self.current_mode, self.current_music_name, self.current_length, self.current_music_player, self.queue, self.loaded_sounds, self.shuffle))
|