Skip to content

Commit fff6a4c

Browse files
committed
Workaround type checker errors.
See sqlalchemy/sqlalchemy#9185 (comment)
1 parent 52c2a8b commit fff6a4c

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

npps4/idol/cache.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@
66
from .. import util
77
from ..db import main
88

9+
from typing import cast
10+
911
ENABLE_CACHE = False
1012

1113

@@ -54,5 +56,5 @@ async def store_response(
5456

5557
async def clear(context: session.BasicSchoolIdolContext, user_id: int):
5658
q = sqlalchemy.delete(main.RequestCache).where(main.RequestCache.user_id == user_id)
57-
result = await context.db.main.execute(q)
59+
result = cast(sqlalchemy.CursorResult, await context.db.main.execute(q))
5860
return result.rowcount

npps4/system/live.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from ..db import main
1919
from ..db import live
2020

21-
from typing import Any, Literal, overload
21+
from typing import Any, Literal, cast, overload
2222

2323

2424
async def unlock_normal_live(context: idol.BasicSchoolIdolContext, user: main.User, live_track_id: int):
@@ -369,7 +369,7 @@ async def get_live_in_progress(context: idol.BasicSchoolIdolContext, user: main.
369369

370370
async def clean_live_in_progress(context: idol.BasicSchoolIdolContext, user: main.User):
371371
q = sqlalchemy.delete(main.LiveInProgress).where(main.LiveInProgress.user_id == user.id)
372-
result = await context.db.main.execute(q)
372+
result = cast(sqlalchemy.CursorResult, await context.db.main.execute(q))
373373
return result.rowcount > 0
374374

375375

npps4/system/unit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from ..db import main
2121
from ..db import unit
2222

23-
from typing import Literal, overload
23+
from typing import Literal, cast, overload
2424

2525

2626
@dataclasses.dataclass
@@ -889,7 +889,7 @@ async def detach_unit_removable_skill(context: idol.BasicSchoolIdolContext, unit
889889
main.UnitRemovableSkill.unit_owning_user_id == unit.id,
890890
main.UnitRemovableSkill.unit_removable_skill_id == removable_skill_id,
891891
)
892-
result = await context.db.main.execute(q)
892+
result = cast(sqlalchemy.CursorResult, await context.db.main.execute(q))
893893
return result.rowcount > 0
894894

895895

0 commit comments

Comments
 (0)