Added all app files

This commit is contained in:
dylan-fournier 2024-05-27 12:26:47 -04:00
parent ce785b5b50
commit 889137712a
13 changed files with 810 additions and 0 deletions

View file

@ -0,0 +1,78 @@
import os
from mutagen.easyid3 import EasyID3
from mutagen.mp3 import MP3
import requests, ast, time, math, argparse, json
loading = ["-","\\","|","/"]
apikeylastfm="YourLastfmKeyHere"
songFiles = os.listdir(r'./sound')
parser=argparse.ArgumentParser(description="Options for the generation of the song database")
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()
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":
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)