def get_setting():
"""
get setting
:return: respond model
"""
request_model = RequestModel(request)
respond_model = RespondModel()
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
setting = get_setting_by_username(user_info_jwt.get('name'))
if setting:
respond_model.data = setting.get('setting')
else:
respond_model.data = {}
return respond_model
def install_by_version():
"""
get all user info
:return:
"""
respond_model = RespondModel()
respond_model.data = get_all_user_info()
return respond_model
def all_plugin_info(lang):
"""
return all plugin info available in the server
:return: respond_model
"""
respond_model = RespondModel()
respond_model.data = get_all_plugin_info(lang)
return respond_model
def get_node():
"""
get current node info
"""
respond_model = RespondModel()
node_info = get_node_info()
if node_info:
respond_model.data = node_info
return respond_model
def del_user(username):
"""
del user
:return:
"""
respond_model = RespondModel()
respond_model.data = del_user_by_username(username)
delete_user_setting(username)
return respond_model
def cluster_run_scan():
"""
heart beat check
"""
request_model = RequestModel(request)
data = request_model.data
respond_model = RespondModel()
respond_model.data = run_scan_from_master(data.get('plugin_name'), data.get('user_setting'), data.get('query'))
return respond_model
def plugin_infos():
"""
return all plugins info which user have the authority to access
:return: respond_model
"""
plugin_names = request.args.get("pluginNames")
lang = request.args.get("lang")
respond_model = RespondModel()
respond_model.data = get_plugin_infos(plugin_names, lang)
return respond_model
def get_plugin_version_info():
"""
get all plugin version from github
:return:
"""
respond_model = RespondModel()
request_model = RequestModel(request)
respond_model.data = get_plugin_version_from_github(
request_model.data.get('github'))
return respond_model
def del_user(username):
"""
del user
:return:
"""
respond_model = RespondModel()
respond_model.data = del_user_by_username(username)
del_libraries_setting_by_username(username)
del_plugin_setting_by_username(username)
return respond_model
def sign_up():
"""
user sign up
:return:
"""
request_model = RequestModel(request)
respond_model = RespondModel()
respond_model.data = user_service.user_sign_up(request_model.data.get('user_info'))
respond_model.code = 20000
return respond_model.dump_json()
def test_canvas():
"""
test canvas setting
:return: respond_model
"""
request_model = RequestModel(request)
user = request_model.data.get('user_info')
respond_model = RespondModel()
respond_model.data = check_canvas_setting(user)
return respond_model
def libraries_settings():
"""
return all libraries settings by user
:return: respond_model
"""
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
respond_model = RespondModel()
respond_model.data = get_libraries_settings(user_info_jwt['name'])
return respond_model
def detail():
"""
return all libraries settings detail
:return: respond_model
"""
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
respond_model = RespondModel()
respond_model.data = libraries_detail(user_info_jwt['name'], request_model.data.get('libraries'))
return respond_model
def install_new_plugin():
"""
install new plugin
:return:
"""
request_model = RequestModel(request)
respond_model = RespondModel()
respond_model.data = install_plugin(request_model.data.get('github'))
# install plugin requirements
install_plugin_require()
return respond_model
def check_enable_sign_up():
"""
check if the server enable user sign up
: return: respond model with boolean
"""
enable_sign_up = get_config(APP_CONFIG)['ENABLE_SIGN_UP']
respond_model = RespondModel()
respond_model.data = {"enable_sign_up": enable_sign_up}
respond_model.message = 'success'
respond_model.code = 20000
return respond_model.dump_json(), 200
def user_info():
"""
get user info
:return: respond model
"""
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
respond_model = RespondModel()
respond_model.message = 'success'
respond_model.data = user_info_jwt
return respond_model
def install_by_version():
"""
install plugin by version
:return:
"""
request_model = RequestModel(request)
respond_model = RespondModel()
respond_model.data = install_plugin_version(
request_model.data.get('github'), request_model.data.get('version'))
# install plugin requirements
install_plugin_require()
return respond_model
def manual_scan():
"""
manual scan
:return: respond model with metaDate
"""
respond_model = RespondModel()
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
meta_data_list = run_manual_scan(request_model.data, user_info_jwt)
respond_model.data = meta_data_list
return respond_model
def plugin_info(lang, plugin_name):
"""
return plugin info if user have authority
:param lang: language for i18n
:param plugin_name: plugin name
:return: respond_model
"""
respond_model = RespondModel()
if not plugin_authorization(plugin_name):
respond_model.message = 'authorization error'
return respond_model
respond_model.data = get_plugin_info(plugin_name, lang)
return respond_model
def user_info():
"""
get user info
:return: respond model
"""
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
respond_model = RespondModel()
respond_model.message = 'success'
if 'admin' in user_info_jwt.get('roles'):
user_info_jwt['roles'] = 'admin,' + get_all_plugin_name()
respond_model.data = user_info_jwt
return respond_model
def get_admin_log():
"""
get plugin full log
:return: {
"length" : "length of log file",
"log" : "log string"
}
"""
plugin_log_file = 'log/general.log'
respond_model = RespondModel()
file_length = get_log_file_length(plugin_log_file)
log_string = get_log_string(plugin_log_file)
respond_model.data = {"length": file_length, "log": log_string}
return respond_model
def get_plugin_log(plugin_name):
"""
get plugin full log
:param plugin_name: plugin name
:return: {
"length" : "length of log file",
"log" : "log string"
}
"""
plugin_log_file = 'log/plugin/%s.log' % plugin_name
respond_model = RespondModel()
file_length = get_log_file_length(plugin_log_file)
log_string = get_log_string(plugin_log_file)
respond_model.data = {"length": file_length, "log": log_string}
return respond_model
def get_admin_log_new(line_number):
"""
get plugin log from specific line
:param line_number: specific line number
:return: {
"length" : "length of log file",
"log" : "log string"
}
"""
plugin_log_file = 'log/general.log'
respond_model = RespondModel()
file_length = get_log_file_length(plugin_log_file)
log_string = get_new_log(plugin_log_file, line_number)
respond_model.data = {"length": file_length, "log": log_string}
return respond_model
def get_plugin_log_new(plugin_name, line_number):
"""
get plugin log from specific line
:param line_number: specific line number
:param plugin_name: plugin name
:return: {
"length" : "length of log file",
"log" : "log string"
}
"""
plugin_log_file = 'log/plugin/%s.log' % plugin_name
respond_model = RespondModel()
file_length = get_log_file_length(plugin_log_file)
log_string = get_new_log(plugin_log_file, line_number)
respond_model.data = {"length": file_length, "log": log_string}
return respond_model
def plugin_setting(lang, plugin_name):
"""
get plugin setting form and user plugin setting
:param lang: language for i18n
:param plugin_name: plugin name
:return: respond_model
"""
respond_model = RespondModel()
request_model = RequestModel(request)
jwt = request_model.token
user_info_jwt = decode_jwt(jwt)['user_info']
if not plugin_authorization(plugin_name):
respond_model.message = 'authorization error'
return respond_model
data = {
'form': get_plugin_setting(plugin_name, lang),
'userSetting': get_user_plugin_setting(plugin_name, user_info_jwt)
}
respond_model.data = data
return respond_model