def get_item_name(sp, item_type, item_id):
"""
Fetch the name of the item.
:param sp: Spotify Client
:param item_type: Type of the item
:param item_id: id of the item
:return String indicating the name of the item
"""
if item_type == 'playlist':
name = sp.playlist(playlist_id=item_id, fields='name').get('name')
elif item_type == 'album':
name = sp.album(album_id=item_id).get('name')
elif item_type == 'track':
name = sp.track(track_id=item_id).get('name')
return sanitize(name)
def test_sanitize_flename():
name = sanitize("Dank: Tunes<> $ V The* Dranks |Strike Back/")
assert name == "Dank Tunes V The Dranks Strike Back"
name2 = sanitize("Dank: Tunes<> $ V \" The* Dra?nks |Strike Back/", "#")
assert name2 == "Dank# Tunes## # V # The# Dra#nks #Strike Back#"
def download_songs(songs,
download_directory,
format_string,
skip_mp3,
keep_playlist_order=False):
"""
Downloads songs from the YouTube URL passed to either current directory or download_directory, is it is passed.
:param songs: Dictionary of songs and associated artist
:param download_directory: Location where to save
:param format_string: format string for the file conversion
:param skip_mp3: Whether to skip conversion to MP3
:param keep_playlist_order: Whether to keep original playlist ordering. Also, prefixes songs files with playlist num
"""
log.debug(f"Downloading to {download_directory}")
for song in songs:
query = f"{song.get('artist')} - {song.get('name')} Lyrics".replace(
":", "").replace("\"", "")
download_archive = path.join(download_directory,
'downloaded_songs.txt')
file_name = sanitize(f"{song.get('artist')} - {song.get('name')}",
'#') # youtube-dl automatically replaces with #
if keep_playlist_order:
# add song number prefix
file_name = f"{song.get('playlist_num')} - {file_name}"
file_path = path.join(download_directory, file_name)
outtmpl = f"{file_path}.%(ext)s"
ydl_opts = {
'format':
format_string,
'download_archive':
download_archive,
'outtmpl':
outtmpl,
'default_search':
'ytsearch',
'noplaylist':
True,
'postprocessor_args': [
'-metadata', 'title=' + song.get('name'), '-metadata',
'artist=' + song.get('artist'), '-metadata',
'album=' + song.get('album')
]
}
if not skip_mp3:
mp3_postprocess_opts = {
'key': 'FFmpegExtractAudio',
'preferredcodec': 'mp3',
'preferredquality': '192',
}
ydl_opts['postprocessors'] = [mp3_postprocess_opts.copy()]
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
try:
ydl.download([query])
except Exception as e:
log.debug(e)
print(
'Failed to download: {}, please ensure YouTubeDL is up-to-date. '
.format(query))
continue
if not skip_mp3:
song_file = MP3(path.join(f"{file_path}.mp3"), ID3=EasyID3)
song_file['date'] = song.get('year')
if keep_playlist_order:
song_file['tracknumber'] = str(song.get('playlist_num'))
else:
song_file['tracknumber'] = str(song.get('num')) + '/' + str(
song.get('num_tracks'))
song_file['genre'] = song.get('genre')
song_file.save()
song_file = MP3(path.join(
download_directory,
f"{song.get('artist')} - {song.get('name')}.mp3"),
ID3=ID3)
if song.get('cover') is not None:
song_file.tags['APIC'] = APIC(encoding=3,
mime='image/jpeg',
type=3,
desc=u'Cover',
data=urllib.request.urlopen(
song.get('cover')).read())
song_file.save()
def default_filename(song):
return sanitize(f"{song.get('artist')} - {song.get('name')}"
) # youtube-dl automatically replaces with #
