From e2eb66daeb62f188738a0ae447e03ea6d2c17297 Mon Sep 17 00:00:00 2001 From: KimigaiiWuyi <444835641@qq.com> Date: Sun, 15 Dec 2024 23:50:21 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=E4=BF=AE=E5=A4=8D=E8=AE=A2?= =?UTF-8?q?=E9=98=85=E6=B6=88=E6=81=AF=E7=9A=84BUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gsuid_core/subscribe.py | 6 +++--- gsuid_core/utils/database/models.py | 2 ++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/gsuid_core/subscribe.py b/gsuid_core/subscribe.py index bb998d9..57d696c 100644 --- a/gsuid_core/subscribe.py +++ b/gsuid_core/subscribe.py @@ -55,9 +55,9 @@ class GsCoreSubscribe: ) if not condi: - await Subscribe.insert_data( - event.user_id, - event.bot_id, + await Subscribe.full_insert_data( + user_id=event.user_id, + bot_id=event.bot_id, group_id=event.group_id, task_name=task_name, bot_self_id=event.bot_self_id, diff --git a/gsuid_core/utils/database/models.py b/gsuid_core/utils/database/models.py index af05e10..5669994 100644 --- a/gsuid_core/utils/database/models.py +++ b/gsuid_core/utils/database/models.py @@ -46,6 +46,8 @@ class Subscribe(BaseModel, table=True): user_id=self.user_id, bot_self_id=self.bot_self_id, user_type=self.user_type, # type: ignore + group_id=self.group_id, + real_bot_id=self.bot_id, ) bot = Bot(BOT, ev) await bot.send_option(