f567a3e093
i also made an api key mandatory to ping the last fm server at all, so it wont try with an empty key
79 lines
3.1 KiB
Python
79 lines
3.1 KiB
Python
import os
|
|
from mutagen.easyid3 import EasyID3
|
|
from mutagen.mp3 import MP3
|
|
import requests, ast, time, math, argparse, json
|
|
|
|
loading = ["-","\\","|","/"]
|
|
songFiles = os.listdir(r'./sound')
|
|
parser=argparse.ArgumentParser(description="Options for the generation of the song database")
|
|
parser.add_argument('-k','--apikey', help='String: LastFM api key', default="")
|
|
parser.add_argument('-m', '--mode', help='new/update: Remake database or update current', default= "update")
|
|
parser.add_argument('-a', '--art', help="True/False: Add art to the database using LastFm (takes minimum 0.25s per song)", default="True")
|
|
args = parser.parse_args()
|
|
apikeylastfm = args.apikey
|
|
# if you want to set the api key permenantly for your setup just uncomment the next line
|
|
# apikeylastfm = "KeyHere"
|
|
if args.mode == "update":
|
|
try:
|
|
with open('songDatabase.json', 'r') as handle:
|
|
songDatabaseList = json.load(handle)
|
|
except:
|
|
songDatabaseList=[]
|
|
|
|
for i in songDatabaseList:
|
|
try:
|
|
songFiles.index(i["file"]) != -1
|
|
except:
|
|
print("deleted: " + i["file"] + " from database")
|
|
songDatabaseList.pop(songDatabaseList.index(i))
|
|
|
|
for i in songDatabaseList:
|
|
songFiles.pop(songFiles.index(i["file"]))
|
|
print("new songs: " + str(songFiles))
|
|
elif args.mode=="new":
|
|
songDatabaseList = []
|
|
|
|
if args.art.lower() == "true":
|
|
x = len(songFiles)*0.25
|
|
if x > 60:
|
|
print("ETA "+ str(x/60) + " minutes")
|
|
else:
|
|
print("ETA "+ str(x) + " seconds")
|
|
|
|
for i in songFiles:
|
|
try:
|
|
song = EasyID3("sound/"+i)
|
|
title = song['title'][0]
|
|
artist = song['artist'][0]
|
|
except:
|
|
try:
|
|
song = i.split("_")
|
|
title = song[0]
|
|
artist = song[1].split(".")[0]
|
|
except:
|
|
title = i
|
|
artist = None
|
|
if args.art.lower() == "true" and not(args.apikey == ""):
|
|
try:
|
|
image = ast.literal_eval(requests.post(url="http://ws.audioscrobbler.com/2.0/?method=track.getInfo&api_key="+apikeylastfm+"&artist="+artist+"&track="+title+"&format=json").text)["track"]["album"]["image"][1]["#text"]
|
|
if image == "":
|
|
image = ast.literal_eval(requests.post(url="http://ws.audioscrobbler.com/2.0/?method=track.getInfo&api_key="+apikeylastfm+"&artist="+artist+"&track="+title+"&format=json").text)["track"]["album"]["image"][2]["#text"]
|
|
if image == "":
|
|
image = None
|
|
time.sleep(0.25)
|
|
except:
|
|
image=None
|
|
else:
|
|
image=None
|
|
try:
|
|
length = math.ceil(MP3("sound/"+i).info.length)
|
|
except:
|
|
length = 0
|
|
if len(songFiles) != 1:
|
|
index = (songFiles.index(i))%4
|
|
print("\r" + str(loading[index] + str(math.floor((songFiles.index(i)/(len(songFiles)-1))*100))+ "%"), end='', flush=True)
|
|
# each "song" is stored as a dictionary containing the below stuff, and each dictionary is put into a list
|
|
songDatabaseList.append({"file":i,"title":title,"artist":artist,"art":image,"length":length})
|
|
|
|
with open('songDatabase.json', 'w') as handle:
|
|
json.dump(songDatabaseList, handle)
|