6FPS/tidal-dl
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

👷 Switch to match function for placeholder replacement.

Browse Source
This commit is contained in:
Robert Honz
2024-04-05 09:39:47 +02:00
parent 301a4c5ba4
commit a9b0b285d1
1 changed files with 67 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tidal_dl_ng/helper/path.py
+67 -66
View File
@@ -61,75 +61,76 @@ def format_path_media(fmt_template: str, media: Track | Album | Playlist | UserP
return result return result
def format_str_media(name: str, media: Track | Album | Playlist | UserPlaylist | Video | Mix) -> str: def format_str_media(name: str, media: Track | Album | Playlist | UserPlaylist | Video | Mix) -> str: # noqa: C901
result: str = name result: str = name
try: try:
if name == "artist_name": match name:
if hasattr(media, "artists"): case "artist_name":
result = name_builder_artist(media) if hasattr(media, "artists"):
elif hasattr(media, "artist"): result = name_builder_artist(media)
result = media.artist.name elif hasattr(media, "artist"):
elif name == "album_artist": result = media.artist.name
result = name_builder_album_artist(media) case "album_artist":
elif name == "track_title": result = name_builder_album_artist(media)
if isinstance(media, Track | Video): case "track_title":
result = name_builder_title(media) if isinstance(media, Track | Video):
elif name == "mix_name": result = name_builder_title(media)
if isinstance(media, Mix): case "mix_name":
result = media.title if isinstance(media, Mix):
elif name == "playlist_name": result = media.title
if isinstance(media, Playlist | UserPlaylist): case "playlist_name":
result = media.name if isinstance(media, Playlist | UserPlaylist):
elif name == "album_title": result = media.name
if isinstance(media, Album): case "album_title":
result = media.name if isinstance(media, Album):
elif isinstance(media, Track): result = media.name
result = media.album.name elif isinstance(media, Track):
elif name == "album_track_num": result = media.album.name
if isinstance(media, Track | Video): case "album_track_num":
num_tracks: int = media.album.num_tracks if hasattr(media, "album") else 1 if isinstance(media, Track | Video):
count_digits: int = int(math.log10(num_tracks)) + 1 num_tracks: int = media.album.num_tracks if hasattr(media, "album") else 1
result = str(media.track_num).zfill(count_digits) count_digits: int = int(math.log10(num_tracks)) + 1
elif name == "album_num_tracks": result = str(media.track_num).zfill(count_digits)
if isinstance(media, Track | Video): case "album_num_tracks":
result = str(media.album.num_tracks if hasattr(media, "album") else 1) if isinstance(media, Track | Video):
elif name == "track_id": result = str(media.album.num_tracks if hasattr(media, "album") else 1)
if isinstance(media, Track | Video): case "track_id":
result = media.id if isinstance(media, Track | Video):
elif name == "playlist_id": result = media.id
if isinstance(media, Playlist): case "playlist_id":
result = media.id if isinstance(media, Playlist):
elif name == "track_duration_seconds": result = media.id
if isinstance(media, Track | Video): case "track_duration_seconds":
result = str(media.duration) if isinstance(media, Track | Video):
elif name == "track_duration_minutes": result = str(media.duration)
if isinstance(media, Track | Video): case "track_duration_minutes":
m, s = divmod(media.duration, 60) if isinstance(media, Track | Video):
result = f"{m:01d}:{s:02d}" m, s = divmod(media.duration, 60)
elif name == "album_duration_seconds": result = f"{m:01d}:{s:02d}"
if isinstance(media, Album): case "album_duration_seconds":
result = str(media.duration) if isinstance(media, Album):
elif name == "album_duration_minutes": result = str(media.duration)
if isinstance(media, Album): case "album_duration_minutes":
m, s = divmod(media.duration, 60) if isinstance(media, Album):
result = f"{m:01d}:{s:02d}" m, s = divmod(media.duration, 60)
elif name == "playlist_duration_seconds": result = f"{m:01d}:{s:02d}"
if isinstance(media, Album): case "playlist_duration_seconds":
result = str(media.duration) if isinstance(media, Album):
elif name == "playlist_duration_minutes": result = str(media.duration)
if isinstance(media, Album): case "playlist_duration_minutes":
m, s = divmod(media.duration, 60) if isinstance(media, Album):
result = f"{m:01d}:{s:02d}" m, s = divmod(media.duration, 60)
elif name == "album_year": result = f"{m:01d}:{s:02d}"
if isinstance(media, Album): case "album_year":
result = str(media.release_date.year) if isinstance(media, Album):
elif name == "video_quality": result = str(media.release_date.year)
if isinstance(media, Video): case "video_quality":
result = media.video_quality if isinstance(media, Video):
elif name == "track_quality": result = media.video_quality
if isinstance(media, Track): case "track_quality":
result = ", ".join(tag for tag in media.media_metadata_tags) if isinstance(media, Track):
result = ", ".join(tag for tag in media.media_metadata_tags)
except Exception as e: except Exception as e:
# TODO: Implement better exception logging. # TODO: Implement better exception logging.
print(e) print(e)