summaryrefslogtreecommitdiff
path: root/tidal_dl_ng/download.py
diff options
context:
space:
mode:
authorRobert Honz <exislow+github@gmail.com>2024-01-16 06:33:23 +0100
committerRobert Honz <exislow+github@gmail.com>2024-01-16 06:33:23 +0100
commite3d780e36d71ef033d26c6fea68d818df8bb2a89 (patch)
tree3a6f13f7cefd9557ef8f8f5103eddfc991e52dc9 /tidal_dl_ng/download.py
parentf61d7e6d2e6fe4b0613b819f2b50cf7763930ce7 (diff)
downloadtidal-dl-ng-e3d780e36d71ef033d26c6fea68d818df8bb2a89.tar.gz
♻️ Capitalized enum names..
Diffstat (limited to 'tidal_dl_ng/download.py')
-rw-r--r--tidal_dl_ng/download.py12
1 files changed, 6 insertions, 6 deletions
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