✨ Track media items will be re-created with album information. Path format refactored.

This commit is contained in:
Robert Honz
2024-03-13 06:40:32 +01:00
parent 832f5ce443
commit bbb6051fdd
2 changed files with 16 additions and 14 deletions
+10 -9
View File
@@ -156,20 +156,19 @@ class Download:
def instantiate_media(
self,
session: Session,
media_type: type[MediaType.TRACK, MediaType.VIDEO, MediaType.ALBUM, MediaType.PLAYLIST, MediaType.MIX],
id_media: str,
) -> Track | Video:
) -> Track | Video | Album | Playlist | Mix:
if media_type == MediaType.TRACK:
media = Track(session, id_media)
media = self.session.track(id_media, with_album=True)
elif media_type == MediaType.VIDEO:
media = Video(session, id_media)
media = self.session.video(id_media)
elif media_type == MediaType.ALBUM:
media = Album(self.session, id_media)
media = self.session.album(id_media)
elif media_type == MediaType.PLAYLIST:
media = Playlist(self.session, id_media)
media = self.session.playlist(id_media)
elif media_type == MediaType.MIX:
media = Mix(self.session, id_media)
media = self.session.mix(id_media)
else:
raise MediaUnknown
@@ -186,7 +185,9 @@ class Download:
) -> (bool, str):
# If no media instance is provided, we need to create the media instance.
if media_id and media_type:
media = self.instantiate_media(self.session, media_type, media_id)
media = self.instantiate_media(media_type, media_id)
elif isinstance(media, Track):
media = self.session.track(media.id, with_album=True)
elif not media:
raise MediaMissing
@@ -342,7 +343,7 @@ class Download:
):
# If no media instance is provided, we need to create the media instance.
if media_id and media_type:
media = self.instantiate_media(self.session, media_type, media_id)
media = self.instantiate_media(media_type, media_id)
elif not media:
raise MediaMissing