From 679de1fc5eca151087e98d5d84280efce81f15a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E2=80=98KimigaiiWuyi=E2=80=99?= <444835641@qq.com> Date: Fri, 20 Oct 2023 14:41:37 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20=E5=9C=A8=E6=8F=92=E5=85=A5?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E4=B8=AD=E5=A6=82=E6=9E=9C=E6=9C=89=E5=B7=B2?= =?UTF-8?q?=E5=AD=98=E5=9C=A8=E6=95=B0=E6=8D=AE=E5=88=99=E6=9B=B4=E6=96=B0?= =?UTF-8?q?=E8=80=8C=E4=B8=8D=E6=98=AF=E6=96=B0=E5=BB=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gsuid_core/utils/database/base_models.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/gsuid_core/utils/database/base_models.py b/gsuid_core/utils/database/base_models.py index f4a3fde..39eb887 100644 --- a/gsuid_core/utils/database/base_models.py +++ b/gsuid_core/utils/database/base_models.py @@ -159,8 +159,11 @@ class BaseModel(BaseBotIDModel): async def insert_data( cls, session: AsyncSession, user_id: str, bot_id: str, **data ) -> int: - session.add(cls(user_id=user_id, bot_id=bot_id, **data)) - await session.commit() + if await cls.data_exist(user_id=user_id, bot_id=bot_id): + 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 @classmethod