Initial YakPanel commit
This commit is contained in:
Niranjan
419
class_v2/crontabModelV2/scriptModel.py
Normal file
419
class_v2/crontabModelV2/scriptModel.py
Normal file
@@ -0,0 +1,419 @@
|
||||
#coding: utf-8
|
||||
#-------------------------------------------------------------------
|
||||
# yakpanel
|
||||
#-------------------------------------------------------------------
|
||||
# Copyright (c) 2015-2099 yakpanel(http://yakpanel.com All rights reserved.
|
||||
#-------------------------------------------------------------------
|
||||
# Author: hwliang <hwl@yakpanel.com>
|
||||
#-------------------------------------------------------------------
|
||||
|
||||
#------------------------------
|
||||
# 任务脚本
|
||||
#------------------------------
|
||||
|
||||
from crontabModelV2.base import crontabBase
|
||||
import db
|
||||
import public
|
||||
import time
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
|
||||
class main(crontabBase):
|
||||
|
||||
_sql = None
|
||||
_return_type = ['string','int','float']
|
||||
_script_type = ['bash','python']
|
||||
_log_type = 'Task Scheduling - Script Library'
|
||||
|
||||
def __init__(self) -> None:
|
||||
try:
|
||||
super().__init__()
|
||||
except:
|
||||
public.repair_db()
|
||||
super().__init__()
|
||||
self._sql = db.Sql().dbfile(self.dbfile)
|
||||
|
||||
def get_script_list(self,args = None):
|
||||
'''
|
||||
@name 获取脚本列表
|
||||
@author hwliang
|
||||
@param args<dict_obj>{
|
||||
p<int> 分页
|
||||
rows<int> 每页数量
|
||||
search<string> 搜索关键字
|
||||
}
|
||||
@return list
|
||||
'''
|
||||
p = 1
|
||||
if args and 'p' in args: p = int(args.p)
|
||||
rows = 10
|
||||
if args and 'rows' in args: rows = int(args.rows)
|
||||
search = ''
|
||||
if args and 'search' in args: search = args.search
|
||||
tojs = ''
|
||||
if args and 'tojs' in args: tojs = args.tojs
|
||||
where = ''
|
||||
params = []
|
||||
if search:
|
||||
where = "(name like ? or ps like ?)"
|
||||
params.append('%' + search + '%')
|
||||
params.append('%' + search + '%')
|
||||
|
||||
if 'type_id' in args:
|
||||
if where: where += ' and '
|
||||
where += "type_id=?"
|
||||
|
||||
params.append(args.type_id)
|
||||
|
||||
if 'is_baota' in args:
|
||||
if where: where += ' and '
|
||||
where += "is_baota=?"
|
||||
params.append(args.is_baota)
|
||||
|
||||
if 'return_type' in args:
|
||||
if where: where += ' and '
|
||||
where += "return_type=?"
|
||||
params.append(args.return_type)
|
||||
|
||||
if 'script_type' in args:
|
||||
if where: where += ' and '
|
||||
where += "script_type=?"
|
||||
params.append(args.script_type)
|
||||
|
||||
if 'status' in args:
|
||||
if where: where += ' and '
|
||||
where += "status=?"
|
||||
params.append(args.status)
|
||||
|
||||
count = self._sql.table('scripts').where(where,params).count()
|
||||
data = public.get_page(count,p,rows,tojs)
|
||||
data['data'] = self._sql.table('scripts').where(where,params).order('script_id desc').limit(data['row'],data['shift']).select()
|
||||
for i in data['data']:
|
||||
i['last_exec_time'] = self._sql.table('tasks').where('script_id=?',(i['script_id'],)).order('log_id desc').getField('start_time')
|
||||
return public.return_message(0,0,data)
|
||||
|
||||
|
||||
def get_script_list_by_type(self,args):
|
||||
'''
|
||||
@name 按分类获取脚本列表
|
||||
@author hwliang
|
||||
@return list
|
||||
'''
|
||||
try:
|
||||
data = self._sql.table('types').select()
|
||||
for i in data:
|
||||
i['script_list'] = self._sql.table('scripts').where('type_id=?',i['type_id']).select()
|
||||
except Exception as ex:
|
||||
if str(ex).find('integers') != -1:
|
||||
journal = self.dbfile + '-journal'
|
||||
if os.path.exists(self.dbfile) and os.path.exists(journal):
|
||||
os.remove(journal)
|
||||
os.remove(self.dbfile)
|
||||
|
||||
self.create_table()
|
||||
data = self._sql.table('types').select()
|
||||
for i in data:
|
||||
i['script_list'] = self._sql.table('scripts').where('type_id=?',i['type_id']).select()
|
||||
|
||||
return public.return_message(0,0,data)
|
||||
|
||||
def get_type_list(self,args):
|
||||
'''
|
||||
@name 获取分类列表
|
||||
@author hwliang
|
||||
@return list
|
||||
'''
|
||||
return self._sql.table('types').select()
|
||||
|
||||
def get_script_type(self,script_body):
|
||||
'''
|
||||
@name 获取脚本类型
|
||||
@param script_body<string> 脚本内容
|
||||
@return string
|
||||
'''
|
||||
|
||||
if script_body.find('import ') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/bin/bash') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/bin/sh') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/usr/bin/env python') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/usr/bin/env bash') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/usr/bin/env sh') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/usr/bin/python') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/usr/bin/python3') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/usr/bin/python2') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/usr/bin/bash') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/usr/bin/sh') != -1:
|
||||
return 'bash'
|
||||
elif script_body.find('#!/bin/python') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/bin/python3') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#!/bin/python2') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding: utf-8') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding:utf-8') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding:gbk') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding: gbk') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding:gb2312') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding: gb2312') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding:utf-8') != -1:
|
||||
return 'python'
|
||||
elif script_body.find('#coding:utf-8') != -1:
|
||||
return 'python'
|
||||
else:
|
||||
return 'bash'
|
||||
|
||||
|
||||
|
||||
|
||||
def create_script(self,args):
|
||||
'''
|
||||
@name 创建脚本
|
||||
@author hwliang
|
||||
@param args<dict_obj>{
|
||||
name<string> 脚本名称
|
||||
script<string> 脚本内容
|
||||
type_id<int> 脚本类型
|
||||
ps<string> 脚本描述
|
||||
return_type<string> 返回类型 (string|int|float|json_object)
|
||||
}
|
||||
@return bool
|
||||
'''
|
||||
name = args.get('name','')
|
||||
script = args.get('script','')
|
||||
type_id = args.get('type_id',7)
|
||||
is_baota = args.get('is_baota',0)
|
||||
author = args.get('author','')
|
||||
ps = args.get('ps','')
|
||||
return_type = args.get('return_type','string')
|
||||
script_type = self.get_script_type(script)
|
||||
version = args.get('version','1.0')
|
||||
is_args = args.get('is_args/d',0)
|
||||
args_title = args.get('args_title','')
|
||||
args_ps = args.get('args_ps','')
|
||||
if not type_id: return public.return_message(-1,0,'Classification ID cannot be empty')
|
||||
if not name or not script: return public.return_message(-1,0,'Script name or content cannot be empty')
|
||||
if not return_type in self._return_type: return public.return_message(-1,0,'Return type error')
|
||||
if not script_type in self._script_type: return public.return_message(-1,0,'Script type error')
|
||||
|
||||
pdata = {
|
||||
'name':name,
|
||||
'script':script,
|
||||
'type_id':type_id,
|
||||
'is_baota':is_baota,
|
||||
'author':author,
|
||||
'ps':ps,
|
||||
'return_type':return_type,
|
||||
'script_type':script_type,
|
||||
'version':version,
|
||||
'is_args':is_args,
|
||||
'args_title':args_title,
|
||||
'args_ps':args_ps,
|
||||
'create_time':int(time.time())
|
||||
}
|
||||
|
||||
if self._sql.table('scripts').where("name=?",(name,)).count():
|
||||
return public.return_message(-1,0,'Script name already exists')
|
||||
res = self._sql.table('scripts').insert(pdata)
|
||||
if not res:
|
||||
return public.return_message(-1,0,'Creation failed')
|
||||
public.WriteLog(self._log_type,'Successfully created script ['+name+']')
|
||||
return public.return_message(0,0,'Created successfully')
|
||||
|
||||
|
||||
|
||||
|
||||
def remove_script(self,args):
|
||||
'''
|
||||
@name 删除脚本
|
||||
@author hwliang
|
||||
@param args<dict_obj>{
|
||||
script_id<int> 脚本ID
|
||||
}
|
||||
@return dict
|
||||
'''
|
||||
script_id = args.get('script_id',0)
|
||||
if not script_id: return public.return_message(-1,0,public.lang('Script ID cannot be empty'))
|
||||
script_info = self._sql.table('scripts').where('script_id=?',(script_id,)).find()
|
||||
if not script_info: return public.return_message(-1,0,public.lang('The specified script does not exist'))
|
||||
if script_info['is_baota'] == 1: return public.return_message(-1,0,public.lang('The built-in script of yakpanel cannot be deleted'))
|
||||
trigger_name = self._sql.table('trigger').where('script_id=?',(script_id,)).getField('name')
|
||||
if trigger_name:
|
||||
return public.return_message(-1,0,public.lang('This script is currently being used by task [{}], please delete the task first!',(trigger_name)))
|
||||
|
||||
trigger_id = self._sql.table('operator_where').where('script_id=?',(script_id,)).getField('trigger_id')
|
||||
if trigger_id:
|
||||
trigger_name = self._sql.table('trigger').where('id=?',(trigger_id,)).getField('name')
|
||||
return public.return_message(-1,0,public.lang('This script is currently being used by the event of task [{}], please delete the task event first!',(trigger_name)))
|
||||
|
||||
if not self._sql.table('scripts').where('script_id=?',(script_id,)).delete():
|
||||
return public.return_message(-1,0,public.lang('Delete failed'))
|
||||
self._sql.table('tasks').where('script_id=? and where_id=0 and trigger_id=0',(script_id,)).delete()
|
||||
public.WriteLog(public.lang(self._log_type),public.lang('Successfully deleted script ['+script_info['name']+']'))
|
||||
return public.return_message(0,0,public.lang('Delete successfully'))
|
||||
|
||||
|
||||
def modify_script(self,args):
|
||||
'''
|
||||
@name 修改脚本信息
|
||||
@author hwliang
|
||||
@param args<dict_obj>{
|
||||
id<int> 脚本ID
|
||||
name<string> 脚本名称
|
||||
script<string> 脚本内容
|
||||
type<string> 脚本类型
|
||||
ps<string> 脚本描述
|
||||
return_type<string> 返回类型 (string|int|float)
|
||||
}
|
||||
@return bool
|
||||
'''
|
||||
|
||||
script_id = args.get('script_id',0)
|
||||
name = args.get('name','')
|
||||
script = args.get('script','')
|
||||
type_id = args.get('type_id',7)
|
||||
# is_baota = args.get('is_baota',0)
|
||||
# author = args.get('author','')
|
||||
ps = args.get('ps','')
|
||||
return_type = args.get('return_type','string')
|
||||
is_args = args.get('is_args/d',0)
|
||||
args_title = args.get('args_title','')
|
||||
args_ps = args.get('args_ps','')
|
||||
script_type = self.get_script_type(script)
|
||||
# version = args.get('version','1.0')
|
||||
if not script_id: return public.return_message(-1,0,public.lang('Script ID cannot be empty'))
|
||||
if not name or not script: return public.return_message(-1,0,public.lang('Script name or content cannot be empty'))
|
||||
if not return_type in self._return_type: return public.return_message(-1,0,public.lang('Return type error'))
|
||||
if not script_type in self._script_type: return public.return_message(-1,0,public.lang('Script type error'))
|
||||
|
||||
pdata = {
|
||||
'name':name,
|
||||
'script':script,
|
||||
'type_id':type_id,
|
||||
'ps':ps,
|
||||
'return_type':return_type,
|
||||
'script_type':script_type,
|
||||
'is_args':is_args,
|
||||
'args_title':args_title,
|
||||
'args_ps':args_ps,
|
||||
}
|
||||
|
||||
script_info = self._sql.table('scripts').where('script_id=?',(script_id,)).find()
|
||||
if not script_info:
|
||||
return public.return_message(-1,0,public.lang('The specified script does not exist'))
|
||||
if not self._sql.table('scripts').where('script_id=?',(script_id,)).update(pdata):
|
||||
return public.return_message(-1,0,public.lang('Modification failed'))
|
||||
public.WriteLog(public.lang(self._log_type),public.lang('Successfully modified script ['+name+']'))
|
||||
return public.return_message(0,0,public.lang('Modified successfully'))
|
||||
|
||||
def add_task_log(self,script_id,trigger_id,where_id,status,result_succ,result_err,start_time,end_time):
|
||||
'''
|
||||
@name 添加任务日志
|
||||
@param script_id<int> 脚本ID
|
||||
@param trigger_id<int> 触发器ID
|
||||
@param where_id<int> 事件ID
|
||||
@param status<int> 任务状态
|
||||
@param result_succ<string> 成功结果
|
||||
@param result_err<string> 错误结果
|
||||
@param start_time<int> 开始时间
|
||||
@param end_time<int> 结束时间
|
||||
@return bool
|
||||
'''
|
||||
return self._sql.table('tasks').add('script_id,trigger_id,where_id,status,result_succ,result_err,start_time,end_time',
|
||||
(script_id,trigger_id,where_id,status,result_succ,result_err,start_time,end_time))
|
||||
|
||||
|
||||
def get_script_logs(self,args):
|
||||
'''
|
||||
@name 获取脚本执行日志
|
||||
@param args<dict_obj>{
|
||||
"script_id": <int> 脚本ID
|
||||
"p": <int> 页码
|
||||
"rows": <int> 每页数量
|
||||
}
|
||||
'''
|
||||
script_id = int(args.get('script_id',0))
|
||||
if not script_id: return public.return_message(-1,0,'Script ID cannot be empty')
|
||||
p = 1
|
||||
if 'p' in args: p = int(args['p'])
|
||||
rows = 10
|
||||
if 'rows' in args: rows = int(args['rows'])
|
||||
tojs = ''
|
||||
if 'tojs' in args: tojs = args.tojs
|
||||
|
||||
where = 'script_id=?'
|
||||
where_args = (script_id,)
|
||||
count = self._sql.table('tasks').where(where,where_args).count()
|
||||
page = public.get_page(count,p,rows,tojs)
|
||||
page['data'] = self._sql.table('tasks').where(where,where_args).limit(page['row'],page['shift']).order('log_id desc').select()
|
||||
return public.return_message(0,0,page)
|
||||
|
||||
def test_script(self,args):
|
||||
'''
|
||||
@name 测试运行脚本
|
||||
@author hwliang
|
||||
@param args<dict_obj>{
|
||||
script_id<int> 脚本ID
|
||||
}
|
||||
@return dict
|
||||
'''
|
||||
script_start_time = int(time.time())
|
||||
script_id = args.get('script_id',0)
|
||||
if not script_id: return public.return_message(-1,0,public.lang('Script ID cannot be empty'))
|
||||
script_info = self._sql.table('scripts').where('script_id=?',(script_id,)).field('script,script_type,return_type').find()
|
||||
script_args = args.get('args/s','')
|
||||
if script_args: script_args = ' ' + script_args
|
||||
if 'is_args' in script_info and script_info['is_args'] and not script_args:
|
||||
return public.return_message(-1,0,public.lang('This script requires parameters'))
|
||||
if not script_info: return public.return_message(-1,0,public.lang('The specified script does not exist'))
|
||||
tmp_file= '{}/tmp/{}.tmp'.format(public.get_panel_path(),public.md5(script_info['script']))
|
||||
try:
|
||||
public.writeFile(tmp_file,script_info['script'])
|
||||
if script_info['script_type'] == 'bash':
|
||||
result = public.ExecShell("bash {}{}".format(tmp_file,script_args))
|
||||
elif script_info['script_type'] == 'python':
|
||||
python_bin = public.get_python_bin()
|
||||
result = public.ExecShell('{} -u {}{}'.format(python_bin,tmp_file,script_args))
|
||||
result_zero=result[0]
|
||||
search_string=' local variable start_time where it is not associated with a value\n'
|
||||
if result[0].find(search_string) != -1:
|
||||
result_zero = result[0].split(search_string)[1]
|
||||
res = result_zero.strip()
|
||||
if result[1]:
|
||||
self.add_task_log(script_id,0,0,0,result_zero,result[1],script_start_time,int(time.time()))
|
||||
return public.return_message(-1,0,public.lang('Script running error, please check if the script code is correct:\n{}',result_zero + "\n" + result[1].split('.tmp:')[-1]))
|
||||
except Exception as ex:
|
||||
error_msg = public.get_error_info()
|
||||
self.add_task_log(script_id,0,0,0,'',error_msg,script_start_time,int(time.time()))
|
||||
return public.return_message(-1,0,public.lang('Execution failed, error message:\n{}',str(error_msg)))
|
||||
finally:
|
||||
if os.path.exists(tmp_file): os.remove(tmp_file)
|
||||
self.add_task_log(script_id,0,0,1,result_zero,result[1],script_start_time,int(time.time()))
|
||||
return public.return_message(0,0,public.lang("Execution succeeded: \n{}",str(public.xssencode2(res))))
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user