beerpsi/artemis
1
0
Fork 0
forked from Hay1tsme/artemis
Code Pull Requests Activity

i am stupid

Browse Source
This commit is contained in:
beerpsi
2024-06-23 12:58:51 +07:00
parent 6f864d1d7a
commit b6e5cc3065
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
core/data/schema/user.py
View File

@ -140,7 +140,7 @@ class UserData(BaseData):
if result: if result:
return result.fetchone() return result.fetchone()
async def acquire_lock_for_game(self, user_id: int, game: str, extra: dict | None = None): async def check_lock_for_game(self, user_id: int, game: str):
sql = game_locks.select( sql = game_locks.select(
(game_locks.c.user == user_id) (game_locks.c.user == user_id)
& (game_locks.c.game == game) & (game_locks.c.game == game)
@ -150,6 +150,14 @@ class UserData(BaseData):
if result: if result:
return result.fetchone() return result.fetchone()
return None
async def acquire_lock_for_game(self, user_id: int, game: str, extra: dict | None = None):
result = await self.check_lock_for_game(user_id, game)
if result is not None:
return result
sql = ( sql = (
insert(game_locks) insert(game_locks)
.values(user=user_id, game=game, extra=extra) .values(user=user_id, game=game, extra=extra)