|
|
import shutil
|
|
|
import os
|
|
|
import yt_dlp
|
|
|
from .yt_dlp_logger import YtDlpLogger, yt_dlp_log_hook
|
|
|
|
|
|
MEDIA_FOLDER = '/var/www/html/storage/app/music'
|
|
|
|
|
|
|
|
|
def download_album(album, artist):
|
|
|
"""
|
|
|
Take an artist and album dict and use yt-dlp to download the album to the local filestore.
|
|
|
:param album: Dict of album data
|
|
|
:return: dict of save data
|
|
|
"""
|
|
|
response = {
|
|
|
'artist': {},
|
|
|
'album': {},
|
|
|
}
|
|
|
artist_path = MEDIA_FOLDER + '/%s' % artist.get('name')
|
|
|
if not os.path.exists(artist_path):
|
|
|
# Make artist folder and copy the artist image, add
|
|
|
os.mkdir(artist_path)
|
|
|
shutil.copy2(artist.get('image'), artist_path + '/artist.jpg')
|
|
|
response['artist'] = {'url_local': artist_path}
|
|
|
|
|
|
# Create album folder
|
|
|
album_title = album.get('name')
|
|
|
album_path = artist_path + '/%s' % album_title
|
|
|
if not os.path.exists(album_path):
|
|
|
os.mkdir(album_path)
|
|
|
shutil.copy2(album.get('image'), album_path + '/album.jpg')
|
|
|
|
|
|
# Download album
|
|
|
ydl_opts = {
|
|
|
'logger': YtDlpLogger(),
|
|
|
'progress_hooks': [yt_dlp_log_hook],
|
|
|
'format': 'mp3/bestaudio/best',
|
|
|
'outtmpl': album_path + '/%(title)s.%(ext)s',
|
|
|
# ℹ️ See help(yt_dlp.postprocessor) for a list of available Postprocessors and their arguments
|
|
|
'postprocessors': [{ # Extract audio using ffmpeg
|
|
|
'key': 'FFmpegExtractAudio',
|
|
|
'preferredcodec': 'mp3',
|
|
|
}]
|
|
|
}
|
|
|
download_url = 'https://music.youtube.com/' + album.get('url_remote')
|
|
|
|
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
|
try:
|
|
|
ydl.download(download_url)
|
|
|
except Exception as e:
|
|
|
print('yt-dlp download failed: =========')
|
|
|
print(e)
|
|
|
response['album'] = {'url_local': album_path}
|
|
|
return response
|
