mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
67 lines
1.6 KiB
Python
67 lines
1.6 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))
|
|
lastrowid = cur.lastrowid
|
|
|
|
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 []
|
|
|
|
@staticmethod
|
|
def exists(albumhash: str, cur: Cursor = None):
|
|
"""
|
|
Checks if an album exists in the database.
|
|
"""
|
|
|
|
sql = "SELECT COUNT(1) FROM albums WHERE albumhash = ?"
|
|
|
|
def _exists(cur: Cursor):
|
|
cur.execute(sql, (albumhash,))
|
|
count = cur.fetchone()[0]
|
|
|
|
return count != 0
|
|
|
|
if cur:
|
|
return _exists(cur)
|
|
|
|
with SQLiteManager() as cur:
|
|
return _exists(cur)
|