scheduler.py 1.03 KB
Newer Older
1 2 3 4 5 6 7
import datetime
from apscheduler.schedulers.background import BackgroundScheduler

class TaskScheduler:
    def __init__(self):
        self.scheduler = BackgroundScheduler()

文靖昊 committed
8
    def add_task(self,timedTask,timeMinute,*args):
9
        # Define your timed task here
文靖昊 committed
10
        self.scheduler.add_job(timedTask, 'interval', minutes=timeMinute,args=args)
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

    def start_scheduler(self):
        # Add the timed task to the scheduler
        # Start the scheduler
        self.scheduler.start()

    def stop_scheduler(self):
        # Shutdown the scheduler
        self.scheduler.shutdown()

def task1():
    print(datetime.datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3])

def task2():
    print("task2")
# Example usage:
if __name__ == "__main__":
    scheduler = TaskScheduler()
    scheduler.add_task(task1,1)
    scheduler.add_task(task2,2)
    scheduler.start_scheduler()
    # Keep the program running to allow scheduler to execute tasks
    try:
        while True:
            pass
    except KeyboardInterrupt:
        scheduler.stop_scheduler()