From 3c57db9e825825cf04eb89f6f2abdfb144630fdf Mon Sep 17 00:00:00 2001 From: Robert Honz Date: Mon, 26 Feb 2024 07:34:07 +0100 Subject: ✨ Direct link search can handle invalid links. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tidal_dl_ng/helper/tidal.py | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) (limited to 'tidal_dl_ng/helper') 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 -- cgit v1.3.1