From c90988690207e791283c4b0e4bd94ebd815f3a24 Mon Sep 17 00:00:00 2001 From: 283375 Date: Sun, 17 Sep 2023 00:27:20 +0800 Subject: [PATCH] feat(db): `get_songs_by_pack_id` --- src/arcaea_offline/database.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/arcaea_offline/database.py b/src/arcaea_offline/database.py index 1c2e172..a5eca23 100644 --- a/src/arcaea_offline/database.py +++ b/src/arcaea_offline/database.py @@ -130,6 +130,12 @@ class Database(metaclass=Singleton): results = list(session.scalars(stmt)) return results + def get_songs_by_pack_id(self, pack_id: str): + stmt = select(Song).where(Song.set == pack_id) + with self.sessionmaker() as session: + results = list(session.scalars(stmt)) + return results + def get_song(self, song_id: str): stmt = select(Song).where(Song.id == song_id) with self.sessionmaker() as session: