mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
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, Message
|
|
from utils.utils import get_message_text, get_message_imgs
|
|
from models.friend_user import FriendUser
|
|
from models.group_member_info import GroupInfoUser
|
|
|
|
__plugin_name__ = 'AI [Hidden]'
|
|
|
|
|
|
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) or msg in ['你好啊', '你好', '在吗', '在不在', '您好', '您好啊', '你好', '在']:
|
|
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(Message(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(Message(result))
|
|
else:
|
|
await ai.finish(no_result())
|
|
|