Genshin-bots/gsuid_core
1
0
Fork 0
mirror of https://github.com/Genshin-bots/gsuid_core.git synced 2025-05-12 06:55:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

🎨 在插入数据中如果有已存在数据则更新而不是新建

Browse Source
This commit is contained in:
‘KimigaiiWuyi’ 2023-10-20 14:41:37 +08:00
parent 067fde116c
commit 679de1fc5e
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
gsuid_core/utils/database/base_models.py
View File

@ -159,8 +159,11 @@ class BaseModel(BaseBotIDModel):
async def insert_data( async def insert_data(
cls, session: AsyncSession, user_id: str, bot_id: str, **data cls, session: AsyncSession, user_id: str, bot_id: str, **data
) -> int: ) -> int:
session.add(cls(user_id=user_id, bot_id=bot_id, **data)) if await cls.data_exist(user_id=user_id, bot_id=bot_id):
await session.commit() await cls.update_data(user_id, bot_id, **data)
else:
session.add(cls(user_id=user_id, bot_id=bot_id, **data))
await session.commit()
return 0 return 0
@classmethod @classmethod