Merge pull request 'Ongeki: Fix issue with technical award folder counts not being saved properly' (#128) from Kumubou/artemis:develop into develop

Reviewed-on: Hay1tsme/artemis#128
This commit is contained in:
Midorica 2024-04-18 21:45:58 +00:00
commit 36ab38b1ee
2 changed files with 11 additions and 4 deletions

View File

@ -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"],

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