handle watching ~/ dir

+ fix bug that caused duplicate album colors in db

app/db/sqlite/tracks.py

@@ -130,15 +130,9 @@ class SQLiteTrackMethods:
             cur.execute("DELETE FROM tracks WHERE filepath=?", (filepath,))
 
     @staticmethod
-    def track_exists(filepath: str):
-        """
-        Checks if a track exists in the database using its filepath.
-        """
+    def remove_tracks_by_folders(folders: list[str]):
+        sql = "DELETE FROM tracks WHERE folder = ?"
+
         with SQLiteManager() as cur:
-            cur.execute("SELECT * FROM tracks WHERE filepath=?", (filepath,))
-            row = cur.fetchone()
-
-            if row is not None:
-                return True
-
-            return False
+            for folder in folders:
+                cur.execute(sql, (folder,))

app/db/store.py

@@ -88,6 +88,17 @@ class Store:
                 cls.tracks.remove(track)
                 break
 
+    @classmethod
+    def remove_tracks_by_dir_except(cls, dirs: list[str]):
+        """Removes all tracks not in the root directories."""
+        to_remove = set()
+
+        for track in cls.tracks:
+            if not track.folder.startswith(tuple(dirs)):
+                to_remove.add(track.folder)
+
+        tdb.remove_tracks_by_folders(to_remove)
+
     @classmethod
     def count_tracks_by_hash(cls, trackhash: str) -> int:
         """
@@ -197,6 +208,8 @@ class Store:
         """
         Creates a list of folders from the tracks in the store.
         """
+        cls.folders.clear()
+
         all_folders = [track.folder for track in cls.tracks]
         all_folders = set(all_folders)
 
@@ -212,6 +225,7 @@ class Store:
 
             cls.folders.append(folder)
 
+
     @classmethod
     def get_folder(cls, path: str):  # type: ignore
         """
@@ -277,6 +291,8 @@ class Store:
         Loads all albums from the database into the store.
         """
 
+        cls.albums = []
+
         albumhashes = set(t.albumhash for t in cls.tracks)
 
         for albumhash in tqdm(albumhashes, desc="Loading albums"):
@@ -291,9 +307,9 @@ class Store:
             albumhash = album[1]
             colors = json.loads(album[2])
 
-            for al in cls.albums:
-                if al.albumhash == albumhash:
-                    al.set_colors(colors)
+            for _al in cls.albums:
+                if _al.albumhash == albumhash:
+                    _al.set_colors(colors)
                     break
 
     @classmethod