From 72ff4e54ac161cf01981f24d144c4fbd01500480 Mon Sep 17 00:00:00 2001 From: SlyAimer <2289782085@qq.com> Date: Wed, 22 Jan 2025 18:29:44 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E9=87=8D=E5=A4=8D=E6=B3=A8?= =?UTF-8?q?=E5=86=8C=E7=AE=A1=E7=90=86=E5=91=98=E6=8F=90=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/my_sqlite/models/chat.py | 6 ++++-- src/qq_plugins/chat.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/my_sqlite/models/chat.py b/src/my_sqlite/models/chat.py index a58c890..f98327c 100644 --- a/src/my_sqlite/models/chat.py +++ b/src/my_sqlite/models/chat.py @@ -108,7 +108,7 @@ class GroupChatRole(Model): table_description = "群聊ai对应的角色设定" @classmethod - async def blind_admin(cls ,admin_list: str | None,group_id: str | None) -> str | None: + async def blind_admin(cls ,admin_list: list | None,group_id: str | None) -> str | None: """ 初次绑定ai :param admin_list: @@ -118,7 +118,9 @@ class GroupChatRole(Model): # 创建初始化的角色设定 existing_record = await cls.filter(group_id=group_id).first() if existing_record: - existing_record.admin_id.append([admin_list]) + if admin_list[0] in existing_record.admin_id: + return "您已经是管理员，请勿重复注册" + existing_record.admin_id.append(admin_list[0]) await existing_record.save() else: role_name = '初始模型' diff --git a/src/qq_plugins/chat.py b/src/qq_plugins/chat.py index 7072ec4..42a407a 100644 --- a/src/qq_plugins/chat.py +++ b/src/qq_plugins/chat.py @@ -23,7 +23,7 @@ async def handle_function(message: MessageEvent): member_openid, group_openid = message.author.id, message.group_openid password = message.get_plaintext().replace("/管理员注册", "").strip() if password == admin_password: - result = await GroupChatRole.blind_admin(member_openid, group_openid) + result = await GroupChatRole.blind_admin([member_openid], group_openid) await t1.finish(result) else: await t1.finish("管理员注册密码错误。")