113 lines
5 KiB
Python
113 lines
5 KiB
Python
import os
|
|
from mutagen.easyid3 import EasyID3
|
|
from mutagen.mp3 import MP3
|
|
import sqlite3 as sql
|
|
import requests, ast, time, math, argparse
|
|
|
|
loading = ["-","\\","|","/"]
|
|
|
|
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")
|
|
parser.add_argument('-d','--directory',help="Directory of the song files", default="./sound/")
|
|
args = parser.parse_args()
|
|
apikeylastfm = args.apikey
|
|
if args.directory[-1] == "/" or args.directory[-1] == "\\":
|
|
soundLocation = args.directory
|
|
elif "/" in args.directory:
|
|
soundLocation = args.directory + "/"
|
|
else:
|
|
soundLocation = args.directory + "\\"
|
|
# if you want to set the api key/sound directory permenantly for your setup just uncomment the next line
|
|
# apikeylastfm = "KeyHere"
|
|
# soundLocation = "directoryHere"
|
|
songFiles = os.listdir(soundLocation)
|
|
fileOfDB = sql.connect("songDatabase.db")
|
|
songDatabase = fileOfDB.cursor()
|
|
# setting song directory
|
|
songDatabase.execute("CREATE TABLE IF NOT EXISTS meta (id TEXT PRIMARY KEY, data TEXT);")
|
|
try:
|
|
songDatabase.execute("INSERT INTO meta (id, data) VALUES (?,?)",("songDirectory",soundLocation))
|
|
except:
|
|
songDatabase.execute("UPDATE meta SET data = ? WHERE id = 'songDirectory'", (soundLocation,))
|
|
if args.mode.lower() == "update":
|
|
#Create if not exists
|
|
songDatabase.execute("CREATE TABLE IF NOT EXISTS songs (filename TEXT PRIMARY KEY, title TEXT, artist TEXT, art TEXT, length INTEGER);")
|
|
songDatabase.execute("SELECT filename FROM songs;")
|
|
dBfilelist = songDatabase.fetchall()
|
|
dBfilelistSet = set()
|
|
for i in dBfilelist:
|
|
dBfilelistSet.add(i[0])
|
|
# Delete nonexistant files
|
|
deleteySongs = list(dBfilelistSet - set(songFiles))
|
|
songDatabase.executemany("DELETE FROM songs WHERE filename = ?", [(item,) for item in deleteySongs]) # in this line it turns the list of strings into a list of tuples of strings
|
|
print("Deleted: " + ", ".join(deleteySongs)+ " from database")
|
|
# only include new files in list to be used
|
|
songFiles = list(set(songFiles) - dBfilelistSet)
|
|
print("new songs: " + ", ".join(songFiles))
|
|
elif args.mode.lower()=="new":
|
|
songDatabase.execute("DROP TABLE IF EXISTS songs;")
|
|
songDatabase.execute("CREATE TABLE songs (filename TEXT PRIMARY KEY, title TEXT, artist TEXT, art TEXT, length INTEGER);")
|
|
else:
|
|
raise ValueError("Must be \"new\" or \"update\"")
|
|
|
|
if args.art.lower() == "true" and not(args.apikey == ""):
|
|
x = len(songFiles)*0.25
|
|
if x > 60:
|
|
print("ETA "+ str(x/60) + " minutes")
|
|
else:
|
|
print("ETA "+ str(x) + " seconds")
|
|
|
|
for i in songFiles:
|
|
if i[-4:].lower() != ".mp3":
|
|
# skip any non-mp3's (like directories or cover art)
|
|
continue
|
|
try:
|
|
# get the metadata
|
|
song = EasyID3(soundLocation+i)
|
|
title = song['title'][0]
|
|
artist = song['artist'][0]
|
|
except:
|
|
if "_" in i:
|
|
# if metadata is missing, try to use file name following title_artist.mp3
|
|
song = i.split("_")
|
|
title = song[0]
|
|
artist = song[1].split(".")[0]
|
|
elif "-" in i:
|
|
# if there's no underscore, try artist - title.mp3
|
|
song = i.split("-")
|
|
title = song[1].split(".")[0]
|
|
artist = song[0]
|
|
title = title.strip()
|
|
artist = artist.strip()
|
|
else:
|
|
#if the file is not formatted with an underscore, the title is the file name
|
|
title = i
|
|
artist = None
|
|
if args.art.lower() == "true" and not(args.apikey == ""):
|
|
try:
|
|
# get the images from last fm, try 2 different sizes
|
|
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 = 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 = None
|
|
time.sleep(0.25)
|
|
except:
|
|
image=None
|
|
else:
|
|
image=None
|
|
try:
|
|
length = math.ceil(MP3(soundLocation+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/JSON entry following the format seen in the readME
|
|
songDatabase.execute(f"INSERT INTO songs (filename, title, artist, art, length) VALUES (?,?,?,?,?)",(i,title,artist,image,length))
|
|
|
|
|
|
|
|
fileOfDB.commit()
|