first_commit .gitignore

plugins/ai/__init__.py

@@ -0,0 +1,61 @@
+from .data_source import get_qqbot_chat_result, hello, no_result
+from services.log import logger
+from nonebot import on_message
+from nonebot.rule import to_me
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, PrivateMessageEvent
+from util.utils import get_message_text, get_message_imgs
+from models.friend_user import FriendUser
+from models.group_member_info import GroupInfoUser
+
+__plugin_name__ = 'AI'
+
+
+ai = on_message(rule=to_me(), priority=8)
+
+
+@ai.handle()
+async def _(bot: Bot, event: PrivateMessageEvent, state: T_State):
+    msg = get_message_text(event.json())
+    imgs = get_message_imgs(event.json())
+    if str(event.get_message()).find('CQ:xml') != -1:
+        return
+    # 打招呼
+    if not msg and not imgs:
+        await ai.finish(hello())
+    img = imgs[0] if imgs else ''
+    nickname = await FriendUser.get_friend_nickname(event.user_id)
+    if not nickname:
+        nickname = await FriendUser.get_user_name(event.user_id)
+        if not nickname:
+            nickname = "你"
+    result = await get_qqbot_chat_result(msg, img, event.user_id, nickname)
+    logger.info(f"USER {event.user_id} 问题：{msg}\n回答：{result}")
+    if result:
+        await ai.finish(result)
+    else:
+        await ai.finish(no_result())
+
+
+@ai.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    # if await GroupRemind.get_status(event.group_id, 'ai'):
+    msg = get_message_text(event.json())
+    imgs = get_message_imgs(event.json())
+    # 打招呼
+    if not msg and not imgs:
+        await ai.finish(hello())
+    img = imgs[0] if imgs else ''
+    nickname = await GroupInfoUser.get_group_member_nickname(event.user_id, event.group_id)
+    if not nickname:
+        try:
+            nickname = (await GroupInfoUser.select_member_info(event.user_id, event.group_id)).user_name
+        except AttributeError:
+            nickname = "你"
+    result = await get_qqbot_chat_result(msg, img, event.user_id, nickname)
+    logger.info(f"问题：{msg}\n回答：{result}")
+    if result:
+        await ai.finish(result)
+    else:
+        await ai.finish(no_result())
+