mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
111 lines
3.5 KiB
Python
Executable File
111 lines
3.5 KiB
Python
Executable File
from typing import List, Optional, Union, Tuple
|
|
from .data_class import StaticData
|
|
from pathlib import Path
|
|
from ruamel.yaml import YAML
|
|
from ruamel import yaml
|
|
|
|
|
|
_yaml = YAML(typ="safe")
|
|
|
|
|
|
class Plugins2settingsManager(StaticData):
|
|
"""
|
|
插件命令阻塞 管理器
|
|
"""
|
|
|
|
def __init__(self, file: Path):
|
|
self.file = file
|
|
super().__init__(None)
|
|
if file.exists():
|
|
with open(file, "r", encoding="utf8") as f:
|
|
self._data = _yaml.load(f)
|
|
if self._data:
|
|
if "PluginSettings" in self._data.keys():
|
|
self._data = (
|
|
self._data["PluginSettings"] if self._data["PluginSettings"] else {}
|
|
)
|
|
for x in self._data.keys():
|
|
if self._data[x].get("cost_gold") is None:
|
|
self._data[x]["cost_gold"] = 0
|
|
|
|
def add_plugin_settings(
|
|
self,
|
|
plugin: str,
|
|
cmd: Optional[List[str]] = None,
|
|
default_status: Optional[bool] = True,
|
|
level: Optional[int] = 5,
|
|
limit_superuser: Optional[bool] = False,
|
|
plugin_type: Tuple[Union[str, int]] = ("normal",),
|
|
cost_gold: int = 0,
|
|
):
|
|
"""
|
|
添加一个插件设置
|
|
:param plugin: 插件模块名称
|
|
:param cmd: 命令 或 命令别名
|
|
:param default_status: 默认开关状态
|
|
:param level: 功能权限等级
|
|
:param limit_superuser: 功能状态是否限制超级用户
|
|
:param plugin_type: 插件类型
|
|
:param cost_gold: 需要消费的金币
|
|
"""
|
|
level = level or 5
|
|
cmd = cmd or []
|
|
cost_gold = cost_gold or 0
|
|
self._data[plugin] = {
|
|
"level": level if level is not None else 5,
|
|
"default_status": default_status if default_status is not None else True,
|
|
"limit_superuser": (
|
|
limit_superuser
|
|
if limit_superuser is not None
|
|
else False
|
|
),
|
|
"cmd": cmd,
|
|
"plugin_type": list(
|
|
plugin_type if plugin_type is not None else ("normal",)
|
|
),
|
|
"cost_gold": cost_gold,
|
|
}
|
|
|
|
def get_plugin_data(self, module: str) -> dict:
|
|
"""
|
|
通过模块名获取数据
|
|
:param module: 模块名称
|
|
"""
|
|
if self._data.get(module) is not None:
|
|
return self._data.get(module)
|
|
return {}
|
|
|
|
def get_plugin_module(
|
|
self, cmd: str, is_all: bool = False
|
|
) -> Union[str, List[str]]:
|
|
"""
|
|
根据 cmd 获取功能 modules
|
|
:param cmd: 命令
|
|
:param is_all: 获取全部包含cmd的模块
|
|
"""
|
|
keys = []
|
|
for key in self._data.keys():
|
|
if cmd in self._data[key]["cmd"]:
|
|
if is_all:
|
|
keys.append(key)
|
|
else:
|
|
return key
|
|
return keys
|
|
|
|
def save(self, path: Union[str, Path] = None):
|
|
path = path or self.file
|
|
if isinstance(path, str):
|
|
path = Path(path)
|
|
if path:
|
|
with open(path, "w", encoding="utf8") as f:
|
|
yaml.dump({"PluginSettings": self._data}, f, indent=2, Dumper=yaml.RoundTripDumper, allow_unicode=True)
|
|
|
|
def reload(self):
|
|
"""
|
|
重载本地数据
|
|
"""
|
|
if self.file.exists():
|
|
with open(self.file, "r", encoding="utf8") as f:
|
|
self._data: dict = _yaml.load(f)
|
|
self._data = self._data["PluginSettings"]
|