✨ User favorites (hearted items) are now selectable from the left pane's list. Fixes #99

This commit is contained in:
Robert Honz
2024-05-24 07:23:10 +02:00
parent 647e950100
commit 5b04063e83
3 changed files with 42 additions and 11 deletions
+9
View File
@@ -50,3 +50,12 @@ class QueueDownloadStatus(StrEnum):
Finished: str = "✅" Finished: str = "✅"
Failed: str = "❌" Failed: str = "❌"
Skipped: str = "↪️" Skipped: str = "↪️"
FAVORITES: {} = {
"fav_videos": {"name": "Videos", "function_name": "videos"},
"fav_tracks": {"name": "Tracks", "function_name": "tracks"},
"fav_mixes": {"name": "Mixes & Radio", "function_name": "mixes"},
"fav_artists": {"name": "Artists", "function_name": "artists"},
"fav_albums": {"name": "Albums", "function_name": "albums"},
}
+29 -7
View File
@@ -47,7 +47,7 @@ from tidalapi.artist import Artist
from tidalapi.session import SearchTypes from tidalapi.session import SearchTypes
from tidal_dl_ng.config import Settings, Tidal from tidal_dl_ng.config import Settings, Tidal
from tidal_dl_ng.constants import QualityVideo, QueueDownloadStatus, TidalLists from tidal_dl_ng.constants import FAVORITES, QualityVideo, QueueDownloadStatus, TidalLists
from tidal_dl_ng.download import Download from tidal_dl_ng.download import Download
from tidal_dl_ng.logger import XStream, logger_gui from tidal_dl_ng.logger import XStream, logger_gui
from tidal_dl_ng.model.gui_data import ProgressBars, QueueDownloadItem, ResultItem, StatusbarMessage from tidal_dl_ng.model.gui_data import ProgressBars, QueueDownloadItem, ResultItem, StatusbarMessage
@@ -279,20 +279,21 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
twi_playlists: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems( twi_playlists: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems(
TidalLists.Playlists, QtCore.Qt.MatchExactly, 0 TidalLists.Playlists, QtCore.Qt.MatchExactly, 0
)[0] )[0]
twi_favorites: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems(
TidalLists.Favorites, QtCore.Qt.MatchExactly, 0
)[0]
twi_mixes: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems( twi_mixes: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems(
TidalLists.Mixes, QtCore.Qt.MatchExactly, 0 TidalLists.Mixes, QtCore.Qt.MatchExactly, 0
)[0] )[0]
twi_favorites: QtWidgets.QTreeWidgetItem = self.tr_lists_user.findItems(
TidalLists.Favorites, QtCore.Qt.MatchExactly, 0
)[0]
# Remove all children if present # Remove all children if present
for twi in [twi_playlists, twi_mixes, twi_favorites]: for twi in [twi_playlists, twi_mixes]:
for i in reversed(range(twi.childCount())): for i in reversed(range(twi.childCount())):
twi.removeChild(twi.child(i)) twi.removeChild(twi.child(i))
# Populate dynamic user lists
for item in user_lists: for item in user_lists:
if isinstance(item, (UserPlaylist, Playlist)): if isinstance(item, UserPlaylist | Playlist):
twi_child = QtWidgets.QTreeWidgetItem(twi_playlists) twi_child = QtWidgets.QTreeWidgetItem(twi_playlists)
name: str = item.name name: str = item.name
description: str = f" {item.description}" if item.description else "" description: str = f" {item.description}" if item.description else ""
@@ -306,6 +307,16 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
set_user_list_media(twi_child, item) set_user_list_media(twi_child, item)
twi_child.setText(2, info) twi_child.setText(2, info)
# Populate static favorites
for key, favorite in FAVORITES.items():
twi_child = QtWidgets.QTreeWidgetItem(twi_favorites)
name: str = favorite["name"]
info: str = ""
twi_child.setText(0, name)
set_user_list_media(twi_child, key)
twi_child.setText(2, info)
# Stop load spinner # Stop load spinner
self.s_spinner_stop.emit() self.s_spinner_stop.emit()
self.s_pb_reload_status.emit(True) self.s_pb_reload_status.emit(True)
@@ -712,10 +723,16 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
self.tidal.settings_apply() self.tidal.settings_apply()
def on_list_items_show(self, item: QtWidgets.QTreeWidgetItem): def on_list_items_show(self, item: QtWidgets.QTreeWidgetItem):
media_list: Album | Playlist = get_user_list_media_item(item) media_list: Album | Playlist | str = get_user_list_media_item(item)
# Only if clicked item is not a top level item. # Only if clicked item is not a top level item.
if media_list: if media_list:
if isinstance(media_list, str) and media_list.startswith("fav_"):
function_name: str = FAVORITES[media_list]["function_name"]
function_list = getattr(self.tidal.session.user.favorites, function_name)
self.list_items_show_result(favorite_function=function_list)
else:
self.list_items_show_result(media_list) self.list_items_show_result(media_list)
self.cover_show(media_list) self.cover_show(media_list)
@@ -754,13 +771,18 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
media_list: Album | Playlist | Mix | Artist | None = None, media_list: Album | Playlist | Mix | Artist | None = None,
point: QtCore.QPoint | None = None, point: QtCore.QPoint | None = None,
parent: QtGui.QStandardItem = None, parent: QtGui.QStandardItem = None,
favorite_function: Callable = None,
) -> None: ) -> None:
if point: if point:
item = self.tr_lists_user.itemAt(point) item = self.tr_lists_user.itemAt(point)
media_list = get_user_list_media_item(item) media_list = get_user_list_media_item(item)
# Get all results # Get all results
if favorite_function:
media_items: [Track | Video | Album] = favorite_function()
else:
media_items: [Track | Video | Album] = items_results_all(media_list) media_items: [Track | Video | Album] = items_results_all(media_list)
result: [ResultItem] = self.search_result_to_model(media_items) result: [ResultItem] = self.search_result_to_model(media_items)
self.populate_tree_results(result, parent=parent) self.populate_tree_results(result, parent=parent)
+1 -1
View File
@@ -127,7 +127,7 @@ def all_artist_album_ids(media_artist: Artist) -> [int | None]:
return result return result
def paginate_media(func_get_items_media) -> [Track | Video | Album]: def paginate_media(func_get_items_media: [Callable]) -> [Track | Video | Album]:
result: [Track | Video | Album] = [] result: [Track | Video | Album] = []
for func_media in func_get_items_media: for func_media in func_get_items_media: