mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
Small fixes
This commit is contained in:
@@ -2,11 +2,13 @@ from mutagen.easyid3 import EasyID3
|
|||||||
from mutagen.id3 import ID3, APIC
|
from mutagen.id3 import ID3, APIC
|
||||||
from mutagen.mp4 import MP4, MP4Cover
|
from mutagen.mp4 import MP4, MP4Cover
|
||||||
|
|
||||||
|
# urllib2 is urllib.request in python3
|
||||||
try:
|
try:
|
||||||
import urllib2
|
import urllib2
|
||||||
except ImportError:
|
except ImportError:
|
||||||
import urllib.request as urllib2
|
import urllib.request as urllib2
|
||||||
|
|
||||||
|
# check if input file title matches with expected title
|
||||||
def compare(file, metadata):
|
def compare(file, metadata):
|
||||||
try:
|
try:
|
||||||
if file.endswith('.mp3'):
|
if file.endswith('.mp3'):
|
||||||
@@ -33,11 +35,13 @@ def embed(music_file, meta_tags, output_ext):
|
|||||||
print('Cannot embed meta-tags into given output extension')
|
print('Cannot embed meta-tags into given output extension')
|
||||||
|
|
||||||
def embed_mp3(music_file, meta_tags, output_ext):
|
def embed_mp3(music_file, meta_tags, output_ext):
|
||||||
|
#EasyID3 is fun to use ;)
|
||||||
artists = []
|
artists = []
|
||||||
for artist in meta_tags['artists']:
|
for artist in meta_tags['artists']:
|
||||||
artists.append(artist['name'])
|
artists.append(artist['name'])
|
||||||
audiofile = EasyID3('Music/' + music_file + output_ext)
|
audiofile = EasyID3('Music/' + music_file + output_ext)
|
||||||
audiofile['artist'] = artists
|
#audiofile['artist'] = artists
|
||||||
|
audiofile['artist'] = ', '.join(artists)
|
||||||
audiofile['albumartist'] = meta_tags['artists'][0]['name']
|
audiofile['albumartist'] = meta_tags['artists'][0]['name']
|
||||||
audiofile['album'] = meta_tags['album']['name']
|
audiofile['album'] = meta_tags['album']['name']
|
||||||
audiofile['title'] = meta_tags['name']
|
audiofile['title'] = meta_tags['name']
|
||||||
@@ -65,7 +69,6 @@ def embed_mp3(music_file, meta_tags, output_ext):
|
|||||||
audiofile.save(v2_version=3)
|
audiofile.save(v2_version=3)
|
||||||
|
|
||||||
def embed_m4a(music_file, meta_tags, output_ext):
|
def embed_m4a(music_file, meta_tags, output_ext):
|
||||||
# eyed serves only mp3 not aac so using mutagen
|
|
||||||
# Apple has specific tags - see mutagen docs -
|
# Apple has specific tags - see mutagen docs -
|
||||||
# http://mutagen.readthedocs.io/en/latest/api/mp4.html
|
# http://mutagen.readthedocs.io/en/latest/api/mp4.html
|
||||||
tags = {'album': '\xa9alb',
|
tags = {'album': '\xa9alb',
|
||||||
@@ -89,7 +92,8 @@ def embed_m4a(music_file, meta_tags, output_ext):
|
|||||||
for artist in meta_tags['artists']:
|
for artist in meta_tags['artists']:
|
||||||
artists.append(artist['name'])
|
artists.append(artist['name'])
|
||||||
audiofile = MP4('Music/' + music_file + output_ext)
|
audiofile = MP4('Music/' + music_file + output_ext)
|
||||||
audiofile[tags['artist']] = artists
|
#audiofile[tags['artist']] = artists
|
||||||
|
audiofile[tags['artist']] = ', '.join(artists)
|
||||||
audiofile[tags['albumartist']] = meta_tags['artists'][0]['name']
|
audiofile[tags['albumartist']] = meta_tags['artists'][0]['name']
|
||||||
audiofile[tags['album']] = meta_tags['album']['name']
|
audiofile[tags['album']] = meta_tags['album']['name']
|
||||||
audiofile[tags['title']] = meta_tags['name']
|
audiofile[tags['title']] = meta_tags['name']
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ try:
|
|||||||
except:
|
except:
|
||||||
from urllib.request import quote
|
from urllib.request import quote
|
||||||
|
|
||||||
|
# method to input (user playlists) and (track when using manual mode)
|
||||||
def input_link(links):
|
def input_link(links):
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
@@ -22,6 +23,7 @@ def input_link(links):
|
|||||||
except ValueError:
|
except ValueError:
|
||||||
print('Choose a valid number!')
|
print('Choose a valid number!')
|
||||||
|
|
||||||
|
# remove first song from .txt
|
||||||
def trim_song(file):
|
def trim_song(file):
|
||||||
with open(file, 'r') as fin:
|
with open(file, 'r') as fin:
|
||||||
data = fin.read().splitlines(True)
|
data = fin.read().splitlines(True)
|
||||||
@@ -62,13 +64,15 @@ def is_spotify(raw_song):
|
|||||||
else:
|
else:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
# generate filename of the song to be downloaded
|
||||||
def generate_filename(title):
|
def generate_filename(title):
|
||||||
raw_title = title.replace(' ', '_')
|
raw_title = title.replace(' ', '_')
|
||||||
|
# slugify removes any special characters
|
||||||
filename = slugify(raw_title, ok='-_()[]{}', lower=False)
|
filename = slugify(raw_title, ok='-_()[]{}', lower=False)
|
||||||
return fix_encoding(filename)
|
return fix_encoding(filename)
|
||||||
|
|
||||||
|
# please respect this user token :)
|
||||||
def generate_token():
|
def generate_token():
|
||||||
# Please respect this user token :)
|
|
||||||
creds = oauth2.SpotifyClientCredentials(
|
creds = oauth2.SpotifyClientCredentials(
|
||||||
client_id='4fe3fecfe5334023a1472516cc99d805',
|
client_id='4fe3fecfe5334023a1472516cc99d805',
|
||||||
client_secret='0f02b7c483c04257984695007a4a8d5c')
|
client_secret='0f02b7c483c04257984695007a4a8d5c')
|
||||||
@@ -79,6 +83,7 @@ def generate_search_URL(song):
|
|||||||
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + quote(song)
|
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + quote(song)
|
||||||
return URL
|
return URL
|
||||||
|
|
||||||
|
# fix encoding issues in python2
|
||||||
def fix_encoding(query):
|
def fix_encoding(query):
|
||||||
if sys.version_info > (3, 0):
|
if sys.version_info > (3, 0):
|
||||||
return query
|
return query
|
||||||
|
|||||||
37
spotdl.py
37
spotdl.py
@@ -1,4 +1,5 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
|
|
||||||
# -*- coding: UTF-8 -*-
|
# -*- coding: UTF-8 -*-
|
||||||
|
|
||||||
from core import metadata
|
from core import metadata
|
||||||
@@ -13,17 +14,20 @@ import sys
|
|||||||
import os
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
|
|
||||||
|
# urllib2 is urllib.request in python3
|
||||||
try:
|
try:
|
||||||
import urllib2
|
import urllib2
|
||||||
except ImportError:
|
except ImportError:
|
||||||
import urllib.request as urllib2
|
import urllib.request as urllib2
|
||||||
|
|
||||||
|
# generate song title to be searched on YouTube
|
||||||
def generate_songname(raw_song):
|
def generate_songname(raw_song):
|
||||||
if misc.is_spotify(raw_song):
|
if misc.is_spotify(raw_song):
|
||||||
tags = generate_metadata(raw_song)
|
tags = generate_metadata(raw_song)
|
||||||
raw_song = tags['artists'][0]['name'] + ' - ' + tags['name']
|
raw_song = tags['artists'][0]['name'] + ' - ' + tags['name']
|
||||||
return misc.fix_encoding(raw_song)
|
return misc.fix_encoding(raw_song)
|
||||||
|
|
||||||
|
# fetch song's metadata from spotify
|
||||||
def generate_metadata(raw_song):
|
def generate_metadata(raw_song):
|
||||||
if misc.is_spotify(raw_song):
|
if misc.is_spotify(raw_song):
|
||||||
meta_tags = spotify.track(raw_song)
|
meta_tags = spotify.track(raw_song)
|
||||||
@@ -78,6 +82,7 @@ def generate_YouTube_URL(raw_song):
|
|||||||
full_link = "youtube.com" + result
|
full_link = "youtube.com" + result
|
||||||
return full_link
|
return full_link
|
||||||
|
|
||||||
|
# parse track from YouTube
|
||||||
def go_pafy(raw_song):
|
def go_pafy(raw_song):
|
||||||
trackURL = generate_YouTube_URL(raw_song)
|
trackURL = generate_YouTube_URL(raw_song)
|
||||||
if trackURL is None:
|
if trackURL is None:
|
||||||
@@ -85,6 +90,7 @@ def go_pafy(raw_song):
|
|||||||
else:
|
else:
|
||||||
return pafy.new(trackURL)
|
return pafy.new(trackURL)
|
||||||
|
|
||||||
|
# title of the YouTube video
|
||||||
def get_YouTube_title(content, number):
|
def get_YouTube_title(content, number):
|
||||||
title = content.title
|
title = content.title
|
||||||
if number is None:
|
if number is None:
|
||||||
@@ -92,6 +98,7 @@ def get_YouTube_title(content, number):
|
|||||||
else:
|
else:
|
||||||
return str(number) + '. ' + title
|
return str(number) + '. ' + title
|
||||||
|
|
||||||
|
# write tracks into list file
|
||||||
def feed_tracks(file, tracks):
|
def feed_tracks(file, tracks):
|
||||||
with open(file, 'a') as fout:
|
with open(file, 'a') as fout:
|
||||||
for item in tracks['items']:
|
for item in tracks['items']:
|
||||||
@@ -101,6 +108,7 @@ def feed_tracks(file, tracks):
|
|||||||
except KeyError:
|
except KeyError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
# fetch user playlists when using -u option
|
||||||
def feed_playlist(username):
|
def feed_playlist(username):
|
||||||
playlists = spotify.user_playlists(username)
|
playlists = spotify.user_playlists(username)
|
||||||
links = []
|
links = []
|
||||||
@@ -113,6 +121,7 @@ def feed_playlist(username):
|
|||||||
playlist = misc.input_link(links)
|
playlist = misc.input_link(links)
|
||||||
results = spotify.user_playlist(playlist['owner']['id'], playlist['id'], fields="tracks,next")
|
results = spotify.user_playlist(playlist['owner']['id'], playlist['id'], fields="tracks,next")
|
||||||
print('')
|
print('')
|
||||||
|
# slugify removes any special characters
|
||||||
file = slugify(playlist['name'], ok='-_()[]{}') + '.txt'
|
file = slugify(playlist['name'], ok='-_()[]{}') + '.txt'
|
||||||
print('Feeding ' + str(playlist['tracks']['total']) + ' tracks to ' + file)
|
print('Feeding ' + str(playlist['tracks']['total']) + ' tracks to ' + file)
|
||||||
tracks = results['tracks']
|
tracks = results['tracks']
|
||||||
@@ -132,6 +141,7 @@ def download_song(content):
|
|||||||
if link is not None:
|
if link is not None:
|
||||||
link.download(filepath='Music/' + music_file + args.input_ext)
|
link.download(filepath='Music/' + music_file + args.input_ext)
|
||||||
|
|
||||||
|
# convert song from input_ext to output_ext
|
||||||
def convert_song(music_file):
|
def convert_song(music_file):
|
||||||
if not args.input_ext == args.output_ext:
|
if not args.input_ext == args.output_ext:
|
||||||
print('Converting ' + music_file + args.input_ext + ' to ' + args.output_ext[1:])
|
print('Converting ' + music_file + args.input_ext + ' to ' + args.output_ext[1:])
|
||||||
@@ -173,7 +183,7 @@ def convert_with_FFmpeg(music_file):
|
|||||||
# libfdk_aac due to copyrights needs to be compiled by end user
|
# 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?
|
# on MacOS brew install ffmpeg --with-fdk-aac will do just that. Other OS?
|
||||||
# https://trac.ffmpeg.org/wiki/Encode/AAC
|
# https://trac.ffmpeg.org/wiki/Encode/AAC
|
||||||
#
|
|
||||||
if os.name == "nt":
|
if os.name == "nt":
|
||||||
ffmpeg_pre = 'Scripts//ffmpeg.exe '
|
ffmpeg_pre = 'Scripts//ffmpeg.exe '
|
||||||
else:
|
else:
|
||||||
@@ -217,52 +227,56 @@ def convert_with_FFmpeg(music_file):
|
|||||||
|
|
||||||
os.remove('Music/' + music_file + args.input_ext)
|
os.remove('Music/' + music_file + args.input_ext)
|
||||||
|
|
||||||
|
# check if input song already exists in Music folder
|
||||||
def check_exists(music_file, raw_song, islist):
|
def check_exists(music_file, raw_song, islist):
|
||||||
files = os.listdir("Music")
|
files = os.listdir("Music")
|
||||||
for file in files:
|
for file in files:
|
||||||
if file.endswith(".temp"):
|
if file.endswith(".temp"):
|
||||||
os.remove("Music/" + file)
|
os.remove("Music/" + file)
|
||||||
continue
|
continue
|
||||||
|
# check if any file with similar name is already present in Music/
|
||||||
if file.startswith(misc.generate_filename(music_file)):
|
if file.startswith(misc.generate_filename(music_file)):
|
||||||
|
# check if the already downloaded song has correct metadata
|
||||||
already_tagged = metadata.compare(file, generate_metadata(raw_song))
|
already_tagged = metadata.compare(file, generate_metadata(raw_song))
|
||||||
|
# if not, remove it and download again without prompt
|
||||||
if misc.is_spotify(raw_song) and not already_tagged:
|
if misc.is_spotify(raw_song) and not already_tagged:
|
||||||
os.remove("Music/" + file)
|
os.remove("Music/" + file)
|
||||||
return False
|
return False
|
||||||
|
# do not prompt and skip the current song if already downloaded when using list
|
||||||
if islist:
|
if islist:
|
||||||
return True
|
return True
|
||||||
|
# if downloading only single song, prompt to re-download
|
||||||
else:
|
else:
|
||||||
prompt = raw_input('Song with same name has already been downloaded. Re-download? (y/n): ').lower()
|
prompt = raw_input('Song with same name has already been downloaded. Re-download? (y/n): ').lower()
|
||||||
|
|
||||||
if prompt == "y":
|
if prompt == "y":
|
||||||
os.remove("Music/" + file)
|
os.remove("Music/" + file)
|
||||||
return False
|
return False
|
||||||
else:
|
else:
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
# download songs from list
|
||||||
def grab_list(file):
|
def grab_list(file):
|
||||||
with open(file, 'r') as listed:
|
with open(file, 'r') as listed:
|
||||||
lines = (listed.read()).splitlines()
|
lines = (listed.read()).splitlines()
|
||||||
# Ignore blank lines in file (if any)
|
# ignore blank lines in file (if any)
|
||||||
try:
|
try:
|
||||||
lines.remove('')
|
lines.remove('')
|
||||||
except ValueError:
|
except ValueError:
|
||||||
pass
|
pass
|
||||||
print('Total songs in list = ' + str(len(lines)) + ' songs')
|
print('Total songs in list = ' + str(len(lines)) + ' songs')
|
||||||
print('')
|
print('')
|
||||||
# Count the number of song being downloaded
|
# nth input song
|
||||||
number = 1
|
number = 1
|
||||||
for raw_song in lines:
|
for raw_song in lines:
|
||||||
try:
|
try:
|
||||||
grab_single(raw_song, number=number)
|
grab_single(raw_song, number=number)
|
||||||
|
# refresh token when it expires (after 1 hour)
|
||||||
except spotipy.oauth2.SpotifyOauthError:
|
except spotipy.oauth2.SpotifyOauthError:
|
||||||
token = misc.generate_token()
|
token = misc.generate_token()
|
||||||
global spotify
|
global spotify
|
||||||
spotify = spotipy.Spotify(auth=token)
|
spotify = spotipy.Spotify(auth=token)
|
||||||
grab_single(raw_song, number=number)
|
grab_single(raw_song, number=number)
|
||||||
|
# detect network problems
|
||||||
except (urllib2.URLError, TypeError, IOError):
|
except (urllib2.URLError, TypeError, IOError):
|
||||||
lines.append(raw_song)
|
lines.append(raw_song)
|
||||||
misc.trim_song(file)
|
misc.trim_song(file)
|
||||||
@@ -277,7 +291,7 @@ def grab_list(file):
|
|||||||
misc.trim_song(file)
|
misc.trim_song(file)
|
||||||
number += 1
|
number += 1
|
||||||
|
|
||||||
# Logic behind preparing the song to download to finishing meta-tags
|
# logic behind downloading some song
|
||||||
def grab_single(raw_song, number=None):
|
def grab_single(raw_song, number=None):
|
||||||
if number:
|
if number:
|
||||||
islist = True
|
islist = True
|
||||||
@@ -298,7 +312,7 @@ def grab_single(raw_song, number=None):
|
|||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
||||||
# Python 3 compatibility
|
# python 3 compatibility
|
||||||
if sys.version_info > (3, 0):
|
if sys.version_info > (3, 0):
|
||||||
raw_input = input
|
raw_input = input
|
||||||
|
|
||||||
@@ -306,10 +320,11 @@ if __name__ == '__main__':
|
|||||||
if not os.path.exists("Music"):
|
if not os.path.exists("Music"):
|
||||||
os.makedirs("Music")
|
os.makedirs("Music")
|
||||||
|
|
||||||
|
# token is mandatory when using Spotify's API
|
||||||
token = misc.generate_token()
|
token = misc.generate_token()
|
||||||
spotify = spotipy.Spotify(auth=token)
|
spotify = spotipy.Spotify(auth=token)
|
||||||
|
|
||||||
# Set up arguments
|
# set up arguments
|
||||||
args = misc.get_arguments()
|
args = misc.get_arguments()
|
||||||
|
|
||||||
if args.song:
|
if args.song:
|
||||||
|
|||||||
Reference in New Issue
Block a user