summaryrefslogtreecommitdiff
path: root/tidal_dl_ng/helper/tidal.py
diff options
context:
space:
mode:
authorRobert Honz <exislow+github@gmail.com>2024-03-13 06:48:11 +0100
committerRobert Honz <exislow+github@gmail.com>2024-03-13 06:48:11 +0100
commite3a8588324328ee27edd4ff7dcab65d9dda7d31a (patch)
tree2738fb2bb4e5b8c7c3c009d5527162bfa06202f6 /tidal_dl_ng/helper/tidal.py
parentf4365e879b869e508d1e8cc5e94157c30950d799 (diff)
parentbbb6051fdd34fabf7565df480df66c128ffa4ef5 (diff)
downloadtidal-dl-ng-e3a8588324328ee27edd4ff7dcab65d9dda7d31a.tar.gz
Merge branch '41-request-song-organization'
* 41-request-song-organization: ✨ Track media items will be re-created with album information. Path format refactored. ♻️ Refactored format_str method to one `format_str_media` method. # Conflicts: # tidal_dl_ng/download.py # tidal_dl_ng/gui.py
Diffstat (limited to 'tidal_dl_ng/helper/tidal.py')
-rw-r--r--tidal_dl_ng/helper/tidal.py27
1 files changed, 12 insertions, 15 deletions
diff --git a/tidal_dl_ng/helper/tidal.py b/tidal_dl_ng/helper/tidal.py
index e13c909..4ae57bd 100644
--- a/tidal_dl_ng/helper/tidal.py
+++ b/tidal_dl_ng/helper/tidal.py
@@ -111,21 +111,18 @@ def instantiate_media(
session: Session,
media_type: type[MediaType.TRACK, MediaType.VIDEO, MediaType.ALBUM, MediaType.PLAYLIST, MediaType.MIX],
id_media: str,
-) -> Track | Video:
- try:
- if media_type == MediaType.TRACK:
- media = Track(session, id_media)
- elif media_type == MediaType.VIDEO:
- media = Video(session, id_media)
- elif media_type == MediaType.ALBUM:
- media = Album(session, id_media)
- elif media_type == MediaType.PLAYLIST:
- media = Playlist(session, id_media)
- elif media_type == MediaType.MIX:
- media = Mix(session, id_media)
- else:
- raise MediaUnknown
- except:
+) -> Track | Video | Album | Playlist | Mix:
+ if media_type == MediaType.TRACK:
+ media = session.track(id_media, with_album=True)
+ elif media_type == MediaType.VIDEO:
+ media = session.video(id_media)
+ elif media_type == MediaType.ALBUM:
+ media = session.album(id_media)
+ elif media_type == MediaType.PLAYLIST:
+ media = session.playlist(id_media)
+ elif media_type == MediaType.MIX:
+ media = session.mix(id_media)
+ else:
raise MediaUnknown
return media