add function to validate albums

+ extract colors in watchdogg
+ rename color db files

app/lib/watchdogg.py

@@ -1,6 +1,7 @@
 """
 This library contains the classes and functions related to the watchdog file watcher.
 """
+import json
 import os
 import sqlite3
 import time
@@ -9,10 +10,14 @@ from watchdog.events import PatternMatchingEventHandler
 from watchdog.observers import Observer
 
 from app import settings
+
 from app.db.sqlite.settings import SettingsSQLMethods as sdb
 from app.db.sqlite.tracks import SQLiteManager
 from app.db.sqlite.tracks import SQLiteTrackMethods as db
-from app.lib.taglib import get_tags
+from app.db.sqlite.albumcolors import SQLiteAlbumMethods as aldb
+
+from app.lib.colorlib import process_color
+from app.lib.taglib import extract_thumb, get_tags
 from app.logger import log
 from app.models import Artist, Track
 from app.store.albums import AlbumStore
@@ -123,6 +128,22 @@ class Watcher:
         self.run()
 
 
+def handle_colors(cur: sqlite3.Cursor, albumhash: str):
+    exists = aldb.exists(albumhash, cur)
+
+    if exists:
+        return
+
+    colors = process_color(albumhash, is_album=True)
+
+    if colors is None:
+        return
+
+    aldb.insert_one_album(cur=cur, albumhash=albumhash, colors=json.dumps(colors))
+
+    return colors
+
+
 def add_track(filepath: str) -> None:
     """
     Processes the audio tags for a given file ands add them to the database and store.
@@ -138,14 +159,23 @@ def add_track(filepath: str) -> None:
     if tags is None or tags["bitrate"] == 0 or tags["duration"] == 0:
         return
 
+    colors = None
+
     with SQLiteManager() as cur:
         db.insert_one_track(tags, cur)
+        extracted = extract_thumb(filepath, tags["albumhash"] + ".webp")
+
+        if not extracted:
+            return
+
+        colors = handle_colors(cur, tags["albumhash"])
 
     track = Track(**tags)
     TrackStore.add_track(track)
 
     if not AlbumStore.album_exists(track.albumhash):
         album = AlbumStore.create_album(track)
+        album.set_colors(colors)
         AlbumStore.add_album(album)
 
     artists: list[Artist] = track.artist + track.albumartist  # type: ignore
@@ -154,6 +184,7 @@ def add_track(filepath: str) -> None:
         if not ArtistStore.artist_exists(artist.artisthash):
             ArtistStore.add_artist(Artist(artist.name))
 
+    extract_thumb(filepath, track.image)
 
 def remove_track(filepath: str) -> None:
     """
@@ -277,7 +308,7 @@ class Handler(PatternMatchingEventHandler):
 
         if current_size == previous_size:
             # Wait for a short duration to ensure the file write operation is complete
-            time.sleep(0.5)
+            time.sleep(5)
 
             # Check the file size again
             current_size = os.path.getsize(event.src_path)