mirror of
https://github.com/Wojtek242/qobuz-dl.git
synced 2024-10-22 22:50:51 +02:00
36c01a326e
* Handle duplicate IDs with database (close #45) * Prettify the code with f strings * New filename format * Add label tag * Add technical info to the folder for Hi-Res downloads
143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
import os
|
|
import logging
|
|
|
|
from mutagen.flac import FLAC, Picture
|
|
from mutagen.mp3 import EasyMP3
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_title(track_dict):
|
|
title = track_dict["title"]
|
|
version = track_dict.get("version")
|
|
if version:
|
|
title = f"{title} ({version})"
|
|
# for classical works
|
|
if track_dict.get("work"):
|
|
title = "{}: {}".format(track_dict["work"], title)
|
|
|
|
return title
|
|
|
|
|
|
# Use KeyError catching instead of dict.get to avoid empty tags
|
|
def tag_flac(filename, root_dir, final_name, d, album, istrack=True, em_image=False):
|
|
"""
|
|
Tag a FLAC file
|
|
|
|
:param str filename: FLAC file path
|
|
:param str root_dir: Root dir used to get the cover art
|
|
:param str final_name: Final name of the FLAC file (complete path)
|
|
:param dict d: Track dictionary from Qobuz_client
|
|
:param dict album: Album dictionary from Qobuz_client
|
|
:param bool istrack
|
|
:param bool em_image: Embed cover art into file
|
|
"""
|
|
audio = FLAC(filename)
|
|
|
|
audio["TITLE"] = get_title(d)
|
|
|
|
audio["TRACKNUMBER"] = str(d["track_number"]) # TRACK NUMBER
|
|
|
|
if "Disc " in final_name:
|
|
audio["DISCNUMBER"] = str(d["media_number"])
|
|
|
|
try:
|
|
audio["COMPOSER"] = d["composer"]["name"] # COMPOSER
|
|
except KeyError:
|
|
pass
|
|
|
|
try:
|
|
audio["ARTIST"] = d["performer"]["name"] # TRACK ARTIST
|
|
except KeyError:
|
|
if istrack:
|
|
audio["ARTIST"] = d["album"]["artist"]["name"] # TRACK ARTIST
|
|
else:
|
|
audio["ARTIST"] = album["artist"]["name"]
|
|
|
|
try:
|
|
audio["LABEL"] = album["label"]["name"]
|
|
except KeyError:
|
|
pass
|
|
|
|
if istrack:
|
|
audio["GENRE"] = ", ".join(d["album"]["genres_list"]) # GENRE
|
|
audio["ALBUMARTIST"] = d["album"]["artist"]["name"] # ALBUM ARTIST
|
|
audio["TRACKTOTAL"] = str(d["album"]["tracks_count"]) # TRACK TOTAL
|
|
audio["ALBUM"] = d["album"]["title"] # ALBUM TITLE
|
|
audio["DATE"] = d["album"]["release_date_original"]
|
|
else:
|
|
audio["GENRE"] = ", ".join(album["genres_list"]) # GENRE
|
|
audio["ALBUMARTIST"] = album["artist"]["name"] # ALBUM ARTIST
|
|
audio["TRACKTOTAL"] = str(album["tracks_count"]) # TRACK TOTAL
|
|
audio["ALBUM"] = album["title"] # ALBUM TITLE
|
|
audio["DATE"] = album["release_date_original"]
|
|
|
|
if em_image:
|
|
emb_image = os.path.join(root_dir, "cover.jpg")
|
|
multi_emb_image = os.path.join(
|
|
os.path.abspath(os.path.join(root_dir, os.pardir)), "cover.jpg"
|
|
)
|
|
cover_image = emb_image if os.path.isfile(emb_image) else multi_emb_image
|
|
try:
|
|
image = Picture()
|
|
image.type = 3
|
|
image.mime = "image/jpeg"
|
|
image.desc = "cover"
|
|
with open(cover_image, "rb") as img:
|
|
image.data = img.read()
|
|
audio.add_picture(image)
|
|
except Exception as e:
|
|
logger.error(f"Error embedding image: {e}", exc_info=True)
|
|
|
|
audio.save()
|
|
os.rename(filename, final_name)
|
|
|
|
|
|
def tag_mp3(filename, root_dir, final_name, d, album, istrack=True, em_image=False):
|
|
"""
|
|
Tag a mp3 file
|
|
|
|
:param str filename: mp3 file path
|
|
:param str root_dir: Root dir used to get the cover art
|
|
:param str final_name: Final name of the mp3 file (complete path)
|
|
:param dict d: Track dictionary from Qobuz_client
|
|
:param bool istrack: Embed cover art into file
|
|
:param bool em_image: Embed cover art into file
|
|
"""
|
|
# TODO: add embedded cover art support for mp3
|
|
audio = EasyMP3(filename)
|
|
|
|
audio["title"] = get_title(d)
|
|
|
|
audio["tracknumber"] = str(d["track_number"])
|
|
|
|
if "Disc " in final_name:
|
|
audio["discnumber"] = str(d["media_number"])
|
|
|
|
try:
|
|
audio["composer"] = d["composer"]["name"]
|
|
except KeyError:
|
|
pass
|
|
|
|
try:
|
|
audio["artist"] = d["performer"]["name"] # TRACK ARTIST
|
|
except KeyError:
|
|
if istrack:
|
|
audio["artist"] = d["album"]["artist"]["name"] # TRACK ARTIST
|
|
else:
|
|
audio["artist"] = album["artist"]["name"]
|
|
|
|
if istrack:
|
|
audio["genre"] = ", ".join(d["album"]["genres_list"]) # GENRE
|
|
audio["albumartist"] = d["album"]["artist"]["name"] # ALBUM ARTIST
|
|
audio["album"] = d["album"]["title"] # ALBUM TITLE
|
|
audio["date"] = d["album"]["release_date_original"]
|
|
else:
|
|
audio["genre"] = ", ".join(album["genres_list"]) # GENRE
|
|
audio["albumartist"] = album["artist"]["name"] # ALBUM ARTIST
|
|
audio["album"] = album["title"] # ALBUM TITLE
|
|
audio["date"] = album["release_date_original"]
|
|
|
|
audio.save()
|
|
os.rename(filename, final_name)
|