mirror of
https://github.com/cooperhammond/irs.git
synced 2025-02-28 23:26:48 +00:00
Redid the whole thing. Again.
This commit is contained in:
Kepoor Hampond
parent
cab5ee0c37
commit
db64eaa823
212
irs.py
212
irs.py
|
|
@ -1,212 +0,0 @@
|
||||||
import mutagen.id3, mutagen.easyid3, mutagen.mp3, youtube_dl
|
|
||||||
import urllib.request, urllib.parse, re, sys, os, requests
|
|
||||||
from bs4 import BeautifulSoup
|
|
||||||
|
|
||||||
def search_google(song, artist, search_terms=""):
|
|
||||||
def visible(element):
|
|
||||||
if element.parent.name in ['style', 'script', '[document]', 'head', 'title']:
|
|
||||||
return False
|
|
||||||
elif re.match('<!--.*-->', str(element)):
|
|
||||||
return False
|
|
||||||
return True
|
|
||||||
string = "%s %s %s" % (song, artist, search_terms)
|
|
||||||
filename = 'http://www.google.com/search?q=' + urllib.parse.quote_plus(string)
|
|
||||||
hdr = {
|
|
||||||
'User-Agent':'Mozilla/5.0',
|
|
||||||
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
|
|
||||||
}
|
|
||||||
texts = BeautifulSoup(urllib.request.urlopen(urllib.request.Request(filename, \
|
|
||||||
headers=hdr)).read(), 'html.parser').findAll(text=True)
|
|
||||||
return list(filter(visible, texts))
|
|
||||||
|
|
||||||
def parse_metadata(song, artist, location, filename, tracknum="", album=""):
|
|
||||||
googled = search_google(song, artist)
|
|
||||||
mp3file = mutagen.mp3.MP3("%s/%s" % (location, filename), ID3=mutagen.easyid3.EasyID3)
|
|
||||||
print ("%s Metadata parsing:" % color('[+]','OKBLUE'))
|
|
||||||
|
|
||||||
# Song title
|
|
||||||
mp3file['title'] = song
|
|
||||||
mp3file.save()
|
|
||||||
print ("\t%s Title parsed: " % color('[+]','OKGREEN') + mp3file['title'][0])
|
|
||||||
|
|
||||||
# Artist
|
|
||||||
mp3file['artist'] = artist
|
|
||||||
mp3file.save()
|
|
||||||
print ("\t%s Artist parsed: " % color('[+]','OKGREEN') + mp3file['artist'][0])
|
|
||||||
|
|
||||||
# Album
|
|
||||||
if album == "":
|
|
||||||
for i, j in enumerate(googled):
|
|
||||||
if "Album:" in j:
|
|
||||||
album = (googled[i + 1])
|
|
||||||
try:
|
|
||||||
mp3file['album'] = album
|
|
||||||
print ("\t%s Album parsed: " % color('[+]','OKGREEN') + mp3file['album'][0])
|
|
||||||
except Exception:
|
|
||||||
mp3file['album'] = album
|
|
||||||
print ("\t%s Album not parsed" % color('[-]','FAIL'))
|
|
||||||
mp3file.save()
|
|
||||||
|
|
||||||
# Album art
|
|
||||||
if mp3file['album'][0] != "":
|
|
||||||
try:
|
|
||||||
download_album_art(mp3file['album'][0], artist, location=location)
|
|
||||||
embed_mp3("%s/cover.jpg" % location, "%s/%s" % (location, filename))
|
|
||||||
print ("\t%s Album art parsed" % color('[+]','OKGREEN'))
|
|
||||||
except Exception as e:
|
|
||||||
print ("\t%s Album art not parsed" % color('[-]','FAIL'))
|
|
||||||
|
|
||||||
# Release date
|
|
||||||
for i, j in enumerate(googled):
|
|
||||||
if "Released:" in j:
|
|
||||||
date = (googled[i + 1])
|
|
||||||
try:
|
|
||||||
mp3file['date'] = date
|
|
||||||
print ("\t%s Release date parsed" % color('[+]','OKGREEN'))
|
|
||||||
except Exception:
|
|
||||||
mp3file['date'] = ""
|
|
||||||
mp3file.save()
|
|
||||||
|
|
||||||
# Track number
|
|
||||||
if tracknum != "":
|
|
||||||
mp3file['tracknumber'] = str(tracknum)
|
|
||||||
mp3file.save()
|
|
||||||
|
|
||||||
print ("\n%s \"%s\" downloaded successfully!\n" % (color('[+]','OKGREEN'), song))
|
|
||||||
|
|
||||||
def embed_mp3(art_location, song_location):
|
|
||||||
music = mutagen.id3.ID3(song_location)
|
|
||||||
music.delall("APIC")
|
|
||||||
music.add(mutagen.id3.APIC(encoding=0, mime="image/jpg", type=3, desc='', \
|
|
||||||
data=open(art_location, 'rb').read()))
|
|
||||||
music.save()
|
|
||||||
|
|
||||||
def download_album_art(album, artist, location=""):
|
|
||||||
try:
|
|
||||||
search = "%s %s" % (album, artist)
|
|
||||||
url = "http://www.seekacover.com/cd/" + urllib.parse.quote_plus(search)
|
|
||||||
soup = BeautifulSoup(requests.get(url).text, 'html.parser')
|
|
||||||
done = False
|
|
||||||
for img in soup.findAll('img'):
|
|
||||||
if done == False:
|
|
||||||
try:
|
|
||||||
if search.lower() in img['title'].lower():
|
|
||||||
urllib.request.urlretrieve(img['src'], "%s/cover.jpg" % location)
|
|
||||||
done = True
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def strip_special_chars(string):
|
|
||||||
special_chars = "\ / : * ? \" < > | - ( )".split(" ")
|
|
||||||
for char in special_chars:
|
|
||||||
string.replace(char, "")
|
|
||||||
return string
|
|
||||||
|
|
||||||
def color(text, type):
|
|
||||||
types = {'HEADER': '\033[95m', 'OKBLUE': '\033[94m', 'OKGREEN': '\033[92m',
|
|
||||||
'WARNING': '\033[93m','FAIL': '\033[91m','ENDC': '\033[0m','BOLD': '\033[1m'
|
|
||||||
,'UNDERLINE': '\033[4m'}
|
|
||||||
return types[type] + text + types['ENDC']
|
|
||||||
|
|
||||||
def find_mp3(song, artist):
|
|
||||||
search_terms = song + " " + artist
|
|
||||||
print ("\"%s\" by %s" % (song, artist))
|
|
||||||
query_string = urllib.parse.urlencode({"search_query" : (search_terms)})
|
|
||||||
html_content = urllib.request.urlopen("http://www.youtube.com/results?" + query_string)
|
|
||||||
search_results = re.findall(r'href=\"\/watch\?v=(.{11})', html_content.read().decode())
|
|
||||||
in_title = False
|
|
||||||
i = -1
|
|
||||||
given_up_score = 0
|
|
||||||
while in_title == False:
|
|
||||||
i += 1
|
|
||||||
given_up_score += 1
|
|
||||||
if given_up_score >= 10:
|
|
||||||
in_title = True
|
|
||||||
audio_url = ("http://www.youtube.com/watch?v=" + search_results[i])
|
|
||||||
title = strip_special_chars((BeautifulSoup(urllib.request.urlopen(audio_url), 'html.parser')).title.string.lower())
|
|
||||||
song_title = song.lower().split("/")
|
|
||||||
for song in song_title:
|
|
||||||
if strip_special_chars(song) in strip_special_chars(title):
|
|
||||||
in_title = True
|
|
||||||
return search_results[i]
|
|
||||||
|
|
||||||
def rip_album(album, artist, tried=False, search="album"):
|
|
||||||
visible_texts = search_google(album, artist, search)
|
|
||||||
try:
|
|
||||||
songs = []
|
|
||||||
num = True
|
|
||||||
for i, j in enumerate(visible_texts):
|
|
||||||
if 'Songs' in j:
|
|
||||||
if visible_texts[i + 1] == "1":
|
|
||||||
indexed = i
|
|
||||||
while num == True:
|
|
||||||
try:
|
|
||||||
if type(int(visible_texts[indexed])) is int:
|
|
||||||
a = visible_texts[indexed + 1]
|
|
||||||
songs.append(a)
|
|
||||||
indexed += 1
|
|
||||||
except:
|
|
||||||
indexed += 1
|
|
||||||
if indexed >= 1000:
|
|
||||||
num = False
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
|
|
||||||
for i, j in enumerate(songs):
|
|
||||||
rip_mp3(j, artist, part_of_album=True, album=album, tracknum=i + 1)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
if str(e) == "local variable 'indexed' referenced before assignment" or str(e) == 'list index out of range':
|
|
||||||
if tried != True:
|
|
||||||
print ("%s Trying to find album ..." % color('[*]','OKBLUE'))
|
|
||||||
rip_album(album, artist, tried=True, search="")
|
|
||||||
else:
|
|
||||||
print ("%s Could not find album '%s'" % (color('[-]','FAIL'), album))
|
|
||||||
else:
|
|
||||||
print ("%s There was an error with getting the contents \
|
|
||||||
of the album '%s'" % (color('[-]','FAIL'), album))
|
|
||||||
|
|
||||||
def rip_mp3(song, artist, part_of_album=False, album="", tracknum=""):
|
|
||||||
audio_code = find_mp3(song, artist)
|
|
||||||
filename = strip_special_chars(song) + ".mp3"
|
|
||||||
ydl_opts = {
|
|
||||||
'format': 'bestaudio/best',
|
|
||||||
#'quiet': True,
|
|
||||||
'postprocessors': [{
|
|
||||||
'key': 'FFmpegExtractAudio',
|
|
||||||
'preferredcodec': 'mp3',
|
|
||||||
}],
|
|
||||||
}
|
|
||||||
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
||||||
ydl.download(["http://www.youtube.com/watch?v=" + audio_code])
|
|
||||||
|
|
||||||
artist_folder = artist
|
|
||||||
if not os.path.isdir(artist_folder):
|
|
||||||
os.makedirs(artist_folder)
|
|
||||||
if not part_of_album:
|
|
||||||
location = artist_folder
|
|
||||||
|
|
||||||
if album != "" and part_of_album:
|
|
||||||
album_folder = artist + "/" + album
|
|
||||||
if not os.path.isdir(album_folder):
|
|
||||||
os.makedirs(album_folder)
|
|
||||||
location = album_folder
|
|
||||||
|
|
||||||
for file in os.listdir("."):
|
|
||||||
if audio_code in file:
|
|
||||||
os.rename(file, location + "/" + filename)
|
|
||||||
|
|
||||||
parse_metadata(song, artist, location, filename, tracknum=tracknum, album=album)
|
|
||||||
|
|
||||||
def main():
|
|
||||||
media = sys.argv[1]
|
|
||||||
args = " ".join(sys.argv[2:]).split(" by ")
|
|
||||||
if media == "song":
|
|
||||||
rip_mp3(args[0], args[1])
|
|
||||||
elif media == "album":
|
|
||||||
rip_album(args[0], args[1])
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
||||||
|
|
@ -1,4 +0,0 @@
|
||||||
mutagen==1.31
|
|
||||||
requests==2.11.1
|
|
||||||
youtube_dl==2016.7.7
|
|
||||||
beautifulsoup4==4.5.1
|
|
||||||
Loading…
Reference in a new issue