mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
256 lines
10 KiB
Python
256 lines
10 KiB
Python
#!/bin/python
|
|
|
|
import mechanize
|
|
from bs4 import BeautifulSoup
|
|
import pafy
|
|
import os
|
|
import requests
|
|
from os.path import expanduser
|
|
import spotipy
|
|
#import spotipy.util as util
|
|
import eyed3
|
|
|
|
opsys = ''
|
|
spotify = spotipy.Spotify()
|
|
|
|
if not os.path.exists("Music"):
|
|
os.makedirs("Music")
|
|
|
|
script_dir = expanduser("~") + '/YTMusic'
|
|
os.chdir(script_dir)
|
|
|
|
for m in os.listdir('Music/'):
|
|
if m.endswith(".temp") or m.endswith("webm"):
|
|
os.remove('Music/' + m)
|
|
print ''
|
|
|
|
def Main():
|
|
Title = ''
|
|
while True:
|
|
try:
|
|
print('')
|
|
print('')
|
|
raw_song = raw_input('Enter a song/cmd: ')
|
|
print ''
|
|
if raw_song == "exit":
|
|
exit()
|
|
elif raw_song == "play":
|
|
if not Title == '':
|
|
if opsys == 'win':
|
|
if os.path.isfile(script_dir + "\Music\\" + Title + ".mp3"):
|
|
os.system('"' + script_dir + "\Music\\" + Title + ".mp3" + '"')
|
|
else:
|
|
os.system('"' + script_dir + "\Music\\" + Title + ".webm" + '"')
|
|
elif opsys == 'linux':
|
|
if os.path.isfile(script_dir + '/Music/' + Title + '.mp3'):
|
|
os.system('mplayer "' + script_dir + '/Music/' + Title + '.mp3"')
|
|
else:
|
|
os.system('mplayer "' + script_dir + '/Music/' + Title + '.webm"')
|
|
elif raw_song == "lyrics":
|
|
if not Title == '':
|
|
if not label == '':
|
|
url = 'http://www.google.com/search?q=' + Title + ' metrolyrics'
|
|
else:
|
|
url = 'http://www.google.com/search?q=' + label + ' metrolyrics'
|
|
req = requests.get(url)
|
|
response = req.content
|
|
result=response
|
|
link_start=result.find('http://www.metrolyrics.com')
|
|
link_end=result.find('html',link_start+1)
|
|
link = result[link_start:link_end+4]
|
|
lyrics_html = link
|
|
try:
|
|
a = requests.get(lyrics_html)
|
|
#print (lyrics_html)
|
|
html_doc= a.content
|
|
soup = BeautifulSoup(html_doc, 'html.parser')
|
|
#album_name = soup.find(id = "album-name-link").text
|
|
raw_lyrics= (soup.findAll('p', attrs={'class' : 'verse'}))
|
|
test1=unicode.join(u'\n',map(unicode,raw_lyrics))
|
|
test1 = (test1.replace('<p class="verse">','\n'))
|
|
test1 = (test1.replace('</br>',''))
|
|
test1 = (test1.replace('<br>',''))
|
|
test1 = test1.replace('</p>','')
|
|
print (test1)
|
|
except:
|
|
print 'Sorry, could not find lyrics'
|
|
else:
|
|
print 'No log to read from..'
|
|
|
|
elif raw_song == "list":
|
|
f = open('Music/list.txt')
|
|
lines = f.readlines()
|
|
f.close()
|
|
x = 0
|
|
y = 0
|
|
for songie in lines:
|
|
if not songie == '\n' or not songie == '':
|
|
x = x + 1
|
|
print 'Total songs in list = ' + str(x) + ' songs'
|
|
for songie in lines:
|
|
try:
|
|
if not songie == '\n' or not songie == '':
|
|
if (len(songie) == 22 and songie.replace(" ", "%20") == songie) or (songie.find('spotify') > -1):
|
|
song = songie.replace(songie[-1:], "")
|
|
content = spotify.track(song)
|
|
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + (content['artists'][0]['name'] + ' - ' + content['name']).replace(" ", "%20")
|
|
else:
|
|
song = songie.replace(" ", "%20")
|
|
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + song
|
|
song = ''
|
|
|
|
print ''
|
|
br = mechanize.Browser()
|
|
br.set_handle_robots(False)
|
|
br.addheaders = [("User-agent","Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.2.13) Gecko/20101206 Ubuntu/10.10 (maverick) Firefox/3.6.13")]
|
|
# print URL
|
|
items = br.open(URL)
|
|
# print items
|
|
items = items.read()
|
|
# print items
|
|
zoom1 = items.find('yt-uix-tile-link')
|
|
zoom2 = items.find('yt-uix-tile-link', zoom1+1)
|
|
zoom3 = items.find('yt-uix-tile-link', zoom2+1)
|
|
part = items[zoom1-100: zoom2]
|
|
items_parse = BeautifulSoup(part, "html.parser")
|
|
|
|
#items_parse = soup(items, "html.parser")
|
|
br.close()
|
|
first_result = items_parse.find(attrs={'class':'yt-uix-tile-link'})['href']
|
|
|
|
full_link = "youtube.com" + first_result
|
|
#print full_link
|
|
|
|
video = pafy.new(full_link)
|
|
Title = ((video.title).replace("\\", "_").replace("/", "_").replace(":", "_").replace("*", "_").replace("?", "_").replace('"', "_").replace("<", "_").replace(">", "_").replace("|", "_").replace(" ", "_"))
|
|
y = y + 1
|
|
print str(y) + '. ' + ((video.title).encode("utf-8"))
|
|
if os.path.exists("Music/" + Title + ".webm.temp"):
|
|
os.remove("Music/" + Title + ".webm.temp")
|
|
download = 1
|
|
if os.path.exists("Music/" + Title + ".mp3"):
|
|
audiofile = eyed3.load("Music/" + Title + '.mp3')
|
|
if ((len(raw_song) == 22 and raw_song.replace(" ", "%20") == raw_song) or (raw_song.find('spotify') > -1)) and not audiofile.tag.title == content['name']:
|
|
os.remove("Music/" + Title + '.mp3')
|
|
else:
|
|
with open('Music/list.txt', 'r') as fin:
|
|
data = fin.read().splitlines(True)
|
|
with open('Music/list.txt', 'w') as fout:
|
|
fout.writelines(data[1:])
|
|
download = 0
|
|
if download == 1:
|
|
a = video.getbestaudio(preftype='webm')
|
|
a.download(filepath="Music/" + Title + ".webm")
|
|
with open('Music/list.txt', 'r') as fin:
|
|
data = fin.read().splitlines(True)
|
|
with open('Music/list.txt', 'w') as fout:
|
|
fout.writelines(data[1:])
|
|
print ''
|
|
print 'Converting ' + Title + '.webm' + ' to mp3..'
|
|
os.system('sudo avconv -loglevel 0 -i ' + script_dir + '"/Music/' + Title + '.webm" ' + script_dir + '"/Music/' + Title + '.mp3"')
|
|
os.remove("Music/" + Title + '.webm')
|
|
os.system('sudo chmod 777 "' + script_dir + '/Music/' + Title + '.mp3"')
|
|
if (len(songie) == 22 and songie.replace(" ", "%20") == songie) or (songie.find('spotify') > -1):
|
|
print 'Fixing meta-tags..'
|
|
audiofile = eyed3.load("Music/" + Title + '.mp3')
|
|
audiofile.tag.artist = content['artists'][0]['name']
|
|
audiofile.tag.album = content['album']['name']
|
|
audiofile.tag.title = content['name']
|
|
os.system("wget -q " + content['album']['images'][0]['url'] + " -O " + script_dir + "/Music/last_albumart.jpg")
|
|
bla = open("Music/last_albumart.jpg","rb").read()
|
|
audiofile.tag.images.set(3,bla,"image/jpeg")
|
|
audiofile.tag.save()
|
|
else:
|
|
with open('Music/list.txt', 'r') as fin:
|
|
data = fin.read().splitlines(True)
|
|
with open('Music/list.txt', 'w') as fout:
|
|
fout.writelines(data[1:])
|
|
|
|
except KeyboardInterrupt:
|
|
Main()
|
|
except:
|
|
lines.append(songie)
|
|
with open('Music/list.txt', 'r') as fin:
|
|
data = fin.read().splitlines(True)
|
|
with open('Music/list.txt', 'w') as fout:
|
|
fout.writelines(data[1:])
|
|
with open("Music/list.txt", "a") as myfile:
|
|
myfile.write(songie)
|
|
print 'Could not complete a Song download, will try later..'
|
|
|
|
else:
|
|
song = raw_song.replace(" ", "%20")
|
|
br = mechanize.Browser()
|
|
br.set_handle_robots(False)
|
|
br.addheaders = [("User-agent","Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.2.13) Gecko/20101206 Ubuntu/10.10 (maverick) Firefox/3.6.13")]
|
|
if (len(raw_song) == 22 and raw_song == song) or (raw_song.find('spotify') > -1):
|
|
content = spotify.track(song)
|
|
label = (content['artists'][0]['name'] + ' - ' + content['name']).replace(" ", "%20")
|
|
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + label
|
|
else:
|
|
URL = "https://www.youtube.com/results?sp=EgIQAQ%253D%253D&q=" + song
|
|
items = br.open(URL).read()
|
|
|
|
zoom1 = items.find('yt-uix-tile-link')
|
|
zoom2 = items.find('yt-uix-tile-link', zoom1+1)
|
|
zoom3 = items.find('yt-uix-tile-link', zoom2+1)
|
|
part = items[zoom1-100: zoom2]
|
|
items_parse = BeautifulSoup(part, "html.parser")
|
|
|
|
#items_parse = soup(items, "html.parser")
|
|
br.close()
|
|
first_result = items_parse.find(attrs={'class':'yt-uix-tile-link'})['href']
|
|
|
|
full_link = "youtube.com" + first_result
|
|
#print full_link
|
|
|
|
video = pafy.new(full_link)
|
|
Title = ((video.title).replace("\\", "_").replace("/", "_").replace(":", "_").replace("*", "_").replace("?", "_").replace('"', "_").replace("<", "_").replace(">", "_").replace("|", "_").replace(" ", "_"))
|
|
print ((video.title).encode("utf-8"))
|
|
if os.path.exists("Music/" + Title + ".webm.temp"):
|
|
os.remove("Music/" + Title + ".webm.temp")
|
|
|
|
download = 1
|
|
|
|
if os.path.exists("Music/" + Title + ".mp3"):
|
|
audiofile = eyed3.load("Music/" + Title + '.mp3')
|
|
if ((len(raw_song) == 22 and raw_song.replace(" ", "%20") == raw_song) or (raw_song.find('spotify') > -1)) and not audiofile.tag.title == content['name']:
|
|
os.remove("Music/" + Title + '.mp3')
|
|
else:
|
|
prompt = raw_input('Song with same name has already been downloaded.. re-download? (y/n/play): ')
|
|
if prompt == "y":
|
|
os.remove("Music/" + Title + ".mp3")
|
|
download = 1
|
|
elif prompt =="play":
|
|
download = 0
|
|
if opsys == 'win':
|
|
if os.path.isfile(script_dir + "\Music\\" + Title + ".mp3"):
|
|
os.system('"' + script_dir + "\Music\\" + Title + ".mp3" + '"')
|
|
else:
|
|
if os.path.isfile(script_dir + '/Music/' + Title + '.mp3'):
|
|
os.system('mplayer "' + script_dir + '/Music/' + Title + '.mp3"')
|
|
else:
|
|
download = 0
|
|
if download == 1:
|
|
a = video.getbestaudio(preftype="webm")
|
|
a.download(filepath="Music/" + Title + ".webm")
|
|
print ''
|
|
print 'Converting ' + Title + '.webm' + ' to mp3..'
|
|
os.system('sudo avconv -loglevel 0 -i ' + script_dir + '"/Music/' + Title + '.webm" ' + script_dir + '"/Music/' + Title + '.mp3"')
|
|
os.remove("Music/" + Title + '.webm')
|
|
os.system('sudo chmod 777 "' + script_dir + '/Music/' + Title + '.mp3"')
|
|
if (len(raw_song) == 22 and raw_song.replace(" ", "%20") == raw_song) or (raw_song.find('spotify') > -1):
|
|
print 'Fixing meta-tags..'
|
|
audiofile = eyed3.load("Music/" + Title + '.mp3')
|
|
audiofile.tag.artist = content['artists'][0]['name']
|
|
audiofile.tag.album = content['album']['name']
|
|
audiofile.tag.title = content['name']
|
|
os.system("wget -q " + content['album']['images'][0]['url'] + " -O " + script_dir + "/Music/last_albumart.jpg")
|
|
bla = open("Music/last_albumart.jpg","rb").read()
|
|
audiofile.tag.images.set(3,bla,"image/jpeg")
|
|
audiofile.tag.save()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
Main()
|