feat(gui): ✨ enhance user media handling with folder support
* Updated `user_media_lists` to return a dictionary containing both `playlists` and `mixes`, including `Folder` objects. * Modified `on_populate_tree_lists` to handle `Folder` items and display their contents. * Introduced new methods for loading and displaying folder contents in the user interface.
This commit is contained in:
@@ -5,6 +5,7 @@ from PySide6 import QtCore, QtGui, QtWidgets
|
||||
from tidalapi import Album, Mix, Playlist, Track, UserPlaylist, Video
|
||||
from tidalapi.artist import Artist
|
||||
from tidalapi.media import Quality
|
||||
from tidalapi.playlist import Folder
|
||||
|
||||
from tidal_dl_ng.constants import QualityVideo
|
||||
|
||||
@@ -33,8 +34,8 @@ def get_results_media_item(
|
||||
return result
|
||||
|
||||
|
||||
def get_user_list_media_item(item: QtWidgets.QTreeWidgetItem) -> Mix | Playlist | UserPlaylist:
|
||||
result: Mix | Playlist | UserPlaylist = get_table_data(item, 1)
|
||||
def get_user_list_media_item(item: QtWidgets.QTreeWidgetItem) -> Mix | Playlist | UserPlaylist | Folder | str:
|
||||
result: Mix | Playlist | UserPlaylist | Folder | str = get_table_data(item, 1)
|
||||
|
||||
return result
|
||||
|
||||
@@ -73,7 +74,9 @@ def get_queue_download_quality_video(
|
||||
|
||||
|
||||
def set_table_data(
|
||||
item: QtWidgets.QTreeWidgetItem, data: Track | Video | Album | Artist | Mix | Playlist | UserPlaylist, column: int
|
||||
item: QtWidgets.QTreeWidgetItem,
|
||||
data: Track | Video | Album | Artist | Mix | Playlist | UserPlaylist | Folder | str,
|
||||
column: int,
|
||||
):
|
||||
item.setData(column, QtCore.Qt.ItemDataRole.UserRole, data)
|
||||
|
||||
@@ -83,7 +86,8 @@ def set_results_media(item: QtWidgets.QTreeWidgetItem, media: Track | Video | Al
|
||||
|
||||
|
||||
def set_user_list_media(
|
||||
item: QtWidgets.QTreeWidgetItem, media: Track | Video | Album | Artist | Mix | Playlist | UserPlaylist
|
||||
item: QtWidgets.QTreeWidgetItem,
|
||||
media: Track | Video | Album | Artist | Mix | Playlist | UserPlaylist | Folder | str,
|
||||
):
|
||||
set_table_data(item, media, 1)
|
||||
|
||||
|
||||
@@ -193,12 +193,42 @@ def paginate_results(func_get_items_media: [Callable]) -> [Track | Video | Album
|
||||
return result
|
||||
|
||||
|
||||
def user_media_lists(session: Session) -> [Playlist | UserPlaylist | Mix]:
|
||||
user_playlists: [Playlist | UserPlaylist] = paginate_results([session.user.playlist_and_favorite_playlists])
|
||||
user_mixes: [Mix] = session.mixes().categories[0].items
|
||||
result: [Playlist | UserPlaylist | Mix] = user_playlists + user_mixes
|
||||
def user_media_lists(session: Session) -> dict[str, list]:
|
||||
"""Fetch user media lists using tidalapi's built-in pagination where available.
|
||||
|
||||
return result
|
||||
Returns a dictionary with 'playlists' and 'mixes' keys containing lists of media items.
|
||||
For playlists, includes both Folder and Playlist objects at the root level.
|
||||
|
||||
Args:
|
||||
session (Session): TIDAL session object.
|
||||
|
||||
Returns:
|
||||
dict[str, list]: Dictionary with 'playlists' (includes Folder and Playlist) and 'mixes' lists.
|
||||
"""
|
||||
# Use built-in pagination for playlists (root level only)
|
||||
playlists = session.user.favorites.playlists_paginated()
|
||||
|
||||
# Fetch root-level folders manually (no paginated version available)
|
||||
folders = []
|
||||
offset = 0
|
||||
limit = 50
|
||||
|
||||
while True:
|
||||
batch = session.user.favorites.playlist_folders(limit=limit, offset=offset, parent_folder_id="root")
|
||||
if not batch:
|
||||
break
|
||||
folders.extend(batch)
|
||||
if len(batch) < limit:
|
||||
break
|
||||
offset += limit
|
||||
|
||||
# Combine folders and playlists
|
||||
all_playlists = folders + playlists
|
||||
|
||||
# Get mixes
|
||||
user_mixes = session.mixes().categories[0].items
|
||||
|
||||
return {"playlists": all_playlists, "mixes": user_mixes}
|
||||
|
||||
|
||||
def instantiate_media(
|
||||
|
||||
Reference in New Issue
Block a user