mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
94 lines
3.9 KiB
Python
94 lines
3.9 KiB
Python
import re
|
||
import os
|
||
import yaml
|
||
from nonebot.rule import to_me
|
||
from nonebot.plugin import on_command
|
||
from nonebot.adapters.qq import MessageEvent
|
||
from src.my_sqlite.models.chat import ChatRole, GroupChatRole
|
||
|
||
with open(os.getcwd() + '/src/ai_chat/config/chat_ai.yaml', 'r', encoding='utf-8') as f1:
|
||
chat = yaml.load(f1, Loader=yaml.FullLoader).get('chat_ai')
|
||
admin_password = chat.get('admin_password')
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
t1 = on_command("管理员注册", rule=to_me(), priority=10, block=True)
|
||
@t1.handle()
|
||
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)
|
||
await t1.finish(result)
|
||
else:
|
||
await t1.finish("管理员注册密码错误。")
|
||
|
||
|
||
t3 = on_command("开启ai",aliases={"关闭ai"},rule=to_me(),priority=10,block=True)
|
||
@t3.handle()
|
||
async def handle_function(message: MessageEvent):
|
||
|
||
member_openid, group_openid, content = message.author.id, message.group_openid, message.get_plaintext()
|
||
# 判断是否为管理员
|
||
if not await GroupChatRole.get_admin_list(group_openid, member_openid):
|
||
await t3.finish("您没有权限使用此功能。")
|
||
else:
|
||
status = await GroupChatRole.is_on(group_openid)
|
||
if not status and content == "/开启ai":
|
||
await GroupChatRole.ai_switch(group_openid)
|
||
await t3.finish("成功开启语言模型对话功能。一起来聊天吧~")
|
||
elif status and content == "/开启ai":
|
||
await t3.finish("当前群已开启ai聊天。请勿重复开启")
|
||
elif status and content == "/关闭ai":
|
||
await GroupChatRole.ai_switch(group_openid)
|
||
await t3.finish("成功关闭语言模型对话功能。")
|
||
elif not status and content == "/关闭ai":
|
||
await t3.finish("当前群已关闭ai聊天,请勿重复关闭。")
|
||
|
||
|
||
tt = on_command("角色列表", rule=to_me(), priority=10, block=True)
|
||
@tt.handle()
|
||
async def handle_function(message: MessageEvent):
|
||
|
||
member_openid, group_openid = message.author.id, message.group_openid
|
||
if not await GroupChatRole.get_admin_list(group_openid, member_openid):
|
||
await tt.finish("您没有权限使用此功能。")
|
||
role_setting = await ChatRole.get_role_list()
|
||
if (role_setting is None) | (not role_setting):
|
||
await tt.finish("角色设定库为空，请添加角色以及设定")
|
||
else:
|
||
await tt.finish(str(role_setting))
|
||
|
||
|
||
t2 = on_command("添加人设",aliases={"更新人设","删除人设","切换人设"},rule=to_me(),priority=10,block=True)
|
||
@t2.handle()
|
||
async def handle_function(message: MessageEvent):
|
||
|
||
member_openid, group_openid = message.author.id, message.group_openid
|
||
if not await GroupChatRole.get_admin_list(group_openid, member_openid):
|
||
await t2.finish("您没有权限使用此功能。")
|
||
value = message.get_plaintext().strip().split(" ")
|
||
action, role_name = value[0], value[1]
|
||
if len(value[1])>10:
|
||
await t2.finish("角色名称过长，请重新输入")
|
||
if action == "/删除人设":
|
||
result = await ChatRole.delete_role(role_name)
|
||
await t2.finish(result)
|
||
if action == "/切换人设":
|
||
result = await GroupChatRole.set_chat_role(group_openid,role_name)
|
||
await t2.finish(result)
|
||
if len(value) < 3:
|
||
await t2.finish("请输入角色名称和设定，格式为：命令 角色名称 角色设定")
|
||
role_setting = re.sub(r'[\n\\n\s"‘’]', '', ''.join(value[2:]))
|
||
if action == "/添加人设":
|
||
result = await ChatRole.insert_role_setting(role_name, role_setting)
|
||
await t2.finish(result)
|
||
if action == "/更新人设":
|
||
result = await ChatRole.update_role_setting(role_name, role_setting)
|
||
await t2.finish(result)
|
||
|