first commit

swingmusic/store/albums.py

@@ -0,0 +1,122 @@
+from collections.abc import Iterable
+
+from swingmusic.lib.tagger import create_albums
+from swingmusic.models import Album, Track
+from swingmusic.store.artists import ArtistStore
+from swingmusic.store.tracks import TrackStore
+from swingmusic.utils.auth import get_current_userid
+
+ALBUM_LOAD_KEY = ""
+
+
+class AlbumMapEntry:
+    def __init__(self, album: Album, trackhashes: set[str]) -> None:
+        self.album = album
+        self.trackhashes = trackhashes
+
+    @property
+    def basetitle(self):
+        return self.album.base_title
+
+    def increment_playcount(self, duration: int, timestamp: int, playcount: int = 1):
+        self.album.lastplayed = timestamp
+        self.album.playduration += duration
+        self.album.playcount += playcount
+
+    def toggle_favorite_user(self, userid: int | None = None):
+        if userid is None:
+            userid = get_current_userid()
+
+        self.album.toggle_favorite_user(userid)
+
+    def set_color(self, color: str):
+        self.album.color = color
+
+
+class AlbumStore:
+    albummap: dict[str, AlbumMapEntry] = {}
+
+    @classmethod
+    def load_albums(cls, instance_key: str):
+        """
+        Loads all albums from the database into the store.
+        """
+        global ALBUM_LOAD_KEY
+        ALBUM_LOAD_KEY = instance_key
+
+        print("Loading albums... ", end="")
+
+        cls.albummap = {
+            album.albumhash: AlbumMapEntry(album=album, trackhashes=trackhashes)
+            for album, trackhashes in create_albums()
+        }
+        print("Done!")
+
+    @classmethod
+    def index_new_album(cls, album: Album, trackhashes: set[str]):
+        cls.albummap[album.albumhash] = AlbumMapEntry(
+            album=album, trackhashes=trackhashes
+        )
+
+    @classmethod
+    def get_flat_list(cls):
+        """
+        Returns a flat list of all albums.
+        """
+        return [a.album for a in cls.albummap.values()]
+
+    @classmethod
+    def get_album_by_hash(cls, albumhash: str) -> Album | None:
+        """
+        Returns an album by its hash.
+        """
+        entry = cls.albummap.get(albumhash)
+        if entry is not None:
+            return entry.album
+
+    @classmethod
+    def get_albums_by_hashes(cls, albumhashes: Iterable[str]) -> list[Album]:
+        """
+        Returns albums by their hashes.
+        """
+        albums = []
+        for albumhash in albumhashes:
+            entry = cls.albummap.get(albumhash)
+            if entry is not None:
+                albums.append(entry.album)
+
+        return albums
+
+    @classmethod
+    def get_albums_by_artisthash(cls, hash: str):
+        """
+        Returns all albums by the given artist hash.
+        """
+        artist = ArtistStore.artistmap.get(hash)
+
+        if not artist:
+            return []
+
+        return [cls.albummap[albumhash].album for albumhash in artist.albumhashes]
+
+    @classmethod
+    def get_albums_by_artisthashes(cls, hashes: Iterable[str]):
+        """
+        Returns all albums by the given artist hashes.
+        """
+        albums = []
+        for hash in hashes:
+            albums.extend(cls.get_albums_by_artisthash(hash))
+
+        return albums
+
+    @classmethod
+    def get_album_tracks(cls, albumhash: str) -> list[Track]:
+        """
+        Returns all tracks for the given album hash.
+        """
+        album = cls.albummap.get(albumhash)
+        if not album:
+            return []
+
+        return TrackStore.get_tracks_by_trackhashes(album.trackhashes)