86 lines
3.2 KiB
Python
86 lines
3.2 KiB
Python
|
"""History management mixin for MainWindow.
|
||
|
|
|
||
|
|
Handles download history and duplicate prevention.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from PySide6 import QtCore
|
||
|
|
from tidalapi import Track
|
||
|
|
|
||
|
|
from tidal_dl_ng.dialog import DialogPreferences
|
||
|
|
from tidal_dl_ng.dialog_history import DialogHistory
|
||
|
|
from tidal_dl_ng.logger import logger_gui
|
||
|
|
from tidal_dl_ng.model.gui_data import StatusbarMessage
|
||
|
|
|
||
|
|
|
||
|
|
class HistoryMixin:
|
||
|
|
"""Mixin containing download history management methods."""
|
||
|
|
|
||
|
|
def on_view_history(self) -> None:
|
||
|
|
"""Open the download history dialog."""
|
||
|
|
DialogHistory(history_service=self.history_service, parent=self)
|
||
|
|
|
||
|
|
def on_toggle_duplicate_prevention(self, enabled: bool) -> None:
|
||
|
|
"""Toggle duplicate download prevention on or off."""
|
||
|
|
self.history_service.update_settings(preventDuplicates=enabled)
|
||
|
|
status_msg = "enabled" if enabled else "disabled"
|
||
|
|
logger_gui.info(f"Duplicate download prevention {status_msg}")
|
||
|
|
self.s_statusbar_message.emit(StatusbarMessage(message=f"Duplicate prevention {status_msg}.", timeout=2500))
|
||
|
|
|
||
|
|
def on_mark_track_as_downloaded(self, track: Track, index: QtCore.QModelIndex) -> None:
|
||
|
|
"""Mark a track as downloaded in history."""
|
||
|
|
track_id = str(track.id)
|
||
|
|
|
||
|
|
source_type = "manual"
|
||
|
|
source_id = None
|
||
|
|
source_name = None
|
||
|
|
|
||
|
|
if hasattr(track, "album") and track.album:
|
||
|
|
source_type = "album"
|
||
|
|
source_id = str(track.album.id)
|
||
|
|
source_name = track.album.name
|
||
|
|
|
||
|
|
self.history_service.add_track_to_history(
|
||
|
|
track_id=track_id, source_type=source_type, source_id=source_id, source_name=source_name
|
||
|
|
)
|
||
|
|
|
||
|
|
self._update_downloaded_column(index, True)
|
||
|
|
logger_gui.info(f"Marked track as downloaded: {track.name}")
|
||
|
|
|
||
|
|
def on_mark_track_as_not_downloaded(self, track_id: str, index: QtCore.QModelIndex) -> None:
|
||
|
|
"""Remove a track from download history."""
|
||
|
|
success = self.history_service.remove_track_from_history(track_id)
|
||
|
|
|
||
|
|
if success:
|
||
|
|
self._update_downloaded_column(index, False)
|
||
|
|
logger_gui.info(f"Unmarked track (ID: {track_id})")
|
||
|
|
|
||
|
|
def _update_downloaded_column(self, index: QtCore.QModelIndex, is_downloaded: bool) -> None:
|
||
|
|
"""Update the Downloaded? column for a specific index."""
|
||
|
|
source_index = self.proxy_tr_results.mapToSource(index)
|
||
|
|
|
||
|
|
item = self.model_tr_results.itemFromIndex(source_index)
|
||
|
|
if not item:
|
||
|
|
return
|
||
|
|
|
||
|
|
row = item.row()
|
||
|
|
parent = item.parent()
|
||
|
|
|
||
|
|
downloaded_item = self.model_tr_results.item(row, 8) if parent is None else parent.child(row, 8)
|
||
|
|
|
||
|
|
if downloaded_item:
|
||
|
|
if is_downloaded:
|
||
|
|
downloaded_item.setText("✅")
|
||
|
|
downloaded_item.setTextAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
|
||
|
|
else:
|
||
|
|
downloaded_item.setText("")
|
||
|
|
|
||
|
|
def on_preferences(self) -> None:
|
||
|
|
"""Open the preferences dialog."""
|
||
|
|
DialogPreferences(settings=self.settings, settings_save=self.s_settings_save, parent=self)
|
||
|
|
|
||
|
|
def on_settings_save(self) -> None:
|
||
|
|
"""Save settings and re-apply them to the GUI."""
|
||
|
|
self.settings.save()
|
||
|
|
self.apply_settings(self.settings)
|
||
|
|
self._init_dl()
|