From 83fbdc7658ea5bd2a4bea6e43844f81ec8251750 Mon Sep 17 00:00:00 2001
From: Kristy Fournier <124598538+kristy-fournier@users.noreply.github.com>
Date: Mon, 10 Mar 2025 10:32:27 -0400
Subject: [PATCH] Update mode back to working, non mp3's are not added to the
 database

---
 Server/databaseGenerator.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/Server/databaseGenerator.py b/Server/databaseGenerator.py
index 465b96d..5494c3f 100644
--- a/Server/databaseGenerator.py
+++ b/Server/databaseGenerator.py
@@ -22,17 +22,17 @@ else:
 # apikeylastfm = "KeyHere"
 # soundLocation = "directoryHere"
 songFiles = os.listdir(soundLocation)
-if args.mode == "update":
+if args.mode.lower() == "update":
     try:   
         with open('songDatabase.json', 'r') as handle:
             songDatabaseList = json.load(handle)
     except:
         songDatabaseList={"songDirectory":soundLocation,'songData':{}}
-
+    deleteySongs = []
     for i in songDatabaseList["songData"]:
-        deleteySongs = []
         try:
-            songFiles.index(i) != -1
+            if songFiles.index(i) == -1:
+                deleteySongs.append(i)
         except:
             deleteySongs.append(i)
     if deleteySongs:
@@ -41,11 +41,12 @@ if args.mode == "update":
         songDatabaseList["songData"].pop(i)
     for i in songDatabaseList["songData"]:
         songFiles.remove(i)
-    print("new songs: " + str(songFiles))
-elif args.mode=="new":
+    print("new songs: " + ", ".join(songFiles))
+elif args.mode.lower()=="new":
     songDatabaseList={"songDirectory":soundLocation,'songData':{}}
-
-if args.art.lower() == "true":
+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")
@@ -53,6 +54,9 @@ if args.art.lower() == "true":
         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)