283375/arcaea-offline
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline.git synced 2025-04-09 17:40:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(db): minor fixes

Browse Source
This commit is contained in:
283375 2023-09-17 00:26:56 +08:00
parent 4899e0383f
commit 589826004b
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/arcaea_offline/database.py
View File

@ -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):