image.png

    1. import logging
    3. from crontab import CronTab
    4. from datetime import datetime
    5. import json
    7. from utils.log import log
    8. from utils.jsonfile import JsonFile
    11. class CronOpt:
    12.     #创建定时任务
    13.     def __init__(self,user="root"):
    14.         self.initialize(user)
    16.     #初始化
    17.     def initialize(self,user="root"):
    18.         self.cron = CronTab(user)
    20.     #查询列表
    21.     def select(self,reInit = False):
    22.         if reInit != False:
    23.             # 强制重新读取列表
    24.             self.initialize()
    25.         cronArray = []
    26.         for job in self.cron:
    27.             # print job.command
    28.             schedule = job.schedule(date_from=datetime.now())
    29.             cronDict = {
    30.                 "job"    : job,
    31.                 "task"   : (job.command).replace(r'>/dev/null 2>&1', ''),
    32.                 "next"   : str(schedule.get_next()),
    33.                 "prev"   : str(schedule.get_prev()),
    34.                 "comment": job.comment
    35.             }
    36.             cronArray.append(cronDict)
    37.         return cronArray
    39.     #新增定时任务
    40.     def add(self, command, timeStr, commentName):
    41.         # 创建任务
    42.         job = self.cron.new(command=command)
    43.         # 设置任务执行周期
    44.         job.setall(timeStr)
    45.         # 备注，也是命令id
    46.         job.set_comment(commentName)
    47.         # 写入到定时任务
    48.         self.cron.write()
    49.         # 返回更新后的列表
    50.         return self.select(True)
    52.     #删除定时任务
    53.     def delCron(self,commentName):
    54.         # 创建任务
    55.         for job in self.cron :
    56.             if job.comment == commentName:
    57.                 self.cron.remove(job)
    58.                 self.cron.write()
    59.         return self.select(True)
    61.     #更新定时任务
    62.     def update(self, command, timeStr, commentName):
    63.         # 先删除任务
    64.         self.delCron(commentName)
    65.         # 再创建任务,以达到更新的结果
    66.         return self.add(command, timeStr, commentName)
    68.     def start(self):
    69.         #     c = CronOpt()
    70.         #     print(c.select())
    71.         #     # log.info(json.loads(c.select()))
    72.         #     # 新增定时任务
    73.         #     #print(c.add("ls /home","*/10 * * * *","测试"))
    74.         #     # 删除定时任务
    75.         #     # print(c.delCron("测试"))
    76.         '''
    77.             思路：
    78.             读取配置文件
    79.             如果系统的定时任务中不存在配置文件中的定时任务id，则新增该定时任务
    80.             如果配置文件不存在系统中的定时任务id，则删除该任务
    81.             '''
    82.         c = CronOpt()
    83.         jsonfile = JsonFile()
    84.         contabjson = jsonfile.file_to_json("/home/zhanghui/script/lanhui/robot/mycrontab.json")
    85.         crontabjsondict = {}
    86.         for onecron in contabjson:
    87.             crontabjsondict[onecron["commentName"]] = onecron
    89.         selectdict = {}
    90.         for oneselect in c.select():
    91.             selectdict[oneselect["comment"]] = oneselect
    92.             if oneselect["comment"] not in crontabjsondict and oneselect["comment"] != "":
    93.                 # print("删除定时任务")
    94.                 log.info("删除定时任务:"+str(c.delCron(oneselect["comment"])))
    96.         for onecron in contabjson:
    97.             if onecron["commentName"] not in selectdict:
    98.                 # print("增加定时任务")
    99.                 log.info("增加定时任务:"+str(c.add(onecron["command"],onecron["timeStr"], onecron["commentName"])))