86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
"""发言频率控制插件 — 新 SDK 版本
|
||
|
||
通过 /chat 命令设置和查看聊天频率。
|
||
"""
|
||
|
||
from maibot_sdk import MaiBotPlugin, Command
|
||
|
||
|
||
class BetterFrequencyPlugin(MaiBotPlugin):
|
||
"""聊天频率控制插件"""
|
||
|
||
@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
|
||
|
||
|
||
def create_plugin():
|
||
return BetterFrequencyPlugin()
|