python異步實現(xiàn)定時任務和周期任務的方法
一. 如何調用
def f1(arg1, arg2): print('f1', arg1, arg2) def f2(arg1): print('f2', arg1) def f3(): print('f3') def f4(): print('周期任務', int(time.time())) timer = TaskTimer() # 把任務加入任務隊列 timer.join_task(f1, [1, 2], timing=15.5) # 每天15:30執(zhí)行 timer.join_task(f2, [3], timing=14) # 每天14:00執(zhí)行 timer.join_task(f3, [], timing=15) # 每天15:00執(zhí)行 timer.join_task(f4, [], interval=10) # 每10秒執(zhí)行1次 # 開始執(zhí)行(此時才會創(chuàng)建線程) timer.start()
f1~f4是我們需要定時執(zhí)行的函數(shù)。
首先創(chuàng)建TaskTimer對象(TaskTimer的代碼在下面)。調用join_task函數(shù),把需要執(zhí)行的函數(shù)加入到任務隊列。最后調用start,任務就開始執(zhí)行了。
join_task參數(shù):
fun:需要執(zhí)行的函數(shù)
arg:fun的參數(shù),如果沒有就傳一個空列表
interval:如果有此參數(shù),說明任務是周期任務,單位為秒(注意interval最少5秒)
timing:如果有此參數(shù),說明任務是定時任務,單位為時
注意:interval和timing只能選填1個
二. 源碼
import datetime import time from threading import Thread from time import sleep class TaskTimer: __instance = None def __new__(cls, *args, **kwargs): """ 單例模式 """ if not cls.__instance: cls.__instance = object.__new__(cls) return cls.__instance def __init__(self): if not hasattr(self, 'task_queue'): setattr(self, 'task_queue', []) if not hasattr(self, 'is_running'): setattr(self, 'is_running', False) def write_log(self, level, msg): cur_time = datetime.datetime.now() with open('./task.log', mode='a+', encoding='utf8') as file: s = "[" + str(cur_time) + "][" + level + "] " + msg print(s) file.write(s + "\n") def work(self): """ 處理任務隊列 """ while True: for task in self.task_queue: if task['interval']: self.cycle_task(task) elif task['timing']: self.timing_task(task) sleep(5) def cycle_task(self, task): """ 周期任務 """ if task['next_sec'] <= int(time.time()): try: task['fun'](*task['arg']) self.write_log("正常", "周期任務:" + task['fun'].__name__ + " 已執(zhí)行") except Exception as e: self.write_log("異常", "周期任務:" + task['fun'].__name__ + " 函數(shù)內部異常:" + str(e)) finally: task['next_sec'] = int(time.time()) + task['interval'] def timing_task(self, task): """ 定時任務 """ # 今天已過秒數(shù) today_sec = self.get_today_until_now() # 到了第二天,就重置任務狀態(tài) if task['today'] != self.get_today(): task['today'] = self.get_today() task['today_done'] = False # 第一次執(zhí)行 if task['first_work']: if today_sec >= task['task_sec']: task['today_done'] = True task['first_work'] = False else: task['first_work'] = False # 今天還沒有執(zhí)行 if not task['today_done']: if today_sec >= task['task_sec']: # 到點了,開始執(zhí)行任務 try: task['fun'](*task['arg']) self.write_log("正常", "定時任務:" + task['fun'].__name__ + " 已執(zhí)行") except Exception as e: self.write_log("異常", "定時任務:" + task['fun'].__name__ + " 函數(shù)內部異常:" + str(e)) finally: task['today_done'] = True if task['first_work']: task['first_work'] = False def get_today_until_now(self): """ 獲取今天凌晨到現(xiàn)在的秒數(shù) """ i = datetime.datetime.now() return i.hour * 3600 + i.minute * 60 + i.second def get_today(self): """ 獲取今天的日期 """ i = datetime.datetime.now() return i.day def join_task(self, fun, arg, interval=None, timing=None): """ interval和timing只能存在1個 :param fun: 你要調用的任務 :param arg: fun的參數(shù) :param interval: 周期任務,單位秒 :param timing: 定時任務,取值:[0,24) """ # 參數(shù)校驗 if (interval != None and timing != None) or (interval == None and timing == None): raise Exception('interval和timing只能選填1個') if timing and not 0 <= timing < 24: raise Exception('timing的取值范圍為[0,24)') if interval and interval < 5: raise Exception('interval最少為5') # 封裝一個task task = { 'fun': fun, 'arg': arg, 'interval': interval, 'timing': timing, } # 封裝周期或定時任務相應的參數(shù) if timing: task['task_sec'] = timing * 3600 task['today_done'] = False task['first_work'] = True task['today'] = self.get_today() elif interval: task['next_sec'] = int(time.time()) + interval # 把task加入任務隊列 self.task_queue.append(task) self.write_log("正常", "新增任務:" + fun.__name__) def start(self): """ 開始執(zhí)行任務 返回線程標識符 """ if not self.is_running: thread = Thread(target=self.work) thread.start() self.is_running = True self.write_log("正常", "TaskTimer已開始運行!") return thread.ident self.write_log("警告", "TaskTimer已運行,請勿重復啟動!")
以上這篇python異步實現(xiàn)定時任務和周期任務的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
python中@property和property函數(shù)常見使用方法示例
這篇文章主要介紹了python中@property和property函數(shù)常見使用方法,結合實例形式分析了Python @property和property函數(shù)功能、使用方法及相關操作注意事項,需要的朋友可以參考下2019-10-10python基礎入門學習筆記(Python環(huán)境搭建)
這篇文章主要介紹了python基礎入門學習筆記,這是開啟學習python基礎知識的第一篇,夯實Python基礎,才能走的更遠,感興趣的小伙伴們可以參考一下2016-01-01以SortedList為例詳解Python的defaultdict對象使用自定義類型的方法
這篇文章主要介紹了以SortedList為例詳解Python的defaultdict對象使用自定義類型的方法,文章圍繞主題展開詳細的內容介紹,具有一定的參考價值,需要的朋友可以參考一下2022-07-07