Initial YakPanel commit
This commit is contained in:
Niranjan
0
class_v2/crontabModelV2/__init__.py
Normal file
0
class_v2/crontabModelV2/__init__.py
Normal file
256
class_v2/crontabModelV2/base.py
Normal file
256
class_v2/crontabModelV2/base.py
Normal file
@@ -0,0 +1,256 @@
|
||||
#coding: utf-8
|
||||
#-------------------------------------------------------------------
|
||||
# yakpanel
|
||||
#-------------------------------------------------------------------
|
||||
# Copyright (c) 2015-2099 yakpanel(http://yakpanel.com) All rights reserved.
|
||||
#-------------------------------------------------------------------
|
||||
# Author: hwliang <hwl@yakpanel.com>
|
||||
#-------------------------------------------------------------------
|
||||
import os,sys,time,json
|
||||
panelPath = '/www/server/panel'
|
||||
os.chdir(panelPath)
|
||||
if not panelPath + "/class/" in sys.path:
|
||||
sys.path.insert(0, panelPath + "/class/")
|
||||
if not panelPath + "/class_v2/" in sys.path:
|
||||
sys.path.insert(0, panelPath + "/class_v2/")
|
||||
import public
|
||||
import db
|
||||
class crontabBase:
|
||||
dbfile = public.get_panel_path() + '/data/db/script.db'
|
||||
__columns = None
|
||||
def __init__(self) -> None:
|
||||
if not os.path.exists(self.dbfile):
|
||||
self.create_table()
|
||||
self.check_column()
|
||||
self.sync_scripts()
|
||||
self.sync_types()
|
||||
self.check_yakpanel_column()
|
||||
|
||||
def exists_column(self, sql_pbj,table,column):
|
||||
'''
|
||||
@name 判断列是否存在
|
||||
@author hwliang
|
||||
@param sql_pbj <object> 数据库对象
|
||||
@param table <str> 表名
|
||||
@param column <str> 列名
|
||||
@return <bool> True 存在，False 不存在
|
||||
'''
|
||||
if not self.__columns:
|
||||
self.__columns = sql_pbj.query('PRAGMA table_info('+table+')',())
|
||||
for col in self.__columns:
|
||||
if len(col) < 3: continue
|
||||
if col[1] == column: return True
|
||||
return False
|
||||
|
||||
def check_column(self):
|
||||
'''
|
||||
@name 检查表结构是否完整，不完整则补充完整
|
||||
@autho hwliang
|
||||
@return void
|
||||
'''
|
||||
|
||||
sql_obj = db.Sql().dbfile(self.dbfile)
|
||||
if not sql_obj: return
|
||||
if not self.exists_column(sql_obj,'scripts','is_args'):
|
||||
sql_obj.execute("ALTER TABLE 'scripts' ADD 'is_args' INTEGER DEFAULT 0",())
|
||||
if not self.exists_column(sql_obj,'scripts','args_title'):
|
||||
sql_obj.execute("ALTER TABLE 'scripts' ADD 'args_title' VARCHAR",())
|
||||
if not self.exists_column(sql_obj,'scripts','args_ps'):
|
||||
sql_obj.execute("ALTER TABLE 'scripts' ADD 'args_ps' VARCHAR",())
|
||||
if not self.exists_column(sql_obj,'operator_where','args'):
|
||||
sql_obj.execute("ALTER TABLE 'operator_where' ADD 'args' VARCHAR",())
|
||||
if not self.exists_column(sql_obj,'trigger','args'):
|
||||
sql_obj.execute("ALTER TABLE 'trigger' ADD 'args' VARCHAR",())
|
||||
if sql_obj.table('types').where('type_id=?',3).getField('name') != 'Alarm notification':
|
||||
sql_obj.table('types').where('type_id=?',3).update({'name':'Alarm notification','title':'Script for sending various alarm notifications'})
|
||||
|
||||
def check_yakpanel_column(self):
|
||||
dbfile = public.get_panel_path() + '/data/default.db'
|
||||
sql_obj = db.Sql().dbfile(dbfile)
|
||||
if not sql_obj: return
|
||||
if not self.exists_column(sql_obj,'crontab','db_type'):
|
||||
sql_obj.execute("ALTER TABLE 'crontab' ADD 'db_type' INTEGER DEFAULT 0",())
|
||||
# if not self.exists_column(sql_obj,'scripts','args_title'):
|
||||
# sql_obj.execute("ALTER TABLE 'scripts' ADD 'args_title' VARCHAR",())
|
||||
# if not self.exists_column(sql_obj,'scripts','args_ps'):
|
||||
# sql_obj.execute("ALTER TABLE 'scripts' ADD 'args_ps' VARCHAR",())
|
||||
# if not self.exists_column(sql_obj,'operator_where','args'):
|
||||
# sql_obj.execute("ALTER TABLE 'operator_where' ADD 'args' VARCHAR",())
|
||||
# if not self.exists_column(sql_obj,'trigger','args'):
|
||||
# sql_obj.execute("ALTER TABLE 'trigger' ADD 'args' VARCHAR",())
|
||||
# if sql_obj.table('types').where('type_id=?',3).getField('name') != '告警通知':
|
||||
# sql_obj.table('types').where('type_id=?',3).update({'name':'告警通知','title':'用于发送各种告警通知的脚本'})
|
||||
|
||||
sql_obj.close()
|
||||
def sync_types(self):
|
||||
panel_path = public.get_panel_path()
|
||||
# 检查文件是否存在
|
||||
tfile = '{}/config/script_types.json'.format(panel_path)
|
||||
if not os.path.exists(tfile): return
|
||||
tbody = public.readFile(tfile)
|
||||
if not tbody: return
|
||||
|
||||
# 检查内容是否变更
|
||||
last_sync_md5_file = '{}/data/last_sync_types_md5.pl'.format(panel_path)
|
||||
tmd5 = public.md5(tbody)
|
||||
if os.path.exists(last_sync_md5_file):
|
||||
last_sync_md5 = public.readFile(last_sync_md5_file)
|
||||
if last_sync_md5 == tmd5: return
|
||||
|
||||
# 同步类型库
|
||||
try:
|
||||
type_list = json.loads(tbody)
|
||||
sql_obj = db.Sql().dbfile(self.dbfile)
|
||||
for type_item in type_list:
|
||||
type_info = sql_obj.table('types').where('type_id=?', (type_item['type_id'],)).find()
|
||||
if type_info:
|
||||
# 更新类型信息
|
||||
sql_obj.table('types').where('type_id=?', (type_item['type_id'],)).update(type_item)
|
||||
else:
|
||||
sql_obj.table('types').insert(type_item)
|
||||
sql_obj.close()
|
||||
|
||||
# 记录本次同步内容的MD5
|
||||
public.writeFile(last_sync_md5_file, tmd5)
|
||||
except:
|
||||
return
|
||||
def sync_scripts(self):
|
||||
'''
|
||||
@name 同步脚本库
|
||||
@autho hwliang
|
||||
@return void
|
||||
'''
|
||||
panel_path = public.get_panel_path()
|
||||
# 检查文件是否存在
|
||||
sfile = '{}/config/crontab.json'.format(panel_path)
|
||||
if not os.path.exists(sfile):
|
||||
return
|
||||
sbody = public.readFile(sfile)
|
||||
if not sbody:
|
||||
return
|
||||
|
||||
# 检查内容是否变更
|
||||
last_sync_md5_file = '{}/data/last_sync_md5.pl'.format(panel_path)
|
||||
smd5 = public.md5(sbody)
|
||||
if os.path.exists(last_sync_md5_file):
|
||||
last_sync_md5 = public.readFile(last_sync_md5_file)
|
||||
if last_sync_md5 == smd5:
|
||||
return
|
||||
|
||||
# 同步脚本库
|
||||
try:
|
||||
script_list = json.loads(sbody)
|
||||
sql_obj = db.Sql().dbfile(self.dbfile)
|
||||
for script in script_list:
|
||||
# public.print_log('脚本id:{}'.format(script['script_id']))
|
||||
if 'script_id' in script.keys():
|
||||
del(script['script_id'])
|
||||
script_info = sql_obj.table('scripts').where('name=?',(script['name'],)).find()
|
||||
if script_info:
|
||||
# 更新脚本版本
|
||||
if script_info['version'] == script['version']: continue
|
||||
sql_obj.table('scripts').where('script_id=?',(script_info['script_id'],)).update(script)
|
||||
else:
|
||||
|
||||
sql_obj.table('scripts').insert(script)
|
||||
sql_obj.close()
|
||||
|
||||
# 记录本次同步内容的MD5
|
||||
public.writeFile(last_sync_md5_file,smd5)
|
||||
except:
|
||||
return
|
||||
|
||||
def create_table(self):
|
||||
'''
|
||||
@name 创建表结构
|
||||
@autho hwliang
|
||||
@return void
|
||||
'''
|
||||
sql_obj = db.Sql().dbfile(self.dbfile)
|
||||
if not sql_obj: return
|
||||
sql = '''
|
||||
CREATE TABLE operator_where (
|
||||
where_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
trigger_id INTEGER,
|
||||
operator VARCHAR,
|
||||
op_value VARCHAR,
|
||||
args VARCHAR,
|
||||
run_script_id INTEGER,
|
||||
run_script TEXT,
|
||||
create_time INTEGER DEFAULT (0)
|
||||
)
|
||||
'''
|
||||
sql_obj.execute(sql,())
|
||||
sql_obj.execute('CREATE INDEX tid ON operator_where (trigger_id)',())
|
||||
|
||||
sql = '''
|
||||
CREATE TABLE scripts (
|
||||
script_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
type_id INTEGER DEFAULT (0), is_baota INTEGER DEFAULT (0),
|
||||
name VARCHAR, status INTEGER DEFAULT (1),
|
||||
author VARCHAR, script TEXT,
|
||||
version VARCHAR,
|
||||
return_type VARCHAR DEFAULT string,
|
||||
is_args INTEGER DEFAULT (0),
|
||||
script_type INTEGER DEFAULT (0),
|
||||
ps VARCHAR,
|
||||
create_time INTEGER DEFAULT (0)
|
||||
)
|
||||
'''
|
||||
sql_obj.execute(sql,())
|
||||
sql_obj.execute('CREATE INDEX type_id ON scripts (type_id);',())
|
||||
sql_obj.execute('CREATE INDEX is_baota ON scripts (is_baota);',())
|
||||
sql_obj.execute('CREATE INDEX status ON scripts (status);',())
|
||||
|
||||
sql = '''CREATE TABLE "trigger" (
|
||||
trigger_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name VARCHAR,
|
||||
status INTEGER DEFAULT (1),
|
||||
script_id INTEGER DEFAULT (0),
|
||||
script_body TEXT,
|
||||
cycle_type VARCHAR,
|
||||
cycle_where VARCHAR,
|
||||
cycle_hour INTEGER DEFAULT (0),
|
||||
cycle_minute INTEGER DEFAULT (0),
|
||||
ps VARCHAR,
|
||||
create_time INTEGER DEFAULT (0)
|
||||
)
|
||||
'''
|
||||
|
||||
sql_obj.execute(sql,())
|
||||
sql_obj.execute('CREATE INDEX t_status ON "trigger" (status)',())
|
||||
|
||||
sql = '''
|
||||
CREATE TABLE types (
|
||||
type_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
name VARCHAR, title VARCHAR
|
||||
)
|
||||
'''
|
||||
sql_obj.execute(sql,())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (1, 'Service Management', 'Scripts used to manage service status, such as detecting service status, stopping, reloading, starting, etc')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (2, 'Process Monitor', 'Script used to monitor process status, such as process survival, overhead, etc')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (3, 'Alarm Notification', 'Script for sending various alarm notifications')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (4, 'Load Monitoring', 'Scripts used to monitor system load, such as CPU, memory, network, etc')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (5, 'Website Monitoring', 'Script used to monitor website status, such as specifying URL address access status, content, etc')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (6, 'Other', 'Comprehensive scripts for various purposes, such as system upgrades, panel upgrades, software upgrades, etc')",())
|
||||
sql_obj.execute("INSERT INTO types (type_id, name, title) VALUES (7, 'Custom', 'User defined script');",())
|
||||
|
||||
sql = '''CREATE TABLE tasks (
|
||||
log_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
script_id INTEGER DEFAULT (0),
|
||||
trigger_id INTEGER DEFAULT (0),
|
||||
where_id INTEGER DEFAULT (0),
|
||||
status INTEGER DEFAULT (0),
|
||||
result_succ VARCHAR,
|
||||
result_err VARCHAR,
|
||||
start_time INTEGER DEFAULT (0),
|
||||
end_time INTEGER DEFAULT (0)
|
||||
)
|
||||
'''
|
||||
sql_obj.execute("CREATE INDEX sid ON tasks (script_id)",())
|
||||
sql_obj.execute("CREATE INDEX t_id ON tasks (trigger_id)",())
|
||||
sql_obj.execute("CREATE INDEX wid ON tasks (where_id)",())
|
||||
sql_obj.execute("CREATE INDEX stat ON tasks (status)",())
|
||||
sql_obj.execute(sql,())
|
||||
sql_obj.close()
|
||||
|
||||
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))))
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
1062
class_v2/crontabModelV2/triggerModel.py
Normal file
1062
class_v2/crontabModelV2/triggerModel.py
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user