first commit

swingmusic/db/libdata.py

@@ -0,0 +1,79 @@
+from typing import Any
+
+from sqlalchemy import JSON, Integer, String, delete, select
+from sqlalchemy.orm import Mapped, mapped_column
+
+from swingmusic.config import UserConfig
+from swingmusic.db import Base
+from swingmusic.db.engine import DbEngine
+from swingmusic.db.utils import track_to_dataclass, tracks_to_dataclasses
+
+
+class TrackTable(Base):
+    __tablename__ = "track"
+
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    album: Mapped[str] = mapped_column(String())
+    albumartists: Mapped[str] = mapped_column(String())
+    albumhash: Mapped[str] = mapped_column(String(), index=True)
+    artists: Mapped[str] = mapped_column(String())
+    bitrate: Mapped[int] = mapped_column(Integer())
+    copyright: Mapped[str | None] = mapped_column(String())
+    date: Mapped[int] = mapped_column(Integer(), nullable=True)
+    disc: Mapped[int] = mapped_column(Integer())
+    duration: Mapped[int] = mapped_column(Integer())
+    filepath: Mapped[str] = mapped_column(String(), index=True, unique=True)
+    folder: Mapped[str] = mapped_column(String(), index=True)
+    genres: Mapped[str | None] = mapped_column(String())
+    last_mod: Mapped[float] = mapped_column(Integer())
+    title: Mapped[str] = mapped_column(String())
+    track: Mapped[int] = mapped_column(Integer())
+    trackhash: Mapped[str] = mapped_column(String(), index=True)
+    lastplayed: Mapped[int] = mapped_column(Integer(), default=0)
+    playcount: Mapped[int] = mapped_column(Integer(), default=0)
+    playduration: Mapped[int] = mapped_column(Integer(), default=0)
+    extra: Mapped[dict[str, Any] | None] = mapped_column(JSON(), default_factory=dict)
+
+    @classmethod
+    def get_all(cls):
+        with DbEngine.manager() as conn:
+            config = UserConfig()
+            result = conn.execute(select(cls).execution_options(yield_per=100))
+
+            for i in result.scalars():
+                d = i.__dict__
+                del d["_sa_instance_state"]
+
+                yield track_to_dataclass(d, config)
+
+    @classmethod
+    def get_tracks_by_filepaths(cls, filepaths: list[str]):
+        with DbEngine.manager() as conn:
+            result = conn.execute(
+                select(TrackTable)
+                .where(TrackTable.filepath.in_(filepaths))
+                .order_by(TrackTable.last_mod)
+            )
+            return tracks_to_dataclasses(result.fetchall())
+
+    @classmethod
+    def get_tracks_in_path(cls, path: str):
+        with DbEngine.manager() as conn:
+            result = conn.execute(
+                select(TrackTable)
+                .where(TrackTable.filepath.contains(path))
+                .order_by(TrackTable.last_mod)
+            )
+
+            clean = []
+            for row in result.fetchall():
+                d = row[0].__dict__
+                del d["_sa_instance_state"]
+                clean.append(d)
+
+            return tracks_to_dataclasses(clean)
+
+    @classmethod
+    def remove_tracks_by_filepaths(cls, filepaths: set[str]):
+        with DbEngine.manager(commit=True) as conn:
+            conn.execute(delete(TrackTable).where(TrackTable.filepath.in_(filepaths)))