✨ Direct link search can handle invalid links.

This commit is contained in:
Robert Honz
2024-02-26 07:34:07 +01:00
parent d845f40e54
commit 3c57db9e82
2 changed files with 21 additions and 12 deletions
+14 -11
View File
@@ -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