Implement get_tech_count (it was stubbed out)

This commit is contained in:
Kumubou 2024-04-17 00:45:05 +00:00
parent 39fc5772eb
commit 1346bab911
1 changed files with 7 additions and 1 deletions

View File

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