From 7c94ca20744da46d0d4793b5e227aa47b5ef6dbd Mon Sep 17 00:00:00 2001 From: Dniel97 Date: Wed, 15 Apr 2026 17:26:14 +0200 Subject: [PATCH] idac: fix vs info battle enum --- titles/idac/schema/vs_info_db.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/titles/idac/schema/vs_info_db.py b/titles/idac/schema/vs_info_db.py index 239a27a..d196f87 100644 --- a/titles/idac/schema/vs_info_db.py +++ b/titles/idac/schema/vs_info_db.py @@ -30,10 +30,10 @@ class IDACVsInfoDb(BaseData): return self.mapper.vs_infos_to_dto(vs_infos_db) async def get_vs_info_by_mode( - self, aime_id: int, battle_mode: int + self, aime_id: int, battle_mode: VSInfoBattleModeEnum ) -> Optional[VsInfoDto]: sql = select(VsInfoDb).where( - and_(VsInfoDb.user == aime_id, VsInfoDb.battle_mode == battle_mode) + and_(VsInfoDb.user == aime_id, VsInfoDb.battle_mode == battle_mode.value) ) result = await self.execute(sql) vs_info_db = result.scalars().first() @@ -43,12 +43,12 @@ class IDACVsInfoDb(BaseData): return self.mapper.vs_info_to_dto(vs_info_db) async def get_vs_course_infos_by_mode( - self, aime_id: int, battle_mode: int + self, aime_id: int, battle_mode: VSInfoBattleModeEnum ) -> Optional[list[VsCourseInfoDto]]: sql = select(VsCourseInfoDb).where( and_( VsCourseInfoDb.user == aime_id, - VsCourseInfoDb.battle_mode == battle_mode, + VsCourseInfoDb.battle_mode == battle_mode.value, ) ) result = await self.execute(sql) @@ -75,7 +75,10 @@ class IDACVsInfoDb(BaseData): return result.lastrowid async def put_vs_course_info( - self, aime_id: int, battle_mode: VSInfoBattleModeEnum, course_info_data: VsCourseInfoDto + self, + aime_id: int, + battle_mode: VSInfoBattleModeEnum, + course_info_data: VsCourseInfoDto, ) -> Optional[int]: course_info_data_db = course_info_data.model_dump_db() course_info_data_db["user"] = aime_id