mirror of
https://github.com/Wojtek242/qobuz-dl.git
synced 2024-10-22 22:50:51 +02:00
151 lines
5.9 KiB
Python
151 lines
5.9 KiB
Python
import os
|
|
|
|
from mutagen.flac import FLAC
|
|
from mutagen.mp3 import EasyMP3
|
|
from pathvalidate import sanitize_filename
|
|
|
|
|
|
def tag_flac(file, path, d, album, istrack=True):
|
|
audio = FLAC(file)
|
|
try:
|
|
d["version"]
|
|
except KeyError:
|
|
audio["TITLE"] = d["title"]
|
|
dversion_exist = 0
|
|
else:
|
|
if d["version"] is None:
|
|
audio["TITLE"] = d["title"] # TRACK TITLE
|
|
dversion_exist = 0
|
|
else:
|
|
audio["TITLE"] = d["title"] + " " + "(" + d["version"] + ")"
|
|
dversion_exist = 1
|
|
# if d["version"] is None:
|
|
# audio["TITLE"] = d["title"]# TRACK TITLE
|
|
# else:
|
|
# audio["TITLE"] = d["title"] + ' ' + '(' + d["version"] + ')'
|
|
|
|
audio["TRACKNUMBER"] = str(d["track_number"]) # TRACK 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"]
|
|
|
|
if istrack:
|
|
if dversion_exist == 0:
|
|
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["YEAR"] = d["album"]["release_date_original"].split("-")[0]
|
|
else:
|
|
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"] + " " + "(" + d["album"]["version"] + ")"
|
|
) # ALBUM TITLE
|
|
audio["YEAR"] = d["album"]["release_date_original"].split("-")[0]
|
|
else:
|
|
if dversion_exist == 0:
|
|
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["YEAR"] = album["release_date_original"].split("-")[0] # YEAR
|
|
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["version"] + ")"
|
|
) # ALBUM TITLE
|
|
audio["YEAR"] = album["release_date_original"].split("-")[0] # YEAR
|
|
|
|
audio.save()
|
|
if dversion_exist == 1:
|
|
title = sanitize_filename(d["title"] + " " + "(" + d["version"] + ")")
|
|
else:
|
|
title = sanitize_filename(d["title"])
|
|
try:
|
|
os.rename(file, "{}/{:02}. {}.flac".format(path, d["track_number"], title))
|
|
except FileExistsError:
|
|
print("File already exists. Skipping...")
|
|
|
|
|
|
def tag_mp3(file, path, d, album, istrack=True): # needs to be fixed
|
|
audio = EasyMP3(file)
|
|
try:
|
|
d["version"]
|
|
except KeyError:
|
|
audio["TITLE"] = d["title"]
|
|
dversion_exist = 0
|
|
else:
|
|
if d["version"] is None:
|
|
audio["TITLE"] = d["title"] # TRACK TITLE
|
|
dversion_exist = 0
|
|
else:
|
|
audio["TITLE"] = d["title"] + " " + "(" + d["version"] + ")"
|
|
dversion_exist = 1
|
|
|
|
audio["tracknumber"] = str(d["track_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:
|
|
if dversion_exist == 1:
|
|
audio["genre"] = ", ".join(d["album"]["genres_list"]) # GENRE
|
|
audio["albumartist"] = d["album"]["artist"]["name"] # ALBUM ARTIST
|
|
audio["album"] = (
|
|
d["album"]["title"] + " " + "(" + d["album"]["version"] + ")"
|
|
) # ALBUM TITLE
|
|
audio["date"] = d["album"]["release_date_original"].split("-")[0]
|
|
else:
|
|
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"].split("-")[0]
|
|
else:
|
|
if album["version"] is not None:
|
|
audio["GENRE"] = ", ".join(album["genres_list"]) # GENRE
|
|
audio["albumartist"] = album["artist"]["name"] # ALBUM ARTIST
|
|
try:
|
|
album["version"]
|
|
except KeyError:
|
|
audio["album"] = album["title"]
|
|
else:
|
|
audio["album"] = (
|
|
album["title"] + " " + "(" + album["version"] + ")"
|
|
) # ALBUM TITLE
|
|
audio["date"] = album["release_date_original"].split("-")[0] # YEAR
|
|
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"].split("-")[0] # YEAR
|
|
|
|
audio.save()
|
|
if dversion_exist == 1:
|
|
title = sanitize_filename(d["title"] + " " + "(" + d["version"] + ")")
|
|
else:
|
|
title = sanitize_filename(d["title"])
|
|
try:
|
|
os.rename(file, "{}/{:02}. {}.mp3".format(path, d["track_number"], title))
|
|
except FileExistsError:
|
|
print("File already exists. Skipping...")
|