93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
|
import os
|
|||
|
|
import sys
|
|||
|
|
from typing import Optional, Type, TypeVar
|
|||
|
|
import traceback
|
|||
|
|
from importlib import import_module
|
|||
|
|
|
|||
|
|
from .base_task import BaseTask
|
|||
|
|
from .util import GET_CLASS, get_client_ip, debug_log
|
|||
|
|
|
|||
|
|
|
|||
|
|
T_CLS = TypeVar('T_CLS', bound=BaseTask)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_task_cls_by_function(
|
|||
|
|
name: str,
|
|||
|
|
func_name: str,
|
|||
|
|
is_model: bool = False,
|
|||
|
|
model_index: str = '',
|
|||
|
|
args: Optional[dict] = None,
|
|||
|
|
sub_name: Optional[str] = None,
|
|||
|
|
) -> Optional[Type[T_CLS]]:
|
|||
|
|
"""
|
|||
|
|
从执行函数的结果中获取任务类
|
|||
|
|
@param model_index: 模块来源，例如：新场景就是mod
|
|||
|
|
@param name: 名称
|
|||
|
|
@param func_name: 函数名称
|
|||
|
|
@param is_model: 是否在Model中，不在Model中，就应该在插件中
|
|||
|
|
@param args: 请求这个接口的参数， 默认为空
|
|||
|
|
@param sub_name: 自分类名称， 如果有，则会和主名称name做拼接
|
|||
|
|
@return: 返回None 或者有效的任务类
|
|||
|
|
"""
|
|||
|
|
import PluginLoader
|
|||
|
|
real_name = name
|
|||
|
|
if isinstance(sub_name, str):
|
|||
|
|
real_name = "{}/{}".format(name, sub_name)
|
|||
|
|
|
|||
|
|
get_obj = GET_CLASS()
|
|||
|
|
if args is not None and isinstance(args, dict):
|
|||
|
|
for key, value in args.items():
|
|||
|
|
setattr(get_obj, key, value)
|
|||
|
|
try:
|
|||
|
|
if is_model:
|
|||
|
|
get_obj.model_index = model_index
|
|||
|
|
res = PluginLoader.module_run(real_name, func_name, get_obj)
|
|||
|
|
else:
|
|||
|
|
get_obj.fun = func_name
|
|||
|
|
get_obj.s = func_name
|
|||
|
|
get_obj.client_ip = get_client_ip
|
|||
|
|
res = PluginLoader.plugin_run(name, func_name, get_obj)
|
|||
|
|
except:
|
|||
|
|
debug_log(traceback.format_exc())
|
|||
|
|
return None
|
|||
|
|
if isinstance(res, dict):
|
|||
|
|
return None
|
|||
|
|
elif isinstance(res, BaseTask):
|
|||
|
|
return res.__class__
|
|||
|
|
elif issubclass(res, BaseTask):
|
|||
|
|
return res
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_task_cls_by_path(path: str, cls_name: str) -> Optional[Type[T_CLS]]:
|
|||
|
|
try:
|
|||
|
|
# 插件优先检测路径
|
|||
|
|
path_sep = path.split(".")
|
|||
|
|
if len(path_sep) >= 2 and path_sep[0] == "plugin":
|
|||
|
|
plugin_path = "/www/server/panel/plugin/{}".format(path_sep[1])
|
|||
|
|
if not os.path.isdir(plugin_path):
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
module = import_module(path)
|
|||
|
|
cls = getattr(module, cls_name, None)
|
|||
|
|
if issubclass(cls, BaseTask):
|
|||
|
|
return cls
|
|||
|
|
elif isinstance(cls, BaseTask):
|
|||
|
|
return cls.__class__
|
|||
|
|
else:
|
|||
|
|
debug_log("Error: The loaded class is not a subclass of BaseTask")
|
|||
|
|
return None
|
|||
|
|
except ModuleNotFoundError as e:
|
|||
|
|
# todo暂时忽略 ssl_push
|
|||
|
|
if 'mod.base.push_mod.ssl_push' in str(e):
|
|||
|
|
return None
|
|||
|
|
else:
|
|||
|
|
debug_log(traceback.format_exc())
|
|||
|
|
debug_log("ModuleNotFoundError: {}".format(str(e)))
|
|||
|
|
return None
|
|||
|
|
except Exception:
|
|||
|
|
print(traceback.format_exc())
|
|||
|
|
print(sys.path)
|
|||
|
|
debug_log(traceback.format_exc())
|
|||
|
|
return None
|