mirror of https://github.com/midoks/mdserver-web
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.
133 lines
3.2 KiB
133 lines
3.2 KiB
# coding:utf-8
|
|
|
|
import sys
|
|
import io
|
|
import os
|
|
import time
|
|
import json
|
|
|
|
|
|
web_dir = os.getcwd() + "/web"
|
|
if os.path.exists(web_dir):
|
|
sys.path.append(web_dir)
|
|
os.chdir(web_dir)
|
|
|
|
import core.mw as mw
|
|
from utils.crontab import crontab as MwCrontab
|
|
|
|
|
|
app_debug = False
|
|
if mw.isAppleSystem():
|
|
app_debug = True
|
|
|
|
|
|
def getPluginName():
|
|
return 'webstats'
|
|
|
|
|
|
def getPluginDir():
|
|
return mw.getPluginDir() + '/' + getPluginName()
|
|
|
|
|
|
def getServerDir():
|
|
return mw.getServerDir() + '/' + getPluginName()
|
|
|
|
|
|
def getTaskConf():
|
|
conf = getServerDir() + "/task_config.json"
|
|
return conf
|
|
|
|
|
|
def getConfigData():
|
|
try:
|
|
return json.loads(mw.readFile(getTaskConf()))
|
|
except:
|
|
pass
|
|
return {
|
|
"task_id": -1,
|
|
"task_list": ["migrate_hot_logs"],
|
|
"default_execute_hour": 3,
|
|
"default_execute_minute": 15,
|
|
}
|
|
|
|
|
|
def createBgTask():
|
|
cfg = getConfigData()
|
|
name = "[勿删]网站统计插件定时任务"
|
|
res = mw.M("crontab").field("id, name").where("name=?", (name,)).find()
|
|
if res:
|
|
return True
|
|
|
|
if "task_id" in cfg.keys() and cfg["task_id"] > 0:
|
|
res = mw.M("crontab").field("id, name").where("id=?", (cfg["task_id"],)).find()
|
|
if res and res["id"] == cfg["task_id"]:
|
|
print("计划任务已经存在!")
|
|
return True
|
|
|
|
|
|
cmd = "cd " + mw.getPanelDir() + " && nice -n 10 python3 " + getPluginDir() + "/tool_task.py execute"
|
|
params = {
|
|
'name': name,
|
|
'type': 'day',
|
|
'week': "",
|
|
'where1': "",
|
|
'hour': cfg['default_execute_hour'],
|
|
'minute': cfg['default_execute_minute'],
|
|
'save': "",
|
|
'backup_to': "",
|
|
'stype': "toShell",
|
|
'sname': '',
|
|
'sbody': cmd,
|
|
'url_address': '',
|
|
}
|
|
|
|
task_id = MwCrontab.instance().add(params)
|
|
if task_id > 0:
|
|
cfg["task_id"] = task_id
|
|
mw.writeFile(getTaskConf(), json.dumps(cfg))
|
|
|
|
|
|
def removeBgTask():
|
|
cfg = getConfigData()
|
|
if "task_id" in cfg.keys() and cfg["task_id"] > 0:
|
|
res = mw.M("crontab").field("id, name").where(
|
|
"id=?", (cfg["task_id"],)).find()
|
|
if res and res["id"] == cfg["task_id"]:
|
|
data = MwCrontab.instance().delete(cfg["task_id"])
|
|
if data[0]:
|
|
# print(data[1])
|
|
cfg["task_id"] = -1
|
|
mw.writeFile(getTaskConf(), json.dumps(cfg))
|
|
return True
|
|
return False
|
|
|
|
|
|
def execute():
|
|
try:
|
|
import time
|
|
now = time.strftime("%Y-%m-%d", time.localtime())
|
|
print("-" * 30)
|
|
cfg = getConfigData()
|
|
task_list = cfg["task_list"]
|
|
for task in task_list:
|
|
# print(task)
|
|
if task == "migrate_hot_logs":
|
|
try:
|
|
import tool_migrate
|
|
tool_migrate.migrateHotLogs("yesterday")
|
|
except Exception as e:
|
|
print(e)
|
|
print(now)
|
|
print("-" * 30)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
action = sys.argv[1]
|
|
if action == "execute":
|
|
execute()
|
|
elif action == "remove":
|
|
removeBgTask()
|
|
elif action == "add":
|
|
createBgTask()
|
|
|