From e3d780e36d71ef033d26c6fea68d818df8bb2a89 Mon Sep 17 00:00:00 2001 From: Robert Honz Date: Tue, 16 Jan 2024 06:33:23 +0100 Subject: ♻️ Capitalized enum names.. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tidal_dl_ng/download.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'tidal_dl_ng/download.py') diff --git a/tidal_dl_ng/download.py b/tidal_dl_ng/download.py index 949734c..5d219de 100644 --- a/tidal_dl_ng/download.py +++ b/tidal_dl_ng/download.py @@ -138,18 +138,18 @@ class Download: def instantiate_media( self, session: Session, - media_type: type[MediaType.Track, MediaType.Video, MediaType.Album, MediaType.Playlist, MediaType.Mix], + media_type: type[MediaType.TRACK, MediaType.VIDEO, MediaType.ALBUM, MediaType.PLAYLIST, MediaType.MIX], id_media: str, ) -> Track | Video: - if media_type == MediaType.Track: + if media_type == MediaType.TRACK: media = Track(session, id_media) - elif media_type == MediaType.Video: + elif media_type == MediaType.VIDEO: media = Video(session, id_media) - elif media_type == MediaType.Album: + elif media_type == MediaType.ALBUM: media = Album(self.session, id_media) - elif media_type == MediaType.Playlist: + elif media_type == MediaType.PLAYLIST: media = Playlist(self.session, id_media) - elif media_type == MediaType.Mix: + elif media_type == MediaType.MIX: media = Mix(self.session, id_media) else: raise MediaUnknown -- cgit v1.3.1