mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-07-25 05:46:13 +00:00
refactor: 独立resprule模块
This commit is contained in:
@@ -0,0 +1,67 @@
|
||||
from ..utils import context
|
||||
|
||||
|
||||
# 检查消息是否符合泛响应匹配机制
|
||||
def check_response_rule(group_id:int, text: str):
|
||||
config = context.get_config_manager().data
|
||||
|
||||
rules = config['response_rules']
|
||||
|
||||
# 检查是否有特定规则
|
||||
if 'prefix' not in config['response_rules']:
|
||||
if str(group_id) in config['response_rules']:
|
||||
rules = config['response_rules'][str(group_id)]
|
||||
else:
|
||||
rules = config['response_rules']['default']
|
||||
|
||||
# 检查前缀匹配
|
||||
if 'prefix' in rules:
|
||||
for rule in rules['prefix']:
|
||||
if text.startswith(rule):
|
||||
return True, text.replace(rule, "", 1)
|
||||
|
||||
# 检查正则表达式匹配
|
||||
if 'regexp' in rules:
|
||||
for rule in rules['regexp']:
|
||||
import re
|
||||
match = re.match(rule, text)
|
||||
if match:
|
||||
return True, text
|
||||
|
||||
return False, ""
|
||||
|
||||
|
||||
def response_at(group_id: int):
|
||||
config = context.get_config_manager().data
|
||||
|
||||
use_response_rule = config['response_rules']
|
||||
|
||||
# 检查是否有特定规则
|
||||
if 'prefix' not in config['response_rules']:
|
||||
if str(group_id) in config['response_rules']:
|
||||
use_response_rule = config['response_rules'][str(group_id)]
|
||||
else:
|
||||
use_response_rule = config['response_rules']['default']
|
||||
|
||||
if 'at' not in use_response_rule:
|
||||
return True
|
||||
|
||||
return use_response_rule['at']
|
||||
|
||||
|
||||
def random_responding(group_id):
|
||||
config = context.get_config_manager().data
|
||||
|
||||
use_response_rule = config['response_rules']
|
||||
|
||||
# 检查是否有特定规则
|
||||
if 'prefix' not in config['response_rules']:
|
||||
if str(group_id) in config['response_rules']:
|
||||
use_response_rule = config['response_rules'][str(group_id)]
|
||||
else:
|
||||
use_response_rule = config['response_rules']['default']
|
||||
|
||||
if 'random_rate' in use_response_rule:
|
||||
import random
|
||||
return random.random() < use_response_rule['random_rate']
|
||||
return False
|
||||
Reference in New Issue
Block a user