Move server code to this repo (#95)

move server code to this repo

app/db/sqlite/albums.py

@@ -0,0 +1,125 @@
+from sqlite3 import Cursor
+
+from app.db import AlbumMethods
+
+from .utils import SQLiteManager, tuple_to_album, tuples_to_albums
+
+
+class SQLiteAlbumMethods(AlbumMethods):
+    @classmethod
+    def insert_one_album(cls, cur: Cursor, albumhash: str, colors: str):
+        """
+        Inserts one album into the database
+        """
+
+        sql = """INSERT INTO albums(
+            albumhash,
+            colors
+            ) VALUES(?,?)
+            """
+
+        cur.execute(sql, (albumhash, colors))
+
+        return cur.lastrowid
+
+    # @classmethod
+    # def insert_many_albums(cls, albums: list[dict]):
+    #     """
+    #     Takes a generator of albums, and inserts them into the database
+
+    #     Parameters
+    #     ----------
+    #     albums : Generator
+    #         Generator
+    #     """
+    #     with SQLiteManager() as cur:
+    #         for album in albums:
+    #             cls.insert_one_album(cur, album["albumhash"], album["colors"])
+
+    @classmethod
+    def get_all_albums(cls):
+        with SQLiteManager() as cur:
+            cur.execute("SELECT * FROM albums")
+            albums = cur.fetchall()
+
+            if albums is not None:
+                return albums
+
+        return []
+
+    # @staticmethod
+    # def get_album_by_id(album_id: int):
+    #     conn = get_sqlite_conn()
+    #     cur = conn.cursor()
+
+    #     cur.execute("SELECT * FROM albums WHERE id=?", (album_id,))
+    #     album = cur.fetchone()
+
+    #     conn.close()
+
+    #     if album is None:
+    #         return None
+
+    #     return tuple_to_album(album)
+
+    @staticmethod
+    def get_album_by_hash(album_hash: str):
+        with SQLiteManager() as cur:
+            cur.execute("SELECT * FROM albums WHERE albumhash=?", (album_hash,))
+            album = cur.fetchone()
+
+            if album is not None:
+                return tuple_to_album(album)
+
+        return None
+
+    @classmethod
+    def get_albums_by_hashes(cls, album_hashes: list):
+        """
+        Gets all the albums with the specified hashes. Returns a generator of albums or an empty list.
+        """
+        with SQLiteManager() as cur:
+            hashes = ",".join("?" * len(album_hashes))
+            cur.execute(
+                f"SELECT * FROM albums WHERE albumhash IN ({hashes})", album_hashes
+            )
+            albums = cur.fetchall()
+
+            if albums is not None:
+                return tuples_to_albums(albums)
+
+        return []
+
+    # @staticmethod
+    # def update_album_colors(album_hash: str, colors: list[str]):
+    #     sql = "UPDATE albums SET colors=? WHERE albumhash=?"
+
+    #     colors_str = json.dumps(colors)
+
+    #     with SQLiteManager() as cur:
+    #         cur.execute(sql, (colors_str, album_hash))
+
+    @staticmethod
+    def get_albums_by_albumartist(albumartist: str):
+        with SQLiteManager() as cur:
+            cur.execute("SELECT * FROM albums WHERE albumartist=?", (albumartist,))
+            albums = cur.fetchall()
+
+            if albums is not None:
+                return tuples_to_albums(albums)
+
+        return []
+
+    @staticmethod
+    def get_all_albums_raw():
+        """
+        Returns all the albums in the database, as a list of tuples.
+        """
+        with SQLiteManager() as cur:
+            cur.execute("SELECT * FROM albums")
+            albums = cur.fetchall()
+
+            if albums is not None:
+                return albums
+
+        return []