Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-06 20:29:59 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

版本更新

Browse Source
This commit is contained in:
HibiKier
2024-10-20 21:11:16 +08:00
parent 4e33bf3a50
commit d594d17bf6
253 changed files with 34987 additions and 2241 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zhenxun/builtin_plugins/init/manager.py
+22 -18
View File
@@ -73,7 +73,7 @@ class Manager:
def add(
self,
module_path: str,
module: str,
data: BaseBlock | PluginCdBlock | PluginCountBlock | PluginLimit,
):
"""添加限制"""
@@ -106,20 +106,20 @@ class Manager:
max_count=data.max_count,
)
if isinstance(data, PluginCdBlock):
self.cd_data[module_path] = data
self.cd_data[module] = data
elif isinstance(data, PluginCountBlock):
self.count_data[module_path] = data
self.count_data[module] = data
elif isinstance(data, BaseBlock):
self.block_data[module_path] = data
self.block_data[module] = data
def exist(self, module_path: str, type: PluginLimitType):
def exist(self, module: str, type: PluginLimitType):
"""是否存在"""
if type == PluginLimitType.CD:
return module_path in self.cd_data
return module in self.cd_data
elif type == PluginLimitType.BLOCK:
return module_path in self.block_data
return module in self.block_data
elif type == PluginLimitType.COUNT:
return module_path in self.count_data
return module in self.count_data
def init(self):
if not self.cd_file.exists():
@@ -208,6 +208,8 @@ class Manager:
temp = _yaml.load(f)
if "PluginCdLimit" in temp.keys():
for k, v in temp["PluginCdLimit"].items():
if "." in k:
k = k.split(".")[-1]
self.cd_data[k] = PluginCdBlock.parse_obj(v)
def __load_block_file(self):
@@ -217,6 +219,8 @@ class Manager:
temp = _yaml.load(f)
if "PluginBlockLimit" in temp.keys():
for k, v in temp["PluginBlockLimit"].items():
if "." in k:
k = k.split(".")[-1]
self.block_data[k] = BaseBlock.parse_obj(v)
def __load_count_file(self):
@@ -226,6 +230,8 @@ class Manager:
temp = _yaml.load(f)
if "PluginCountLimit" in temp.keys():
for k, v in temp["PluginCountLimit"].items():
if "." in k:
k = k.split(".")[-1]
self.count_data[k] = PluginCountBlock.parse_obj(v)
def __replace_data(
@@ -270,10 +276,10 @@ class Manager:
if not db_data:
return (
PluginLimit(
module=k.split(".")[-1],
module_path=k,
module=k,
module_path=module2plugin[k].module_path,
limit_type=limit_type,
plugin=module2plugin.get(k),
plugin=module2plugin[k],
cd=getattr(limit, "cd", None),
max_count=getattr(limit, "max_count", None),
status=limit.status,
@@ -329,12 +335,10 @@ class Manager:
]
if data := self.__get_file_data(limit_type):
db_type_limit_modules = [
(limit.module_path, limit.id) for limit in db_type_limits
(limit.module, limit.id) for limit in db_type_limits
]
delete_list.extend(
id
for module_path, id in db_type_limit_modules
if module_path not in data.keys()
id for module, id in db_type_limit_modules if module not in data.keys()
)
for k, v in data.items():
if not module2plugin.get(k):
@@ -343,7 +347,7 @@ class Manager:
f"插件模块 {k} 未加载，已过滤当前 {v._type} 限制..."
)
continue
db_data = [limit for limit in db_type_limits if limit.module_path == k]
db_data = [limit for limit in db_type_limits if limit.module == k]
db_data, is_create = self.__set_data(
k, db_data[0] if db_data else None, v, limit_type, module2plugin
)
@@ -369,8 +373,8 @@ class Manager:
+ list(self.block_data.keys())
+ list(self.count_data.keys())
)
plugins = await PluginInfo.get_plugins(module_path__in=modules)
module2plugin = {p.module_path: p for p in plugins}
plugins = await PluginInfo.get_plugins(module__in=modules)
module2plugin = {p.module: p for p in plugins}
create_list, update_list, delete_list = self.__set_db_limits(
db_limits, module2plugin, PluginLimitType.CD
)