mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
321 lines
10 KiB
Python
321 lines
10 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: UTF-8 -*-
|
|
|
|
from core import metadata
|
|
from core import misc
|
|
from bs4 import BeautifulSoup
|
|
from titlecase import titlecase
|
|
from slugify import slugify
|
|
import spotipy
|
|
import spotipy.oauth2 as oauth2
|
|
import pafy
|
|
import sys
|
|
import os
|
|
import subprocess
|
|
|
|
try:
|
|
import urllib2
|
|
except ImportError:
|
|
import urllib.request as urllib2
|
|
|
|
def generate_songname(raw_song):
|
|
if misc.is_spotify(raw_song):
|
|
tags = generate_metadata(raw_song)
|
|
raw_song = tags['artists'][0]['name'] + ' - ' + tags['name']
|
|
return misc.fix_encoding(raw_song)
|
|
|
|
def generate_metadata(raw_song):
|
|
if misc.is_spotify(raw_song):
|
|
meta_tags = spotify.track(raw_song)
|
|
else:
|
|
meta_tags = spotify.search(raw_song, limit=1)['tracks']['items'][0]
|
|
|
|
artist = spotify.artist(meta_tags['artists'][0]['id'])
|
|
album = spotify.album(meta_tags['album']['id'])
|
|
|
|
try:
|
|
meta_tags[u'genre'] = titlecase(artist['genres'][0])
|
|
except IndexError:
|
|
meta_tags[u'genre'] = None
|
|
|
|
meta_tags[u'release_date'] = album['release_date']
|
|
meta_tags[u'copyright'] = album['copyrights'][0]['text']
|
|
meta_tags[u'publisher'] = album['label']
|
|
meta_tags[u'total_tracks'] = album['tracks']['total']
|
|
#import pprint
|
|
#pprint.pprint(meta_tags)
|
|
#pprint.pprint(spotify.album(meta_tags['album']['id']))
|
|
return meta_tags
|
|
|
|
def generate_YouTube_URL(raw_song):
|
|
song = generate_songname(raw_song)
|
|
searchURL = misc.generate_search_URL(song)
|
|
item = urllib2.urlopen(searchURL).read()
|
|
items_parse = BeautifulSoup(item, "html.parser")
|
|
check = 1
|
|
if args.manual:
|
|
links = []
|
|
print(song)
|
|
print('')
|
|
print('0. Skip downloading this song')
|
|
for x in items_parse.find_all('h3', {'class': 'yt-lockup-title'}):
|
|
if not x.find('channel') == -1 or not x.find('googleads') == -1:
|
|
print(str(check) + '. ' + x.get_text())
|
|
links.append(x.find('a')['href'])
|
|
check += 1
|
|
print('')
|
|
result = misc.input_link(links)
|
|
if result is None:
|
|
return None
|
|
else:
|
|
result = items_parse.find_all(
|
|
attrs={'class': 'yt-uix-tile-link'})[0]['href']
|
|
while not result.find('channel') == - \
|
|
1 or not result.find('googleads') == -1:
|
|
result = items_parse.find_all(
|
|
attrs={'class': 'yt-uix-tile-link'})[check]['href']
|
|
check += 1
|
|
full_link = "youtube.com" + result
|
|
return full_link
|
|
|
|
def go_pafy(raw_song):
|
|
trackURL = generate_YouTube_URL(raw_song)
|
|
if trackURL is None:
|
|
return None
|
|
else:
|
|
return pafy.new(trackURL)
|
|
|
|
def get_YouTube_title(content, number):
|
|
title = content.title
|
|
if number is None:
|
|
return title
|
|
else:
|
|
return str(number) + '. ' + title
|
|
|
|
def feed_tracks(file, tracks):
|
|
with open(file, 'a') as fout:
|
|
for item in tracks['items']:
|
|
track = item['track']
|
|
try:
|
|
fout.write(track['external_urls']['spotify'] + '\n')
|
|
except KeyError:
|
|
pass
|
|
|
|
def feed_playlist(username):
|
|
playlists = spotify.user_playlists(username)
|
|
links = []
|
|
check = 1
|
|
for playlist in playlists['items']:
|
|
print(str(check) + '. ' + misc.fix_encoding(playlist['name']) + ' (' + str(playlist['tracks']['total']) + ' tracks)')
|
|
links.append(playlist)
|
|
check += 1
|
|
print('')
|
|
playlist = misc.input_link(links)
|
|
results = spotify.user_playlist(playlist['owner']['id'], playlist['id'], fields="tracks,next")
|
|
print('')
|
|
file = slugify(playlist['name'], ok='-_()[]{}') + '.txt'
|
|
print('Feeding ' + str(playlist['tracks']['total']) + ' tracks to ' + file)
|
|
tracks = results['tracks']
|
|
feed_tracks(file, tracks)
|
|
while tracks['next']:
|
|
tracks = spotify.next(tracks)
|
|
feed_tracks(file, tracks)
|
|
|
|
def download_song(content):
|
|
music_file = misc.generate_filename(content.title)
|
|
if args.input_ext == '.webm':
|
|
link = content.getbestaudio(preftype='webm')
|
|
if link is not None:
|
|
link.download(filepath='Music/' + music_file + args.input_ext)
|
|
else:
|
|
link = content.getbestaudio(preftype='m4a')
|
|
if link is not None:
|
|
link.download(filepath='Music/' + music_file + args.input_ext)
|
|
|
|
def convert_song(music_file):
|
|
if not args.input_ext == args.output_ext:
|
|
print('Converting ' + music_file + args.input_ext + ' to ' + args.output_ext[1:])
|
|
if args.ffmpeg:
|
|
convert_with_FFmpeg(music_file)
|
|
else:
|
|
convert_with_libav(music_file)
|
|
|
|
def convert_with_libav(music_file):
|
|
if os.name == 'nt':
|
|
avconv_path = 'Scripts\\avconv.exe'
|
|
else:
|
|
avconv_path = 'avconv'
|
|
|
|
if args.verbose:
|
|
level = 'debug'
|
|
else:
|
|
level = '0'
|
|
|
|
#print([avconv_path,
|
|
# '-loglevel', level,
|
|
# '-i', 'Music/' + music_file + args.input_ext,
|
|
# '-ab', '192k',
|
|
# 'Music/' + music_file + args.output_ext])
|
|
|
|
subprocess.call([avconv_path,
|
|
'-loglevel', level,
|
|
'-i', 'Music/' + music_file + args.input_ext,
|
|
'-ab', '192k',
|
|
'Music/' + music_file + args.output_ext])
|
|
|
|
os.remove('Music/' + music_file + args.input_ext)
|
|
|
|
def convert_with_FFmpeg(music_file):
|
|
# What are the differences and similarities between ffmpeg, libav, and avconv?
|
|
# https://stackoverflow.com/questions/9477115
|
|
# ffmeg encoders high to lower quality
|
|
# libopus > libvorbis >= libfdk_aac > aac > libmp3lame
|
|
# libfdk_aac due to copyrights needs to be compiled by end user
|
|
# on MacOS brew install ffmpeg --with-fdk-aac will do just that. Other OS?
|
|
# https://trac.ffmpeg.org/wiki/Encode/AAC
|
|
#
|
|
if os.name == "nt":
|
|
ffmpeg_pre = 'Scripts//ffmpeg.exe '
|
|
else:
|
|
ffmpeg_pre = 'ffmpeg '
|
|
|
|
ffmpeg_pre += '-y '
|
|
if not args.verbose:
|
|
ffmpeg_pre += '-hide_banner -nostats -v panic '
|
|
|
|
if args.input_ext == '.m4a':
|
|
if args.output_ext == '.mp3':
|
|
ffmpeg_params = ' -codec:v copy -codec:a libmp3lame -q:a 2 '
|
|
elif output_ext == '.webm':
|
|
ffmpeg_params = ' -c:a libopus -vbr on -b:a 192k -vn '
|
|
else:
|
|
return
|
|
elif args.input_ext == '.webm':
|
|
if args.output_ext == '.mp3':
|
|
ffmpeg_params = ' -ab 192k -ar 44100 -vn '
|
|
elif args.output_ext == '.m4a':
|
|
ffmpeg_params = ' -cutoff 20000 -c:a libfdk_aac -b:a 256k -vn '
|
|
else:
|
|
return
|
|
else:
|
|
print('Unknown formats. Unable to convert.', args.input_ext, args.output_ext)
|
|
return
|
|
|
|
#command = (ffmpeg_pre +
|
|
# '-i "Music/' + music_file + args.input_ext + '"' +
|
|
# ffmpeg_params +
|
|
# '"Music/' + music_file + args.output_ext + '"').split(' ')
|
|
|
|
commandos = (ffmpeg_pre +
|
|
'-i "Music/' + music_file + args.input_ext + '" ' +
|
|
ffmpeg_params +
|
|
'"Music/' + music_file + args.output_ext + '" ')
|
|
#print(command)
|
|
#print(commandos)
|
|
os.system(commandos)
|
|
#subprocess.call(command)
|
|
|
|
os.remove('Music/' + music_file + args.input_ext)
|
|
|
|
def check_exists(music_file, raw_song, islist):
|
|
files = os.listdir("Music")
|
|
for file in files:
|
|
if file.endswith(".temp"):
|
|
os.remove("Music/" + file)
|
|
continue
|
|
|
|
if file.startswith(misc.generate_filename(music_file)):
|
|
|
|
already_tagged = metadata.compare(file, generate_metadata(raw_song))
|
|
|
|
if misc.is_spotify(raw_song) and not already_tagged:
|
|
os.remove("Music/" + file)
|
|
return False
|
|
|
|
if islist:
|
|
return True
|
|
else:
|
|
prompt = raw_input('Song with same name has already been downloaded. Re-download? (y/n): ').lower()
|
|
|
|
if prompt == "y":
|
|
os.remove("Music/" + file)
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def grab_list(file):
|
|
with open(file, 'r') as listed:
|
|
lines = (listed.read()).splitlines()
|
|
# Ignore blank lines in file (if any)
|
|
try:
|
|
lines.remove('')
|
|
except ValueError:
|
|
pass
|
|
print('Total songs in list = ' + str(len(lines)) + ' songs')
|
|
print('')
|
|
# Count the number of song being downloaded
|
|
number = 1
|
|
for raw_song in lines:
|
|
try:
|
|
grab_single(raw_song, number=number)
|
|
except spotipy.oauth2.SpotifyOauthError:
|
|
token = misc.generate_token()
|
|
global spotify
|
|
spotify = spotipy.Spotify(auth=token)
|
|
grab_single(raw_song, number=number)
|
|
except (urllib2.URLError, TypeError, IOError):
|
|
lines.append(raw_song)
|
|
misc.trim_song(file)
|
|
with open(file, 'a') as myfile:
|
|
myfile.write(raw_song)
|
|
print('Failed to download song. Will retry after other songs.')
|
|
continue
|
|
except KeyboardInterrupt:
|
|
misc.grace_quit()
|
|
finally:
|
|
print('')
|
|
misc.trim_song(file)
|
|
number += 1
|
|
|
|
# Logic behind preparing the song to download to finishing meta-tags
|
|
def grab_single(raw_song, number=None):
|
|
if number:
|
|
islist = True
|
|
else:
|
|
islist = False
|
|
content = go_pafy(raw_song)
|
|
if content is None:
|
|
return
|
|
print(get_YouTube_title(content, number))
|
|
music_file = misc.generate_filename(content.title)
|
|
if not check_exists(music_file, raw_song, islist=islist):
|
|
download_song(content)
|
|
print('')
|
|
convert_song(music_file)
|
|
meta_tags = generate_metadata(raw_song)
|
|
if not args.no_metadata:
|
|
metadata.embed(music_file, meta_tags, args.output_ext)
|
|
|
|
if __name__ == '__main__':
|
|
|
|
# Python 3 compatibility
|
|
if sys.version_info > (3, 0):
|
|
raw_input = input
|
|
|
|
os.chdir(sys.path[0])
|
|
if not os.path.exists("Music"):
|
|
os.makedirs("Music")
|
|
|
|
token = misc.generate_token()
|
|
spotify = spotipy.Spotify(auth=token)
|
|
|
|
# Set up arguments
|
|
args = misc.get_arguments()
|
|
|
|
if args.song:
|
|
grab_single(raw_song=args.song)
|
|
elif args.list:
|
|
grab_list(file=args.list)
|
|
elif args.username:
|
|
feed_playlist(username=args.username)
|