mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 19:25:34 +00:00
120 lines
3.2 KiB
Python
120 lines
3.2 KiB
Python
"""
|
|
Contains the SQLiteTrackMethods class which contains methods for
|
|
interacting with the tracks table.
|
|
"""
|
|
|
|
from collections import OrderedDict
|
|
from sqlite3 import Cursor
|
|
|
|
from app.db.sqlite.utils import tuple_to_track, tuples_to_tracks
|
|
|
|
from .utils import SQLiteManager
|
|
|
|
|
|
class SQLiteTrackMethods:
|
|
"""
|
|
This class contains all methods for interacting with the tracks table.
|
|
"""
|
|
|
|
@classmethod
|
|
def insert_one_track(cls, track: dict, cur: Cursor):
|
|
"""
|
|
Inserts a single track into the database.
|
|
"""
|
|
sql = """INSERT INTO tracks(
|
|
album,
|
|
albumartist,
|
|
albumhash,
|
|
artist,
|
|
bitrate,
|
|
copyright,
|
|
date,
|
|
disc,
|
|
duration,
|
|
filepath,
|
|
folder,
|
|
genre,
|
|
title,
|
|
track,
|
|
trackhash
|
|
) VALUES(:album, :albumartist, :albumhash, :artist, :bitrate, :copyright,
|
|
:date, :disc, :duration, :filepath, :folder, :genre, :title, :track, :trackhash)
|
|
"""
|
|
|
|
track = OrderedDict(sorted(track.items()))
|
|
cur.execute(sql, track)
|
|
|
|
@classmethod
|
|
def insert_many_tracks(cls, tracks: list[dict]):
|
|
"""
|
|
Inserts a list of tracks into the database.
|
|
"""
|
|
with SQLiteManager() as cur:
|
|
for track in tracks:
|
|
cls.insert_one_track(track, cur)
|
|
|
|
@staticmethod
|
|
def get_all_tracks():
|
|
"""
|
|
Get all tracks from the database and return a generator of Track objects
|
|
or an empty list.
|
|
"""
|
|
with SQLiteManager() as cur:
|
|
cur.execute("SELECT * FROM tracks")
|
|
rows = cur.fetchall()
|
|
|
|
if rows is not None:
|
|
return tuples_to_tracks(rows)
|
|
|
|
return []
|
|
|
|
@staticmethod
|
|
def get_track_by_trackhash(trackhash: str):
|
|
"""
|
|
Gets a track using its trackhash. Returns a Track object or None.
|
|
"""
|
|
with SQLiteManager() as cur:
|
|
cur.execute("SELECT * FROM tracks WHERE trackhash=?", (trackhash,))
|
|
row = cur.fetchone()
|
|
|
|
if row is not None:
|
|
return tuple_to_track(row)
|
|
|
|
return None
|
|
|
|
@staticmethod
|
|
def get_tracks_by_trackhashes(hashes: list[str]):
|
|
"""
|
|
Gets all tracks in a list of trackhashes.
|
|
Returns a generator of Track objects or an empty list.
|
|
"""
|
|
|
|
sql = "SELECT * FROM tracks WHERE trackhash IN ({})".format(
|
|
",".join("?" * len(hashes))
|
|
)
|
|
|
|
with SQLiteManager() as cur:
|
|
cur.execute(sql, hashes)
|
|
rows = cur.fetchall()
|
|
|
|
if rows is not None:
|
|
return tuples_to_tracks(rows)
|
|
|
|
return []
|
|
|
|
@staticmethod
|
|
def remove_track_by_filepath(filepath: str):
|
|
"""
|
|
Removes a track from the database using its filepath.
|
|
"""
|
|
with SQLiteManager() as cur:
|
|
cur.execute("DELETE FROM tracks WHERE filepath=?", (filepath,))
|
|
|
|
@staticmethod
|
|
def remove_tracks_by_folders(folders: set[str]):
|
|
sql = "DELETE FROM tracks WHERE folder = ?"
|
|
|
|
with SQLiteManager() as cur:
|
|
for folder in folders:
|
|
cur.execute(sql, (folder,))
|