import datetime import os.path import shutil import webbrowser from enum import Enum, StrEnum from pathlib import Path from PySide6 import QtCore, QtGui, QtWidgets from tidalapi import Quality as QualityAudio from tidal_dl_ng import __version__ from tidal_dl_ng.config import Settings from tidal_dl_ng.constants import CoverDimensions, QualityVideo from tidal_dl_ng.model.cfg import HelpSettings from tidal_dl_ng.model.cfg import Settings as ModelSettings from tidal_dl_ng.model.meta import ReleaseLatest from tidal_dl_ng.ui.dialog_login import Ui_DialogLogin from tidal_dl_ng.ui.dialog_settings import Ui_DialogSettings from tidal_dl_ng.ui.dialog_version import Ui_DialogVersion class DialogVersion(QtWidgets.QDialog): """Version dialog.""" ui: Ui_DialogVersion def __init__( self, parent=None, update_check: bool = False, update_available: bool = False, update_info: ReleaseLatest = None ): super().__init__(parent) # Create an instance of the GUI self.ui = Ui_DialogVersion() # Run the .setupUi() method to show the GUI self.ui.setupUi(self) # Set the version. self.ui.l_version.setText("v" + __version__) if not update_check: self.update_info_hide() self.error_hide() else: self.update_info(update_available, update_info) # Show self.exec() def update_info(self, update_available: bool, update_info: ReleaseLatest): if not update_available and update_info.version == "v0.0.0": self.update_info_hide() self.ui.l_error_details.setText( "Cannot retrieve update information. Maybe something is wrong with your internet connection." ) else: self.error_hide() if not update_available: self.ui.l_h_version_new.setText("Latest available version:") self.changelog_hide() else: self.ui.l_changelog_details.setText(update_info.release_info) self.ui.pb_download.clicked.connect(lambda: webbrowser.open(update_info.url)) self.ui.l_version_new.setText(update_info.version) def error_hide(self): self.ui.l_error.setHidden(True) self.ui.l_error_details.setHidden(True) def update_info_hide(self): self.ui.l_h_version_new.setHidden(True) self.ui.l_version_new.setHidden(True) self.changelog_hide() def changelog_hide(self): self.ui.l_changelog.setHidden(True) self.ui.l_changelog_details.setHidden(True) self.ui.pb_download.setHidden(True) class DialogLogin(QtWidgets.QDialog): """Version dialog.""" ui: Ui_DialogLogin url_redirect: str def __init__(self, url_login: str, hint: str, expires_in: int, parent=None): super().__init__(parent) datetime_current: datetime.datetime = datetime.datetime.now() datetime_expires: datetime.datetime = datetime_current + datetime.timedelta(0, expires_in) # Create an instance of the GUI self.ui = Ui_DialogLogin() # Run the .setupUi() method to show the GUI self.ui.setupUi(self) # Set data. self.ui.tb_url_login.setText(f'https://{url_login}') self.ui.l_hint.setText(hint) self.ui.l_expires_date_time.setText(datetime_expires.strftime("%Y-%m-%d %H:%M")) # Show self.return_code = self.exec() class DialogPreferences(QtWidgets.QDialog): """Preferences dialog.""" ui: Ui_DialogSettings settings: Settings data: ModelSettings s_settings_save: QtCore.Signal icon: QtGui.QIcon help_settings: HelpSettings parameters_checkboxes: [str] parameters_combo: [(str, StrEnum)] parameters_line_edit: [str] parameters_spin_box: [str] prefix_checkbox: str = "cb_" prefix_label: str = "l_" prefix_icon: str = "icon_" prefix_line_edit: str = "le_" prefix_combo: str = "c_" prefix_spin_box: str = "sb_" def __init__(self, settings: Settings, settings_save: QtCore.Signal, parent=None): super().__init__(parent) self.settings = settings self.data = settings.data self.s_settings_save = settings_save self.help_settings = HelpSettings() pixmapi: QtWidgets.QStyle.StandardPixmap = QtWidgets.QStyle.SP_MessageBoxQuestion self.icon = self.style().standardIcon(pixmapi) self._init_checkboxes() self._init_comboboxes() self._init_line_edit() self._init_spin_box() # Create an instance of the GUI self.ui = Ui_DialogSettings() # Run the .setupUi() method to show the GUI self.ui.setupUi(self) # Set data. self.gui_populate() # Post setup self.exec() def _init_line_edit(self): self.parameters_line_edit = [ "download_base_path", "format_album", "format_playlist", "format_mix", "format_track", "format_video", "path_binary_ffmpeg", ] def _init_spin_box(self): self.parameters_spin_box = ["album_track_num_pad_min", "downloads_concurrent_max"] def _init_comboboxes(self): self.parameters_combo = [ ("quality_audio", QualityAudio), ("quality_video", QualityVideo), ("metadata_cover_dimension", CoverDimensions), ] def _init_checkboxes(self): self.parameters_checkboxes = [ "lyrics_embed", "lyrics_file", "video_download", "download_delay", "video_convert_mp4", "extract_flac", "metadata_cover_embed", "cover_album_file", "skip_existing", "symlink_to_track", "playlist_create", ] def gui_populate(self): self.populate_checkboxes() self.populate_combo() self.populate_line_edit() self.populate_spin_box() def dialog_chose_file( self, obj_line_edit: QtWidgets.QLineEdit, file_mode: QtWidgets.QFileDialog | QtWidgets.QFileDialog.FileMode = QtWidgets.QFileDialog.Directory, path_default: str = None, ): # If a path is set, use it otherwise the users home directory. path_settings: str = os.path.expanduser(obj_line_edit.text()) if obj_line_edit.text() else "" # Check if obj_line_edit is empty but path_default can be usd instead path_settings = ( path_settings if path_settings else os.path.expanduser(path_default) if path_default else path_settings ) dir_current: str = path_settings if path_settings and os.path.exists(path_settings) else str(Path.home()) dialog: QtWidgets.QFileDialog = QtWidgets.QFileDialog() # Set to directory mode only but show files. dialog.setFileMode(file_mode) dialog.setViewMode(QtWidgets.QFileDialog.Detail) dialog.setOption(QtWidgets.QFileDialog.ShowDirsOnly, False) dialog.setOption(QtWidgets.QFileDialog.DontResolveSymlinks, True) # There is a bug in the PyQt implementation, which hides files in Directory mode. # Thus, we need to use the PyQt dialog instead of the native dialog. if os.name == "nt" and file_mode == QtWidgets.QFileDialog.Directory: dialog.setOption(QtWidgets.QFileDialog.DontUseNativeDialog, True) dialog.setDirectory(dir_current) # Execute dialog and set path is something is choosen. if dialog.exec(): dir_name: str = dialog.selectedFiles()[0] path: Path = Path(dir_name) obj_line_edit.setText(str(path)) def populate_line_edit(self): for pn in self.parameters_line_edit: label_icon: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + self.prefix_icon + pn) label: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + pn) line_edit: QtWidgets.QLineEdit = getattr(self.ui, self.prefix_line_edit + pn) label_icon.setPixmap(QtGui.QPixmap(self.icon.pixmap(QtCore.QSize(16, 16)))) label_icon.setToolTip(getattr(self.help_settings, pn)) label.setText(pn) line_edit.setText(str(getattr(self.data, pn))) # Base Path File Dialog self.ui.pb_download_base_path.clicked.connect(lambda x: self.dialog_chose_file(self.ui.le_download_base_path)) self.ui.pb_path_binary_ffmpeg.clicked.connect( lambda x: self.dialog_chose_file( self.ui.le_path_binary_ffmpeg, file_mode=QtWidgets.QFileDialog.FileMode.ExistingFiles, path_default=shutil.which("ffmpeg"), ) ) def populate_combo(self): for p in self.parameters_combo: pn: str = p[0] values: Enum = p[1] label_icon: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + self.prefix_icon + pn) label: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + pn) combo: QtWidgets.QComboBox = getattr(self.ui, self.prefix_combo + pn) setting_current = getattr(self.data, pn) label_icon.setPixmap(QtGui.QPixmap(self.icon.pixmap(QtCore.QSize(16, 16)))) label_icon.setToolTip(getattr(self.help_settings, pn)) label.setText(pn) for index, v in enumerate(values): combo.addItem(v.name, v) if v == setting_current: combo.setCurrentIndex(index) def populate_checkboxes(self): for pn in self.parameters_checkboxes: checkbox: QtWidgets.QCheckBox = getattr(self.ui, self.prefix_checkbox + pn) checkbox.setText(pn) checkbox.setToolTip(getattr(self.help_settings, pn)) checkbox.setIcon(self.icon) checkbox.setChecked(getattr(self.data, pn)) def populate_spin_box(self): for pn in self.parameters_spin_box: label_icon: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + self.prefix_icon + pn) label: QtWidgets.QLabel = getattr(self.ui, self.prefix_label + pn) spin_box: QtWidgets.QSpinBox = getattr(self.ui, self.prefix_spin_box + pn) label_icon.setPixmap(QtGui.QPixmap(self.icon.pixmap(QtCore.QSize(16, 16)))) label_icon.setToolTip(getattr(self.help_settings, pn)) label.setText(pn) spin_box.setValue(getattr(self.data, pn)) def accept(self): # Get settings. self.to_settings() self.done(1) def to_settings(self): for item in self.parameters_checkboxes: setattr(self.settings.data, item, getattr(self.ui, self.prefix_checkbox + item).isChecked()) for item in self.parameters_line_edit: setattr(self.settings.data, item, getattr(self.ui, self.prefix_line_edit + item).text()) for item in self.parameters_combo: setattr(self.settings.data, item[0], getattr(self.ui, self.prefix_combo + item[0]).currentData()) for item in self.parameters_spin_box: setattr(self.settings.data, item, getattr(self.ui, self.prefix_spin_box + item).value()) self.s_settings_save.emit()