File size: 4,436 Bytes
3a5cf48 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | # coding: utf-8
# -------------------------------------------------------------------
# 宝塔Linux面板
# -------------------------------------------------------------------
# Copyright (c) 2015-2017 宝塔软件(http:#bt.cn) All rights reserved.
# -------------------------------------------------------------------
# Author: wzz <wzz@bt.cn>
# -------------------------------------------------------------------
# ------------------------------
# 网站模型管理控制器
# ------------------------------
import json
import public
import re
class Controller:
def __init__(self):
pass
def model(self, args):
'''
@name 调用指定项目模型
@author wzz <2024/1/24 上午 11:07>
@param {"mod_name":"string<模型名称>","def_name":"string<方法名称>","data":JSON,} <数据类型> 参数描述
@return dict{"status":True/False,"msg":"提示信息"}
'''
try: # 表单验证
if args['mod_name'] in ['base']: return public.return_status_code(1000, '错误的调用!')
public.exists_args('def_name,mod_name', args)
if args['def_name'].find('__') != -1: return public.return_status_code(1000,
'调用的方法名称中不能包含“__”字符')
if not re.match(r"^\w+$", args['mod_name']): return public.return_status_code(1000,
'调用的模块名称中不能包含\w以外的字符')
if not re.match(r"^\w+$", args['def_name']): return public.return_status_code(1000,
'调用的方法名称中不能包含\w以外的字符')
except:
return public.get_error_object()
# 参数处理
module_name = args['mod_name'].strip()
sub_mod_name = args['sub_mod_name'].strip()
mod_name = "{}Mod".format(args['mod_name'].strip())
def_name = args['def_name'].strip()
model_index = None
if 'model_index' in args: model_index = args['model_index']
if not hasattr(args, 'data'): args.data = {}
if args.data:
if isinstance(args.data, str):
try: # 解析为dict_obj
pdata = public.to_dict_obj(json.loads(args.data))
except:
return public.get_error_object()
else:
pdata = args.data
else:
pdata = args
if isinstance(pdata, dict):
pdata = public.to_dict_obj(pdata)
if not isinstance(pdata, public.dict_obj):
return public.return_error("传递的参数不是通用的内部对象")
# 告诉加载器，要加载什么模块
if model_index: pdata.model_index = model_index
if hasattr(args, "_ws"):
pdata._ws = args._ws
# 前置HOOK
hook_index = '{}_{}_LAST'.format(mod_name.upper(), def_name.upper())
hook_result = public.exec_hook(hook_index, pdata)
if isinstance(hook_result, public.dict_obj):
pdata = hook_result # 桥接
elif isinstance(hook_result, dict):
return hook_result # 响应具体错误信息
elif isinstance(hook_result, bool):
if not hook_result: # 直接中断操作
return public.return_data(False, {}, error_msg='前置HOOK中断操作')
# 调用处理方法
# result = run_object(pdata)
import PluginLoader
result = PluginLoader.module_run("{}/{}".format(module_name, sub_mod_name), def_name, pdata)
if isinstance(result, dict):
if 'status' in result and result['status'] == False and 'msg' in result:
if isinstance(result['msg'], str):
if result['msg'].find('Traceback ') != -1:
raise public.PanelError(result['msg'])
# 后置HOOK
hook_index = '{}_{}_END'.format(mod_name.upper(), def_name.upper())
hook_data = public.to_dict_obj({
'args': pdata,
'result': result
})
hook_result = public.exec_hook(hook_index, hook_data)
if isinstance(hook_result, dict):
result = hook_result['result']
return result
|