swingmusic/app/db/sqlite/artists.py

"""
Contains methods for reading and writing to the sqlite artists database.
"""

import json
from sqlite3 import Cursor

from .utils import SQLiteManager


class SQLiteArtistMethods:
    @staticmethod
    def insert_one_artist(cur: Cursor, artisthash: str, colors: list[str]):
        """
        Inserts a single artist into the database.
        """
        sql = """INSERT OR REPLACE INTO artists(
            artisthash,
            colors
            ) VALUES(?,?)
            """
        colors = json.dumps(colors)
        cur.execute(sql, (artisthash, colors))

    @staticmethod
    def get_all_artists(cur_: Cursor = None):
        """
        Get all artists from the database and return a generator of Artist objects
        """
        sql = """SELECT * FROM artists"""

        if not cur_:
            with SQLiteManager() as cur:
                cur.execute(sql)

                for artist in cur.fetchall():
                    yield artist

                cur.close()

        else:
            cur_.execute(sql)

            for artist in cur_.fetchall():
                yield artist