diff --git a/titles/ongeki/base.py b/titles/ongeki/base.py index ca5a38c..57a7363 100644 --- a/titles/ongeki/base.py +++ b/titles/ongeki/base.py @@ -426,9 +426,10 @@ class OngekiBase: userTechCountList = [] for tc in utcl: - tc.pop("id") - tc.pop("user") - userTechCountList.append(tc) + tmp = tc._asdict() + tmp.pop("id") + tmp.pop("user") + userTechCountList.append(tmp) return { "userId": data["userId"], diff --git a/titles/ongeki/schema/score.py b/titles/ongeki/schema/score.py index 4770725..dc3a91d 100644 --- a/titles/ongeki/schema/score.py +++ b/titles/ongeki/schema/score.py @@ -129,7 +129,13 @@ tech_count = Table( class OngekiScoreData(BaseData): async def get_tech_count(self, aime_id: int) -> Optional[List[Dict]]: - return [] + sql = select(tech_count).where(tech_count.c.user == aime_id) + + result = self.execute(sql) + + if result is None: + return None + return result.fetchall() async def put_tech_count(self, aime_id: int, tech_count_data: Dict) -> Optional[int]: tech_count_data["user"] = aime_id