mirror of https://github.com/vitiko98/qobuz-dl
123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
import os
|
|
|
|
import requests
|
|
from pathvalidate import sanitize_filename
|
|
from tqdm import tqdm
|
|
|
|
import qobuz_dl.metadata as metadata
|
|
|
|
|
|
def req_tqdm(url, fname, track_name):
|
|
r = requests.get(url, allow_redirects=True, stream=True)
|
|
total = int(r.headers.get("content-length", 0))
|
|
with open(fname, "wb") as file, tqdm(
|
|
total=total,
|
|
unit="iB",
|
|
unit_scale=True,
|
|
unit_divisor=1024,
|
|
desc=track_name,
|
|
bar_format="{n_fmt}/{total_fmt} /// {desc}",
|
|
) as bar:
|
|
for data in r.iter_content(chunk_size=1024):
|
|
size = file.write(data)
|
|
bar.update(size)
|
|
|
|
|
|
def mkDir(dirn):
|
|
try:
|
|
os.mkdir(dirn)
|
|
except FileExistsError:
|
|
print("Warning: folder already exists. Overwriting...")
|
|
|
|
|
|
def getDesc(u, mt):
|
|
return "{} [{}/{}]".format(mt["title"], u["bit_depth"], u["sampling_rate"])
|
|
|
|
|
|
def getBooklet(i, dirn):
|
|
req_tqdm(i, dirn + "/booklet.pdf", "Downloading booklet")
|
|
|
|
|
|
def getCover(i, dirn):
|
|
req_tqdm(i, dirn + "/cover.jpg", "Downloading cover art")
|
|
|
|
|
|
# Download and tag a file
|
|
def downloadItem(dirn, count, parse, meta, album, url, is_track, mp3):
|
|
fname = (
|
|
"{}/{:02}.mp3".format(dirn, count)
|
|
if mp3
|
|
else "{}/{:02}.flac".format(dirn, count)
|
|
)
|
|
func = metadata.tag_mp3 if mp3 else metadata.tag_flac
|
|
desc = getDesc(parse, meta)
|
|
req_tqdm(url, fname, desc)
|
|
func(fname, dirn, meta, album, is_track)
|
|
|
|
|
|
# Iterate over IDs by type calling downloadItem
|
|
def iterateIDs(client, id, path, quality, album=False):
|
|
count = 0
|
|
|
|
if album:
|
|
meta = client.get_album_meta(id)
|
|
album_title = (
|
|
"{} ({})".format(meta["title"], meta["version"])
|
|
if meta["version"]
|
|
else meta["title"]
|
|
)
|
|
print("\nDownloading: {}\n".format(album_title))
|
|
dirT = (
|
|
meta["artist"]["name"],
|
|
album_title,
|
|
meta["release_date_original"].split("-")[0],
|
|
)
|
|
sanitized_title = sanitize_filename("{} - {} [{}]".format(*dirT))
|
|
dirn = path + sanitized_title
|
|
mkDir(dirn)
|
|
getCover(meta["image"]["large"], dirn)
|
|
if "goodies" in meta:
|
|
try:
|
|
getBooklet(meta["goodies"][0]["url"], dirn)
|
|
except Exception as e:
|
|
print("Error: " + e)
|
|
for i in meta["tracks"]["items"]:
|
|
parse = client.get_track_url(i["id"], quality)
|
|
try:
|
|
url = parse["url"]
|
|
except KeyError:
|
|
print("Track is not available for download")
|
|
return
|
|
if "sample" not in parse:
|
|
is_mp3 = True if int(quality) == 5 else False
|
|
downloadItem(dirn, count, parse, i, meta, url, False, is_mp3)
|
|
else:
|
|
print("Demo. Skipping")
|
|
count = count + 1
|
|
else:
|
|
parse = client.get_track_url(id, quality)
|
|
url = parse["url"]
|
|
|
|
if "sample" not in parse:
|
|
meta = client.get_track_meta(id)
|
|
track_title = (
|
|
"{} ({})".format(meta["title"], meta["version"])
|
|
if meta["version"]
|
|
else meta["title"]
|
|
)
|
|
print("\nDownloading: {}\n".format(track_title))
|
|
dirT = (
|
|
meta["album"]["artist"]["name"],
|
|
track_title,
|
|
meta["album"]["release_date_original"].split("-")[0],
|
|
)
|
|
sanitized_title = sanitize_filename("{} - {} [{}]".format(*dirT))
|
|
dirn = path + sanitized_title
|
|
mkDir(dirn)
|
|
getCover(meta["album"]["image"]["large"], dirn)
|
|
is_mp3 = True if int(quality) == 5 else False
|
|
downloadItem(dirn, count, parse, meta, meta, url, True, is_mp3)
|
|
else:
|
|
print("Demo. Skipping")
|
|
print("\nCompleted\n")
|