diff options
Diffstat (limited to 'tidal_dl_ng/helper')
| -rw-r--r-- | tidal_dl_ng/helper/tidal.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tidal_dl_ng/helper/tidal.py b/tidal_dl_ng/helper/tidal.py index 60f636d..9c4eeba 100644 --- a/tidal_dl_ng/helper/tidal.py +++ b/tidal_dl_ng/helper/tidal.py @@ -1,4 +1,4 @@ -from tidalapi import Album, Mix, Playlist, Session, Track, Video +from tidalapi import Album, Mix, Playlist, Session, Track, UserPlaylist, Video from tidalapi.session import SearchTypes from tidal_dl_ng.constants import MediaType @@ -96,3 +96,11 @@ def items_results_all(media_list: [Mix | Playlist | Album], videos_include: bool done = True return result + + +def user_media_lists(session: Session) -> [Playlist | UserPlaylist | Mix]: + user_playlists: [Playlist | UserPlaylist] = session.user.playlist_and_favorite_playlists() + user_mixes: [Mix] = session.mixes().categories[0].items + result: [Playlist | UserPlaylist | Mix] = user_playlists + user_mixes + + return result |