xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-17 09:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(qqbot): 适配配置管理器

Browse Source
This commit is contained in:
RockChinQ
2023-11-26 23:04:14 +08:00
parent db2e366014
commit 549a7eff7f
2 changed files with 55 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/qqbot/adapter.py
+1
View File
@@ -5,6 +5,7 @@ import mirai
class MessageSourceAdapter: class MessageSourceAdapter:
bot_account_id: int
def __init__(self, config: dict): def __init__(self, config: dict):
pass pass
pkg/qqbot/manager.py
+54 -52
View File
@@ -2,7 +2,7 @@ import json
import os import os
import logging import logging
from mirai import At, GroupMessage, MessageEvent, Mirai, StrangerMessage, WebSocketAdapter, HTTPAdapter, \ from mirai import At, GroupMessage, MessageEvent, StrangerMessage, \
FriendMessage, Image, MessageChain, Plain FriendMessage, Image, MessageChain, Plain
import func_timeout import func_timeout
@@ -19,16 +19,16 @@ from ..qqbot import adapter as msadapter
# 检查消息是否符合泛响应匹配机制 # 检查消息是否符合泛响应匹配机制
def check_response_rule(group_id:int, text: str): def check_response_rule(group_id:int, text: str):
config = context.get_config() config = context.get_config_manager().data
rules = config.response_rules rules = config['response_rules']
# 检查是否有特定规则 # 检查是否有特定规则
if 'prefix' not in config.response_rules: if 'prefix' not in config['response_rules']:
if str(group_id) in config.response_rules: if str(group_id) in config['response_rules']:
rules = config.response_rules[str(group_id)] rules = config['response_rules'][str(group_id)]
else: else:
rules = config.response_rules['default'] rules = config['response_rules']['default']
# 检查前缀匹配 # 检查前缀匹配
if 'prefix' in rules: if 'prefix' in rules:
@@ -48,16 +48,16 @@ def check_response_rule(group_id:int, text: str):
def response_at(group_id: int): def response_at(group_id: int):
config = context.get_config() config = context.get_config_manager().data
use_response_rule = config.response_rules use_response_rule = config['response_rules']
# 检查是否有特定规则 # 检查是否有特定规则
if 'prefix' not in config.response_rules: if 'prefix' not in config['response_rules']:
if str(group_id) in config.response_rules: if str(group_id) in config['response_rules']:
use_response_rule = config.response_rules[str(group_id)] use_response_rule = config['response_rules'][str(group_id)]
else: else:
use_response_rule = config.response_rules['default'] use_response_rule = config['response_rules']['default']
if 'at' not in use_response_rule: if 'at' not in use_response_rule:
return True return True
@@ -66,16 +66,16 @@ def response_at(group_id: int):
def random_responding(group_id): def random_responding(group_id):
config = context.get_config() config = context.get_config_manager().data
use_response_rule = config.response_rules use_response_rule = config['response_rules']
# 检查是否有特定规则 # 检查是否有特定规则
if 'prefix' not in config.response_rules: if 'prefix' not in config['response_rules']:
if str(group_id) in config.response_rules: if str(group_id) in config['response_rules']:
use_response_rule = config.response_rules[str(group_id)] use_response_rule = config['response_rules'][str(group_id)]
else: else:
use_response_rule = config.response_rules['default'] use_response_rule = config['response_rules']['default']
if 'random_rate' in use_response_rule: if 'random_rate' in use_response_rule:
import random import random
@@ -102,25 +102,25 @@ class QQBotManager:
ban_group = [] ban_group = []
def __init__(self, first_time_init=True): def __init__(self, first_time_init=True):
import config config = context.get_config_manager().data
self.timeout = config.process_message_timeout self.timeout = config['process_message_timeout']
self.retry = config.retry_times self.retry = config['retry_times']
# 由于YiriMirai的bot对象是单例的，且shutdown方法暂时无法使用 # 由于YiriMirai的bot对象是单例的，且shutdown方法暂时无法使用
# 故只在第一次初始化时创建bot对象，重载之后使用原bot对象 # 故只在第一次初始化时创建bot对象，重载之后使用原bot对象
# 因此，bot的配置不支持热重载 # 因此，bot的配置不支持热重载
if first_time_init: if first_time_init:
logging.debug("Use adapter:" + config.msg_source_adapter) logging.debug("Use adapter:" + config['msg_source_adapter'])
if config.msg_source_adapter == 'yirimirai': if config['msg_source_adapter'] == 'yirimirai':
from pkg.qqbot.sources.yirimirai import YiriMiraiAdapter from pkg.qqbot.sources.yirimirai import YiriMiraiAdapter
mirai_http_api_config = config.mirai_http_api_config mirai_http_api_config = config['mirai_http_api_config']
self.bot_account_id = config.mirai_http_api_config['qq'] self.bot_account_id = config['mirai_http_api_config']['qq']
self.adapter = YiriMiraiAdapter(mirai_http_api_config) self.adapter = YiriMiraiAdapter(mirai_http_api_config)
elif config.msg_source_adapter == 'nakuru': elif config['msg_source_adapter'] == 'nakuru':
from pkg.qqbot.sources.nakuru import NakuruProjectAdapter from pkg.qqbot.sources.nakuru import NakuruProjectAdapter
self.adapter = NakuruProjectAdapter(config.nakuru_config) self.adapter = NakuruProjectAdapter(config['nakuru_config'])
self.bot_account_id = self.adapter.bot_account_id self.bot_account_id = self.adapter.bot_account_id
else: else:
self.adapter = context.get_qqbot_manager().adapter self.adapter = context.get_qqbot_manager().adapter
@@ -176,7 +176,7 @@ class QQBotManager:
stranger_message_handler, stranger_message_handler,
) )
# nakuru不区分好友和陌生人，故仅为yirimirai注册陌生人事件 # nakuru不区分好友和陌生人，故仅为yirimirai注册陌生人事件
if config.msg_source_adapter == 'yirimirai': if config['msg_source_adapter'] == 'yirimirai':
self.adapter.register_listener( self.adapter.register_listener(
StrangerMessage, StrangerMessage,
on_stranger_message on_stranger_message
@@ -213,12 +213,11 @@ class QQBotManager:
用于在热重载流程中卸载所有事件处理器 用于在热重载流程中卸载所有事件处理器
""" """
import config
self.adapter.unregister_listener( self.adapter.unregister_listener(
FriendMessage, FriendMessage,
on_friend_message on_friend_message
) )
if config.msg_source_adapter == 'yirimirai': if config['msg_source_adapter'] == 'yirimirai':
self.adapter.unregister_listener( self.adapter.unregister_listener(
StrangerMessage, StrangerMessage,
on_stranger_message on_stranger_message
@@ -243,10 +242,10 @@ class QQBotManager:
if hasattr(banlist, "enable_group"): if hasattr(banlist, "enable_group"):
self.enable_group = banlist.enable_group self.enable_group = banlist.enable_group
config = context.get_config() config = context.get_config_manager().data
if os.path.exists("sensitive.json") \ if os.path.exists("sensitive.json") \
and config.sensitive_word_filter is not None \ and config['sensitive_word_filter'] is not None \
and config.sensitive_word_filter: and config['sensitive_word_filter']:
with open("sensitive.json", "r", encoding="utf-8") as f: with open("sensitive.json", "r", encoding="utf-8") as f:
sensitive_json = json.load(f) sensitive_json = json.load(f)
self.reply_filter = qqbot_filter.ReplyFilter( self.reply_filter = qqbot_filter.ReplyFilter(
@@ -258,16 +257,16 @@ class QQBotManager:
self.reply_filter = qqbot_filter.ReplyFilter([]) self.reply_filter = qqbot_filter.ReplyFilter([])
def send(self, event, msg, check_quote=True, check_at_sender=True): def send(self, event, msg, check_quote=True, check_at_sender=True):
config = context.get_config() config = context.get_config_manager().data
if check_at_sender and config.at_sender: if check_at_sender and config['at_sender']:
msg.insert( msg.insert(
0, 0,
Plain(" \n") Plain(" \n")
) )
# 当回复的正文中包含换行时，quote可能会自带at，此时就不再单独添加at，只添加换行 # 当回复的正文中包含换行时，quote可能会自带at，此时就不再单独添加at，只添加换行
if "\n" not in str(msg[1]) or config.msg_source_adapter == 'nakuru': if "\n" not in str(msg[1]) or config['msg_source_adapter'] == 'nakuru':
msg.insert( msg.insert(
0, 0,
At( At(
@@ -278,14 +277,15 @@ class QQBotManager:
self.adapter.reply_message( self.adapter.reply_message(
event, event,
msg, msg,
quote_origin=True if config.quote_origin and check_quote else False quote_origin=True if config['quote_origin'] and check_quote else False
) )
# 私聊消息处理 # 私聊消息处理
def on_person_message(self, event: MessageEvent): def on_person_message(self, event: MessageEvent):
import config
reply = '' reply = ''
config = context.get_config_manager().data
if not self.enable_private: if not self.enable_private:
logging.debug("已在banlist.py中禁用所有私聊") logging.debug("已在banlist.py中禁用所有私聊")
elif event.sender.id == self.bot_account_id: elif event.sender.id == self.bot_account_id:
@@ -299,7 +299,7 @@ class QQBotManager:
for i in range(self.retry): for i in range(self.retry):
try: try:
@func_timeout.func_set_timeout(config.process_message_timeout) @func_timeout.func_set_timeout(config['process_message_timeout'])
def time_ctrl_wrapper(): def time_ctrl_wrapper():
reply = processor.process_message('person', event.sender.id, str(event.message_chain), reply = processor.process_message('person', event.sender.id, str(event.message_chain),
event.message_chain, event.message_chain,
@@ -326,8 +326,10 @@ class QQBotManager:
# 群消息处理 # 群消息处理
def on_group_message(self, event: GroupMessage): def on_group_message(self, event: GroupMessage):
import config
reply = '' reply = ''
config = context.get_config_manager().data
def process(text=None) -> str: def process(text=None) -> str:
replys = "" replys = ""
if At(self.bot_account_id) in event.message_chain: if At(self.bot_account_id) in event.message_chain:
@@ -337,7 +339,7 @@ class QQBotManager:
failed = 0 failed = 0
for i in range(self.retry): for i in range(self.retry):
try: try:
@func_timeout.func_set_timeout(config.process_message_timeout) @func_timeout.func_set_timeout(config['process_message_timeout'])
def time_ctrl_wrapper(): def time_ctrl_wrapper():
replys = processor.process_message('group', event.group.id, replys = processor.process_message('group', event.group.id,
str(event.message_chain).strip() if text is None else text, str(event.message_chain).strip() if text is None else text,
@@ -385,17 +387,17 @@ class QQBotManager:
# 通知系统管理员 # 通知系统管理员
def notify_admin(self, message: str): def notify_admin(self, message: str):
config = context.get_config() config = context.get_config_manager().data
if config.admin_qq != 0 and config.admin_qq != []: if config['admin_qq'] != 0 and config['admin_qq'] != []:
logging.info("通知管理员:{}".format(message)) logging.info("通知管理员:{}".format(message))
if type(config.admin_qq) == int: if type(config['admin_qq']) == int:
self.adapter.send_message( self.adapter.send_message(
"person", "person",
config.admin_qq, config['admin_qq'],
MessageChain([Plain("[bot]{}".format(message))]) MessageChain([Plain("[bot]{}".format(message))])
) )
else: else:
for adm in config.admin_qq: for adm in config['admin_qq']:
self.adapter.send_message( self.adapter.send_message(
"person", "person",
adm, adm,
@@ -403,17 +405,17 @@ class QQBotManager:
) )
def notify_admin_message_chain(self, message): def notify_admin_message_chain(self, message):
config = context.get_config() config = context.get_config_manager().data
if config.admin_qq != 0 and config.admin_qq != []: if config['admin_qq'] != 0 and config['admin_qq'] != []:
logging.info("通知管理员:{}".format(message)) logging.info("通知管理员:{}".format(message))
if type(config.admin_qq) == int: if type(config['admin_qq']) == int:
self.adapter.send_message( self.adapter.send_message(
"person", "person",
config.admin_qq, config['admin_qq'],
message message
) )
else: else:
for adm in config.admin_qq: for adm in config['admin_qq']:
self.adapter.send_message( self.adapter.send_message(
"person", "person",
adm, adm,