From 589826004b2d90368ae388994c50c3e39a200ece Mon Sep 17 00:00:00 2001 From: 283375 Date: Sun, 17 Sep 2023 00:26:56 +0800 Subject: [PATCH] fix(db): minor fixes --- src/arcaea_offline/database.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/arcaea_offline/database.py b/src/arcaea_offline/database.py index 91ad683..1c2e172 100644 --- a/src/arcaea_offline/database.py +++ b/src/arcaea_offline/database.py @@ -155,13 +155,13 @@ class Database(metaclass=Singleton): def get_difficulties_by_song_id(self, song_id: str): stmt = select(Difficulty).where(Difficulty.song_id == song_id) with self.sessionmaker() as session: - results = session.scalars(stmt) + results = list(session.scalars(stmt)) return results def get_difficulties_localized_by_song_id(self, song_id: str): stmt = select(DifficultyLocalized).where(DifficultyLocalized.song_id == song_id) with self.sessionmaker() as session: - results = session.scalars(stmt) + results = list(session.scalars(stmt)) return results def get_difficulty(self, song_id: str, rating_class: int): @@ -194,7 +194,7 @@ class Database(metaclass=Singleton): def get_chart_infos_by_song_id(self, song_id: str): stmt = select(ChartInfo).where(ChartInfo.song_id == song_id) with self.sessionmaker() as session: - results = session.scalars(stmt) + results = list(session.scalars(stmt)) return results def get_chart_info(self, song_id: str, rating_class: int):