From ca9576160f737f9b648d1eaaf60bdcbd34076d1a Mon Sep 17 00:00:00 2001 From: 283375 Date: Tue, 29 Aug 2023 01:32:26 +0800 Subject: [PATCH] fix(db): convert results to list before closing session --- src/arcaea_offline/database.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/arcaea_offline/database.py b/src/arcaea_offline/database.py index 10cca5a..e6b213a 100644 --- a/src/arcaea_offline/database.py +++ b/src/arcaea_offline/database.py @@ -103,8 +103,8 @@ class Database(metaclass=Singleton): def get_packs(self): stmt = select(Pack) with self.sessionmaker() as session: - results = session.scalars(stmt) - return list(results) + results = list(session.scalars(stmt)) + return results def get_pack_by_id(self, pack_id: str): stmt = select(Pack).where(Pack.id == pack_id) @@ -119,14 +119,14 @@ class Database(metaclass=Singleton): def get_charts_by_pack_id(self, pack_id: str): stmt = select(Chart).where(Chart.set == pack_id) with self.sessionmaker() as session: - results = session.scalars(stmt) - return list(results) + results = list(session.scalars(stmt)) + return results def get_charts_by_song_id(self, song_id: str): stmt = select(Chart).where(Chart.song_id == song_id) with self.sessionmaker() as session: - results = session.scalars(stmt) - return list(results) + results = list(session.scalars(stmt)) + return results def get_chart(self, song_id: str, rating_class: int): stmt = select(Chart).where(