mirror of
https://github.com/aaPanel/aaPanel.git
synced 2026-08-17 21:25:47 +02:00
Jack
a24f2e23c6
1. Added Mail Marketing Automation trigger tasks 2. Added Mail Marketing Groups Import, Export, Merge 3. Added Mail Marketing Subscribers to support paste import 4. Added Mail Marketing Template Import Export Duplicate 5. Added Mail server add timed automatic reply 6. Added HTTPS Protection function for Website (disables automatic HTTP to HTTPS redirection when enabled) 7. Optimize Website Interface Button Integration 8. Optimize the response speed of WP Toolkit interface
420 lines
17 KiB
Python
420 lines
17 KiB
Python
#coding: utf-8
|
|
#-------------------------------------------------------------------
|
|
# aapanel
|
|
#-------------------------------------------------------------------
|
|
# Copyright (c) 2015-2099 aapanel(http://aapanel.com All rights reserved.
|
|
#-------------------------------------------------------------------
|
|
# Author: hwliang <hwl@aapanel.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 aapanel 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))))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|