midoks
/
mdserver-web
mirror of https://github.com/midoks/mdserver-web
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
Simple Linux Panel
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11377 Commits
8 Branches
70 Tags
40 MiB
 
 
 
 
 
 
Tag: Branch: Tree: d4cfd90ccd
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'd4cfd90ccd'
${ noResults }
mdserver-web/web/utils/crontab.py

430 lines
15 KiB
Raw Blame History

# coding:utf-8
# ---------------------------------------------------------------------------------
# MW-Linux面板
# ---------------------------------------------------------------------------------
# copyright (c) 2018-∞(https://github.com/midoks/mdserver-web) All rights reserved.
# ---------------------------------------------------------------------------------
# Author: midoks <midoks@163.com>
# ---------------------------------------------------------------------------------
import os
import sys
import time
import json
import threading
import multiprocessing
from admin import model
import core.mw as mw
import thisdb
class crontab(object):
# lock
_instance_lock = threading.Lock()
@classmethod
def instance(cls, *args, **kwargs):
if not hasattr(crontab, "_instance"):
with crontab._instance_lock:
if not hasattr(crontab, "_instance"):
crontab._instance = crontab(*args, **kwargs)
return crontab._instance
# 获取指定任务数据
def getCrondFind(self, cron_id):
return thisdb.getCrond(cron_id)
def add(self, data):
if len(data['name']) < 1:
return mw.returnData(False, '任务名称不能为空!')
is_check_pass, msg = self.cronCheck(data)
if not is_check_pass:
return mw.returnData(is_check_pass, msg)
cmd, title = self.getCrondCycle(data)
cron_path = mw.getServerDir() + '/cron'
cron_name = self.getShell(data)
cmd += ' ' + cron_path + '/' + cron_name + ' >> ' + cron_path + '/' + cron_name + '.log 2>&1'
if not mw.isAppleSystem():
data = self.writeShell(cmd)
if not data['status']:
return data
self.crondReload()
add_dbdata = {}
add_dbdata['name'] = data['name']
add_dbdata['type'] = data['type']
add_dbdata['where1'] = data['where1']
add_dbdata['where_hour'] = data['hour']
add_dbdata['where_minute'] = data['minute']
add_dbdata['save'] = data['save']
add_dbdata['backup_to'] = data['backup_to']
add_dbdata['sname'] = data['sname']
add_dbdata['sbody'] = data['sbody']
add_dbdata['stype'] = data['stype']
add_dbdata['echo'] = cron_name
add_dbdata['url_address'] = data['url_address']
tid = thisdb.addCrontab(add_dbdata)
if tid > 0:
return mw.returnData(True, '添加成功')
return mw.returnData(False, '添加失败')
def delete(self, tid):
data = thisdb.getCrond(tid)
if not self.removeForCrond(data['echo']):
return mw.returnData(False, '无法写入文件，请检查是否开启了系统加固功能!')
cron_path = mw.getServerDir() + '/cron'
cron_file = cron_path + '/' + data['echo']
if os.path.exists(cron_file):
os.remove(cron_file)
cron_file = cron_path + '/' + data['echo'] + '.log'
if os.path.exists(cron_file):
os.remove(cron_file)
thisdb.deleteCronById(tid)
msg = mw.getInfo('删除计划任务[{1}]成功!', (data['name'],))
mw.writeLog('计划任务', msg)
return mw.returnData(True, msg)
def getCrontabHuman(self, data):
rdata = []
for i in range(len(data)):
t = data[i]
if t['type'] == "day":
t['type'] = '每天'
t['cycle'] = mw.getInfo('每天, {1}点{2}分 执行', (str(t['where_hour']), str(t['where_minute'])))
elif t['type'] == "day-n":
t['type'] = mw.getInfo('每{1}天', (str(t['where1']),))
t['cycle'] = mw.getInfo('每隔{1}天, {2}点{3}分 执行', (str(t['where1']), str(t['where_hour']), str(t['where_minute'])))
elif t['type'] == "hour":
t['type'] = '每小时'
t['cycle'] = mw.getInfo('每小时, 第{1}分钟 执行', (str(t['where_minute']),))
elif t['type'] == "hour-n":
t['type'] = mw.getInfo('每{1}小时', (str(t['where1']),))
t['cycle'] = mw.getInfo('每{1}小时, 第{2}分钟 执行', (str(t['where1']), str(t['where_minute'])))
elif t['type'] == "minute-n":
t['type'] = mw.getInfo('每{1}分钟', (str(t['where1']),))
t['cycle'] = mw.getInfo('每隔{1}分钟执行', (str(t['where1']),))
elif t['type'] == "week":
t['type'] = '每周'
if not t['where1']:
t['where1'] = '0'
t['cycle'] = mw.getInfo('每周{1}, {2}点{3}分执行', (self.toWeek(int(t['where1'])), str(t['where_hour']), str(t['where_minute'])))
elif t['type'] == "month":
t['type'] = '每月'
t['cycle'] = mw.getInfo('每月, {1}日 {2}点{3}分执行', (str(t['where1']), str(t['where_hour']), str(t['where_minute'])))
rdata.append(t)
return rdata
# 从crond删除
def removeForCrond(self, echo):
if mw.isAppleSystem():
return True
u_file = '/var/spool/cron/crontabs/root'
if not os.path.exists(u_file):
file = '/var/spool/cron/root'
if not os.path.exists(file):
return False
else:
file = u_file
conf = mw.readFile(file)
rep = ".+" + str(echo) + ".+\n"
conf = re.sub(rep, "", conf)
if not mw.writeFile(file, conf):
return False
self.crondReload()
return True
def getCrontabList(self,
page:int | None = 1,
size:int | None = 10
):
info = thisdb.getCrontabList(page=int(page),size=int(size))
rdata = {}
rdata['data'] = self.getCrontabHuman(info['list'])
rdata['page'] = mw.getPage({'count':info['count'],'tojs':'getCronData','p':page,'row':size})
return rdata
def getCrondCycle(self, params):
cron_cmd = ''
title = ''
if params['type'] == "day":
cron_cmd = self.getDay(params)
title = '每天'
elif params['type'] == "day-n":
cron_cmd = self.getDay_N(params)
title = mw.getInfo('每{1}天', (params['where1'],))
elif params['type'] == "hour":
cron_cmd = self.getHour(params)
title = '每小时'
elif params['type'] == "hour-n":
cron_cmd = self.getHour_N(params)
title = '每小时'
elif params['type'] == "minute-n":
cron_cmd = self.minute_N(params)
elif params['type'] == "week":
params['where1'] = params['week']
cron_cmd = self.week(params)
elif params['type'] == "month":
cron_cmd = self.month(params)
return cron_cmd, title
# 转换大写星期
def toWeek(self, num):
wheres = {
0: '日',
1: '一',
2: '二',
3: '三',
4: '四',
5: '五',
6: '六'
}
try:
return wheres[num]
except:
return ''
# 取任务构造Day
def getDay(self, param):
cmd = "{0} {1} * * * ".format(param['minute'], param['hour'])
return cmd
# 取任务构造Day_n
def getDay_N(self, param):
cmd = "{0} {1} */{2} * * ".format(param['minute'], param['hour'], param['where1'])
return cmd
# 取任务构造Hour
def getHour(self, param):
cmd = "{0} * * * * ".format(param['minute'])
return cmd
# 取任务构造Hour-N
def getHour_N(self, param):
cmd = "{0} */{1} * * * ".format(param['minute'], param['where1'])
return cmd
# 取任务构造Minute-N
def minute_N(self, param):
cmd = "*/{0} * * * * ".format(param['where1'])
return cmd
# 取任务构造week
def week(self, param):
cmd = "{0} {1} * * {2}".format(param['minute'], param['hour'], param['week'])
return cmd
# 取任务构造Month
def month(self, param):
cmd = "{0} {1} {2} * * ".format(param['minute'], param['hour'], param['where1'])
return cmd
# 参数校验
def cronCheck(self, params):
if params['stype'] == 'site' or params['stype'] == 'database' or params['stype'].find('database_') > -1 or params['stype'] == 'logs' or params['stype'] == 'path':
if params['save'] == '':
return False, '保留份数不能为空!'
if params['type'] == 'day':
if params['hour'] == '':
return False, '小时不能为空!'
if params['minute'] == '':
return False, '分钟不能为空!'
if params['type'] == 'day-n':
if params['where1'] == '':
return False, '天不能为空!'
if params['hour'] == '':
return False, '小时不能为空!'
if params['minute'] == '':
return False, '分钟不能为空!'
if params['type'] == 'hour':
if params['minute'] == '':
return False, '分钟不能为空!'
if params['type'] == 'hour-n':
if params['where1'] == '':
return False, '小时不能为空!'
if params['minute'] == '':
return False, '分钟不能为空!'
if params['type'] == 'minute-n':
if params['where1'] == '':
return False, '分钟不能为空!'
if params['type'] == 'week':
if params['hour'] == '':
return False, '小时不能为空!'
if params['minute'] == '':
return False, '分钟不能为空!'
if params['type'] == 'month':
if params['where1'] == '':
return False, '日不能为空!'
if params['hour'] == '':
return False, '小时不能为空!'
if params['minute'] == '':
return False, '分钟不能为空!'
return True, 'OK'
# 取执行脚本
def getShell(self, param):
# try:
stype = param['stype']
if stype == 'toFile':
shell = param.sFile
else:
head = "#!/bin/bash\nPATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin\nexport PATH\n"
start_head = '''
SCRIPT_RUN_TIME="0s"
MW_ToSeconds()
{
SEC=$1
if [ $SEC -lt 60 ]; then
SCRIPT_RUN_TIME="${SEC}s"
elif [ $SEC -ge 60 ] && [ $SEC -lt 3600 ];then
SCRIPT_RUN_TIME="$(( SEC / 60 ))m$(( SEC % 60 ))s"
elif [ $SEC -ge 3600 ]; then
SCRIPT_RUN_TIME="$(( SEC / 3600 ))h$(( (SEC % 3600) / 60 ))m$(( (SEC % 3600) % 60 ))s"
fi
}
START_MW_SHELL_TIME=`date +%s`
'''
source_bin_activate = '''
export LANG=en_US.UTF-8
MW_PATH=%s/bin/activate
if [ -f $MW_PATH ];then
source $MW_PATH
fi''' % (mw.getPanelDir(),)
head = head + start_head + source_bin_activate + "\n"
log = '.log'
#所有
if param['sname'] == 'ALL':
log = ''
script_dir = mw.getPanelDir() + "/scripts"
source_stype = 'database'
if stype.find('database_') > -1:
plugin_name = stype.replace('database_', '')
script_dir = mw.getPanelDir() + "/plugins/" + plugin_name + "/scripts"
source_stype = stype
stype = 'database'
wheres = {
'path': head + "python3 " + script_dir + "/backup.py path " + param['sname'] + " " + str(param['save']),
'site': head + "python3 " + script_dir + "/backup.py site " + param['sname'] + " " + str(param['save']),
'database': head + "python3 " + script_dir + "/backup.py database " + param['sname'] + " " + str(param['save']),
'logs': head + "python3 " + script_dir + "/logs_backup.py " + param['sname'] + log + " " + str(param['save']),
'rememory': head + "/bin/bash " + script_dir + '/rememory.sh'
}
if param['backup_to'] != 'localhost':
cfile = mw.getPluginDir() + "/" + \
param['backup_to'] + "/index.py"
wheres['path'] = head + "python3 " + cfile + \
" path " + param['sname'] + " " + str(param['save'])
wheres['site'] = head + "python3 " + cfile + \
" site " + param['sname'] + " " + str(param['save'])
wheres['database'] = head + "python3 " + cfile + " " + \
source_stype + " " + \
param['sname'] + " " + str(param['save'])
try:
shell = wheres[stype]
except:
if stype == 'toUrl':
shell = head + "curl -sS --connect-timeout 10 -m 60 '" + \
param['urladdress'] + "'"
else:
shell = head + param['sbody'].replace("\r\n", "\n")
shell += '''
echo "----------------------------------------------------------------------------"
endDate=`date +"%Y-%m-%d %H:%M:%S"`
END_MW_SHELL_TIME=`date +"%s"`
((SHELL_COS_TIME=($END_MW_SHELL_TIME-$START_MW_SHELL_TIME)))
MW_ToSeconds $SHELL_COS_TIME
echo "★[$endDate] Successful | Script Run [$SCRIPT_RUN_TIME] "
echo "----------------------------------------------------------------------------"
'''
cron_path = mw.getServerDir() + '/cron'
if not os.path.exists(cron_path):
mw.execShell('mkdir -p ' + cron_path)
if not 'echo' in param:
cron_name = mw.md5(mw.md5(str(time.time()) + '_mw'))
else:
cron_name = param['echo']
file = cron_path + '/' + cron_name
mw.writeFile(file, self.checkScript(shell))
mw.execShell('chmod 750 ' + file)
return cron_name
# 检查脚本
def checkScript(self, shell):
keys = ['shutdown', 'init 0', 'mkfs', 'passwd',
'chpasswd', '--stdin', 'mkfs.ext', 'mke2fs']
for key in keys:
shell = shell.replace(key, '[***]')
return shell
# 将Shell脚本写到文件
def writeShell(self, config):
file = '/var/spool/cron/crontabs/root'
current_os = mw.getOs()
if current_os == 'darwin':
file = '/etc/crontab'
elif current_os.startswith("freebsd"):
file = '/var/cron/tabs/root'
if not os.path.exists(file):
file = '/var/spool/cron/root'
if not os.path.exists(file):
mw.writeFile(file, '')
conf = mw.readFile(file)
conf += str(config) + "\n"
if mw.writeFile(file, conf):
if not os.path.exists(file):
mw.execShell("chmod 600 '" + file +"' && chown root.root " + file)
else:
mw.execShell("chmod 600 '" + file +"' && chown root.crontab " + file)
return mw.returnData(True, 'ok')
return mw.returnData(False, '文件写入失败,请检查是否开启系统加固功能!')
# 重载配置
def crondReload(self):
if mw.isAppleSystem():
if os.path.exists('/etc/crontab'):
pass
else:
if os.path.exists('/etc/init.d/crond'):
mw.execShell('/etc/init.d/crond reload')
elif os.path.exists('/etc/init.d/cron'):
mw.execShell('service cron restart')
else:
mw.execShell("systemctl reload crond")