swingmusic/app/db/sqlite/albums.py

from sqlite3 import Cursor

from .utils import SQLiteManager, tuples_to_albums


class SQLiteAlbumMethods:
    @classmethod
    def insert_one_album(cls, cur: Cursor, albumhash: str, colors: str):
        """
        Inserts one album into the database
        """

        sql = """INSERT OR REPLACE INTO albums(
            albumhash,
            colors
            ) VALUES(?,?)
            """

        cur.execute(sql, (albumhash, colors))
        lastrowid = cur.lastrowid
        cur.close()

        return lastrowid

    @classmethod
    def get_all_albums(cls):
        with SQLiteManager() as cur:
            cur.execute("SELECT * FROM albums")
            albums = cur.fetchall()
            cur.close()

            if albums is not None:
                return albums

        return []

    @staticmethod
    def get_albums_by_albumartist(albumartist: str):
        with SQLiteManager() as cur:
            cur.execute("SELECT * FROM albums WHERE albumartist=?", (albumartist,))
            albums = cur.fetchall()
            cur.close()

            if albums is not None:
                return tuples_to_albums(albums)

        return []