From 31dad06b40b117ccf80053ce2d3183ac247a3e6d Mon Sep 17 00:00:00 2001
From: ClovertaTheTrilobita <cloverta@petalmail.com>
Date: Sat, 25 Jan 2025 20:15:52 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=A7=81=E8=81=8A=E6=97=A0?=
 =?UTF-8?q?=E6=B3=95=E5=9B=9E=E5=A4=8D=E7=9A=84BUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/qq_plugins/check.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/qq_plugins/check.py b/src/qq_plugins/check.py
index e510412..c085219 100644
--- a/src/qq_plugins/check.py
+++ b/src/qq_plugins/check.py
@@ -23,8 +23,12 @@ menu = ['/今日运势','/图','/点歌','/摸摸头','/群老婆','/今日老
 
 async def check_value_in_menu(message: MessageEvent) -> bool:
     value = message.get_plaintext().strip().split(" ")
+    if hasattr(message, 'group_openid'): # 是否有属性group_openid，即是否为群聊消息
+        group_id = message.group_openid
+    else:
+        group_id = "C2C" # 非群聊消息，存为c2c
     #缓存用户id
-    await UserList.insert_user(message.author.id,message.group_openid)
+    await UserList.insert_user(message.author.id,group_id)
     if value[0] in menu:
         return False
     else:
@@ -35,7 +39,12 @@ check = on_message(rule=to_me() & Rule(check_value_in_menu) ,block=True, priorit
 @check.handle()
 async def handle_function(message: MessageEvent):
 
-    member_openid, group_openid,content = message.author.id, message.group_openid,message.get_plaintext()
+    if hasattr(message, 'group_openid'):
+        group_openid = message.group_openid
+    else:
+        group_openid = "C2C"
+
+    member_openid, content = message.author.id, message.get_plaintext()
     status = await GroupChatRole.is_on(group_openid)
     if status:
         msg = await ai_chat.deepseek_chat(group_openid,content)