308 lines
11 KiB
Python
308 lines
11 KiB
Python
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'<a href="https://{url_login}">https://{url_login}</a>')
|
|
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()
|