d13767ee21
- Add support for configuration reload scopes in the plugin runtime. - Implement validation for SDK plugins to ensure required lifecycle methods are overridden. - Update the configuration update handling to include scope information. - Introduce tests for expression auto-check task and NapCat adapter SDK integration. - Refactor configuration management to support callbacks with variable arguments. - Improve plugin loading and error handling for configuration updates. - Ensure that plugins can manage their own configuration updates effectively.
111 lines
4.0 KiB
Python
111 lines
4.0 KiB
Python
"""发言频率控制插件 — 新 SDK 版本
|
||
|
||
通过 /chat 命令设置和查看聊天频率。
|
||
"""
|
||
|
||
from maibot_sdk import Command, MaiBotPlugin
|
||
|
||
|
||
class BetterFrequencyPlugin(MaiBotPlugin):
|
||
"""聊天频率控制插件"""
|
||
|
||
async def on_load(self) -> None:
|
||
"""处理插件加载。"""
|
||
|
||
async def on_unload(self) -> None:
|
||
"""处理插件卸载。"""
|
||
|
||
@Command(
|
||
"set_talk_frequency",
|
||
description="设置当前聊天的talk_frequency值：/chat talk_frequency <数字> 或 /chat t <数字>",
|
||
pattern=r"^/chat\s+(?:talk_frequency|t)\s+(?P<value>[+-]?\d*\.?\d+)$",
|
||
)
|
||
async def handle_set_talk_frequency(self, stream_id: str = "", matched_groups: dict | None = None, **kwargs):
|
||
"""设置当前聊天的 talk_frequency"""
|
||
if not matched_groups or "value" not in matched_groups:
|
||
return False, "命令格式错误", False
|
||
|
||
value_str = matched_groups["value"]
|
||
if not value_str:
|
||
return False, "无法获取数值参数", False
|
||
|
||
try:
|
||
value = float(value_str)
|
||
except ValueError:
|
||
await self.ctx.send.text("数值格式错误，请输入有效的数字", stream_id)
|
||
return False, "数值格式错误", False
|
||
|
||
if not stream_id:
|
||
return False, "无法获取聊天流信息", False
|
||
|
||
# 设置 talk_frequency
|
||
await self.ctx.frequency.set_adjust(stream_id, value)
|
||
|
||
# 获取当前状态
|
||
current = await self.ctx.frequency.get_current_talk_value(stream_id)
|
||
current_val = current if isinstance(current, (int, float)) else 0
|
||
adjust = await self.ctx.frequency.get_adjust(stream_id)
|
||
adjust_val = adjust if isinstance(adjust, (int, float)) else 1
|
||
base_val = current_val / adjust_val if adjust_val else 0
|
||
|
||
msg = (
|
||
f"已设置当前聊天的talk_frequency调整值为: {value}\n"
|
||
f"当前talk_value: {current_val:.2f}\n"
|
||
f"发言频率调整: {adjust_val:.2f}\n"
|
||
f"基础值: {base_val:.2f}"
|
||
)
|
||
await self.ctx.send.text(msg, stream_id)
|
||
return True, None, False
|
||
|
||
@Command(
|
||
"show_frequency",
|
||
description="显示当前聊天的频率控制状态：/chat show 或 /chat s",
|
||
pattern=r"^/chat\s+(?:show|s)$",
|
||
)
|
||
async def handle_show_frequency(self, stream_id: str = "", **kwargs):
|
||
"""显示当前频率控制状态"""
|
||
if not stream_id:
|
||
return False, "无法获取聊天流信息", False
|
||
|
||
current = await self.ctx.frequency.get_current_talk_value(stream_id)
|
||
current_val = current if isinstance(current, (int, float)) else 0
|
||
adjust = await self.ctx.frequency.get_adjust(stream_id)
|
||
adjust_val = adjust if isinstance(adjust, (int, float)) else 1
|
||
base_val = current_val / adjust_val if adjust_val else 0
|
||
|
||
status_msg = (
|
||
"当前聊天频率控制状态\n"
|
||
"Talk Value (发言频率):\n\n"
|
||
f" • 基础值: {base_val:.2f}\n"
|
||
f" • 发言频率调整: {adjust_val:.2f}\n"
|
||
f" • 当前值: {current_val:.2f}\n\n"
|
||
"使用命令:\n"
|
||
" • /chat talk_frequency <数字> 或 /chat t <数字> - 设置发言频率调整\n"
|
||
" • /chat show 或 /chat s - 显示当前状态"
|
||
)
|
||
await self.ctx.send.text(status_msg, stream_id)
|
||
return True, None, False
|
||
|
||
async def on_config_update(self, scope: str, config_data: dict[str, object], version: str) -> None:
|
||
"""处理配置热重载事件。
|
||
|
||
Args:
|
||
scope: 配置变更范围。
|
||
config_data: 最新配置数据。
|
||
version: 配置版本号。
|
||
"""
|
||
|
||
del scope
|
||
del config_data
|
||
del version
|
||
|
||
|
||
def create_plugin() -> BetterFrequencyPlugin:
|
||
"""创建聊天频率插件实例。
|
||
|
||
Returns:
|
||
BetterFrequencyPlugin: 新的聊天频率插件实例。
|
||
"""
|
||
|
||
return BetterFrequencyPlugin()
|