diff options
| author | Robert Honz <exislow+github@gmail.com> | 2024-02-26 07:34:07 +0100 |
|---|---|---|
| committer | Robert Honz <exislow+github@gmail.com> | 2024-02-26 07:34:07 +0100 |
| commit | 3c57db9e825825cf04eb89f6f2abdfb144630fdf (patch) | |
| tree | d4f9b64fbbff63b323105f18cf8411b950df1cf7 /tidal_dl_ng/helper | |
| parent | d845f40e540f33ec719f1cc3d6b2448456b009ac (diff) | |
| download | tidal-dl-ng-3c57db9e825825cf04eb89f6f2abdfb144630fdf.tar.gz | |
✨ Direct link search can handle invalid links.
Diffstat (limited to 'tidal_dl_ng/helper')
| -rw-r--r-- | tidal_dl_ng/helper/tidal.py | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/tidal_dl_ng/helper/tidal.py b/tidal_dl_ng/helper/tidal.py index bdb2abf..e13c909 100644 --- a/tidal_dl_ng/helper/tidal.py +++ b/tidal_dl_ng/helper/tidal.py @@ -112,17 +112,20 @@ def instantiate_media( media_type: type[MediaType.TRACK, MediaType.VIDEO, MediaType.ALBUM, MediaType.PLAYLIST, MediaType.MIX], id_media: str, ) -> Track | Video: - 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: + 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: raise MediaUnknown return media |