forked from cksgf/ServerManagement
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtask.py
181 lines (177 loc) · 7.82 KB
/
task.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
import datetime
import threading
import subprocess
import time
from index import sql
import os
import json
class taskset():
def __init__(self):
self.taskList=[]
self.maxSetTime = 2678400
lastTask = sql.selectTask()
if lastTask[0]:
for i in lastTask[1]:
self.CreatTask(json.loads(i[0]),writeToSql = False)
else:
pass
'''
{'type':'day',
'hour':'12',
'mint':'30',
'senc':'15',
'creatTime'：'2018-12-3 19:48',
'taskID':str(time.time()+random.random()),
'nextRunTime':time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()+int(interval))),
'value'："echo 666"
},
{'type':'week',
'week':'0', #周日为0
'hour':'12',
'mint':'30',
'senc':'15',
'creatTime'：'2018-12-3 19:48',
'taskID':str(time.time()+random.random()),
'nextRunTime':time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()+int(interval))),
'value'："echo 666"
},
{'type':'month',
'day':'7',
'hour':'12',
'mint':'30',
'senc':'15',
'creatTime'：'2018-12-3 19:48',
'taskID':str(time.time()+random.random()),
'nextRunTime':time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()+int(interval))),
'value'："echo 666"
}
'''
def TaskFunc(self,data,delete = False):
if data not in self.taskList:
return True
#检测是否超出最大设定时间,并作出相关的处理
if data['needCheck'] == 'T':
interval = self.GetNextTaskSenc(data)
if interval >= self.maxSetTime:
timer = threading.Timer(self.maxSetTime, self.TaskFunc,(data,))
timer.start()
return True
else:
self.taskList.remove(data)
data['needCheck'] = 'F'
self.taskList.append(data)
if data['type'] == 'once':
timer = threading.Timer(interval, self.TaskFunc,(data,True))
else:
timer = threading.Timer(interval, self.TaskFunc,(data,))
timer.start()
return True
nowTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
if not delete :
self.taskList.remove(data)
interval = self.GetNextTaskSenc(data)
data['nextRunTime'] = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()+int(interval)))
if interval >= self.maxSetTime:
data['needCheck'] = 'T'
interval = self.maxSetTime
else:
data['needCheck'] = 'F'
timer = threading.Timer(interval, self.TaskFunc,(data,))
timer.start()
self.taskList.append(data)
else:
self.DeleteTask(data['taskID'])
logname = 'lib/tasklog/'+data['creatTime'].replace(':','_')+'.log'
with open(logname,'a') as f:
f.write('-'*20+'\n'+nowTime+':\n')
subprocess.Popen(data['value'],shell=True,stdout = open(logname,'a'),stderr = subprocess.STDOUT)
def CreatTask(self,data,writeToSql=True):
interval = self.GetNextTaskSenc(data)
data['nextRunTime'] = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()+int(interval)))
if interval >= self.maxSetTime:
data['needCheck'] = 'T'
interval = self.maxSetTime
else:
data['needCheck'] = 'F'
self.taskList.append(data)
if writeToSql:
sql.insertTask(data)
logname = 'lib/tasklog/'+data['creatTime'].replace(':','_')+'.log'
if not os.path.isfile(logname):
with open(logname,'w') as f:
if data['type'] == 'day':
t = '每天的%s:%s:%s' %(data['hour'],data['mint'],data['senc'])
elif data['type'] == 'month':
t = '每月%s号的%s:%s:%s' %(data['day'],data['hour'],data['mint'],data['senc'])
elif data['type'] == 'week':
t = '每周%s的%s:%s:%s' %(data['week'],data['hour'],data['mint'],data['senc'])
elif data['type'] == 'senc':
t = '每间隔%s秒' %data['senc']
elif data['type'] == 'once':
t = '%s年%s月%s日,%s时%s分%s秒' %(data['year'],data['month'],data['day'],data['hour'],data['mint'],data['senc'])
f.write('计划任务执行日志\n任务创建时间:%s\n计划类型:%s\nSHELL内容:%s\n'%(data['creatTime'],t,data['value']))
timer = threading.Timer(interval, self.TaskFunc,(data,))
timer.start()
def GetTaskList(self):
return self.taskList
def DeleteTask(self,taskID):
for i in self.taskList:
if i['taskID'] == taskID:
self.taskList.remove(i)
sql.deleteTask(i['taskID'])
def GetNextTaskSenc(self,data):
if data['type'] == 'senc':
return int(data['senc'])
#设定周期不为秒的话,计算出下一次执行是几天之后
elif data['type'] == 'day' :
now_time = datetime.datetime.now()
next_time = now_time + datetime.timedelta(days=1)
elif data['type'] == 'week' :
if str(data['week']) not in list(str(i) for i in range(0,8)):
raise ValueError('日期设定错误,星期数值应在1-7内!')
now_time = datetime.datetime.now()
tip = 1 #从第二天计算,避免设定周几和今天相同,产生循环
while True:
next_time = now_time + datetime.timedelta(days=tip)
if next_time.strftime('%w') == data['week']:
break
else:
tip+=1
next_time = now_time + datetime.timedelta(days=tip)
elif data['type'] == 'month' :
if str(data['day']) not in list(str(i) for i in range(1,32)):
raise ValueError('日期设定错误,日期数值应在1-31内!')
now_time = datetime.datetime.now()
tip = 1
while True:
next_time = now_time + datetime.timedelta(days=tip)
if str(next_time.day) == str(data['day']):
break
else:
tip+=1
next_time = now_time + datetime.timedelta(days=tip)
elif data['type'] == 'once' :
if str(data['day']) not in list(str(i) for i in range(1,32)):
raise ValueError('日期设定错误,日期数值应在1-31内!')
now_time = datetime.datetime.now()
tip = 1
while True:
next_time = now_time + datetime.timedelta(days=tip)
if (str(next_time.year) == str(data['year'])) and (str(next_time.month) == str(data['month'])) and (str(next_time.day) == str(data['day'])):
break
else:
tip+=1
next_time = now_time + datetime.timedelta(days=tip)
else:
raise ValueError('无法解析下次执行的日期,请检查设定时间格式!')
#下次执行任务的时间
next_year = next_time.date().year
next_month = next_time.date().month
next_day = next_time.date().day
try:
#根据下次运行的时间,计算出秒数
next_time = datetime.datetime.strptime('%s-%s-%s %s:%s:%s' %(next_year,next_month,next_day,data["hour"],data["mint"],data["senc"]), "%Y-%m-%d %H:%M:%S")
timer_start_time = (next_time - now_time).total_seconds()
except :
raise ValueError('请检查时间格式!')
return (int(timer_start_time)+1 if (timer_start_time%1 > 0) else int(timer_start_time)) #向上取整,只有这一个需求,懒得用math.ceil