mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 19:25:34 +00:00
5487dad27b
+ Remove folder store + Reduce fuzzy search score cutoff from 90% to 75% + use inheritance to init Artist class + misc
37 lines
910 B
Python
37 lines
910 B
Python
"""
|
|
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: str | 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():
|
|
"""
|
|
Get all artists from the database and return a generator of Artist objects
|
|
"""
|
|
sql = """SELECT * FROM artists"""
|
|
|
|
with SQLiteManager() as cur:
|
|
cur.execute(sql)
|
|
|
|
for artist in cur.fetchall():
|
|
yield artist
|