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(