95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
import os
|
||
os.environ['NCATBOT_CONFIG_PATH'] = 'ncatbot.yaml'
|
||
import toml
|
||
from ncatbot.core.event import PrivateMessageEvent, GroupMessageEvent, NoticeEvent, RequestEvent
|
||
from ncatbot.utils import ncatbot_config
|
||
from ncatbot.core import BotClient
|
||
from model.logger import setup_logger
|
||
from control.group import group
|
||
from control.private import private
|
||
from control.request import request
|
||
from control.notice import notice
|
||
|
||
logger = setup_logger()
|
||
|
||
with open("./config.toml", "r", encoding="utf-8") as f:
|
||
config = toml.load(f)
|
||
bt_uin = config.get("bot_qq")
|
||
root = config.get("root_qq")
|
||
ws_uri = config.get("ws_uri")
|
||
webui_uri = config.get("webui_uri")
|
||
webui_token = config.get("webui_token")
|
||
ws_token = config.get("ws_token")
|
||
ws_listen_ip = config.get("ws_listen_ip")
|
||
enable_webui_interaction = config.get("enable_webui_interaction")
|
||
|
||
ncatbot_config.set_bot_uin(bt_uin)
|
||
ncatbot_config.set_root(root)
|
||
ncatbot_config.set_webui_uri(webui_uri)
|
||
ncatbot_config.set_webui_token(webui_token)
|
||
ncatbot_config.set_ws_uri(ws_uri)
|
||
ncatbot_config.set_ws_token(ws_token)
|
||
ncatbot_config.set_ws_listen_ip(ws_listen_ip)
|
||
ncatbot_config.enable_webui_interaction = enable_webui_interaction
|
||
|
||
bot = BotClient()
|
||
|
||
@bot.on_group_message()
|
||
async def on_group_message(msg: GroupMessageEvent):
|
||
logger.info(f"收到消息：{msg.raw_message}，来自{msg.group_id}群聊{msg.user_id}用户")
|
||
if msg.user_id == 2854196310: # qq管家，防止刷屏。bot大战请看
|
||
pass
|
||
else:
|
||
ctrl = group(msg)
|
||
return_message = ctrl.main()
|
||
if return_message is None:
|
||
return
|
||
else:
|
||
# logger.info(f"返回消息：{return_message}")
|
||
await msg.reply(text=return_message)
|
||
|
||
@bot.on_private_message()
|
||
async def on_private_message(msg: PrivateMessageEvent):
|
||
logger.info(f"收到消息：{msg.raw_message}，来自{msg.user_id}用户")
|
||
if msg.user_id == 2854196310: # qq管家，防止刷屏。
|
||
pass
|
||
else:
|
||
ctrl = private(msg)
|
||
return_message = ctrl.main()
|
||
if return_message is None:
|
||
return
|
||
else:
|
||
# logger.info(f"返回消息：{return_message}")
|
||
await msg.reply(text=return_message)
|
||
|
||
|
||
@bot.on_request(filter='friend')
|
||
async def on_request_event(msg: RequestEvent):
|
||
logger.info(f"收到request事件：{msg.request_type}，来自{msg.group_id}群聊，{msg.user_id}用户，验证消息：{msg.comment}")
|
||
logger.info(f"收到request请求，来自{msg.self_id}")
|
||
ctrl = request(msg)
|
||
accept_friend_application = ctrl.main()
|
||
if accept_friend_application is True:
|
||
await msg.reply(True)
|
||
# logger.info("请求已通过")
|
||
else:
|
||
await msg.reply(False)
|
||
# logger.info("请求被拒绝")
|
||
|
||
|
||
|
||
@bot.on_notice()
|
||
async def on_notice_event(msg: NoticeEvent):
|
||
logger.info(f"收到notice事件：{msg['notice_type']}，来自{msg['user_id']}用户")
|
||
ctrl = notice(msg)
|
||
return_message = ctrl.main()
|
||
if return_message is None:
|
||
return
|
||
else:
|
||
# logger.info(f"返回消息：{return_message}")
|
||
await bot.api.post_group_msg(group_id=msg["group_id"], text=return_message)
|
||
|
||
|
||
bot.run_frontend()
|
||
|