mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
5487dad27b
+ Remove folder store + Reduce fuzzy search score cutoff from 90% to 75% + use inheritance to init Artist class + misc
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
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))
|
|
return cur.lastrowid
|
|
|
|
@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_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 []
|