40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import logging
|
|
import tomllib
|
|
|
|
from wcferry import WxMsg, Wcf
|
|
|
|
from robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
|
|
from xiuren.random_pic import get_xiuren_pic
|
|
|
|
|
|
class Xiuren:
|
|
def __init__(self, wcf: Wcf, gbm: GroupBotManager):
|
|
self.LOG = logging.getLogger(__name__)
|
|
self.wcf = wcf # 假设 wcf 对象在此类中初始化
|
|
self.gbm = gbm # 权限功能
|
|
with open("xiuren/config.toml", "rb") as f:
|
|
plugin_config = tomllib.load(f)
|
|
|
|
config = plugin_config["Xiuren"]
|
|
|
|
self.enable = config["enable"]
|
|
self.command = config["command"]
|
|
self.LOG.info(f"[秀人] 组件初始化完成，指令： {self.command}")
|
|
|
|
def handle_message(self, message: WxMsg):
|
|
if not self.enable:
|
|
return
|
|
|
|
content = str(message.content).strip()
|
|
command = content.split(" ")
|
|
|
|
if command[0] not in self.command:
|
|
return
|
|
|
|
# 如果触发了指令，但是没有权限，则返回权限不足
|
|
if self.gbm.get_group_permission(message.roomid, Feature.PIC) == PermissionStatus.DISABLED:
|
|
return
|
|
|
|
pic_path = get_xiuren_pic()
|
|
self.wcf.send_file(pic_path, message.roomid)
|