|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler |
|
|
from apscheduler.triggers.cron import CronTrigger |
|
|
import pytz |
|
|
from core.config import get_settings |
|
|
|
|
|
class SchedulerManager: |
|
|
def __init__(self): |
|
|
self.scheduler = AsyncIOScheduler(timezone=pytz.timezone(get_settings().TIMEZONE)) |
|
|
self._configure_jobs() |
|
|
|
|
|
def _configure_jobs(self): |
|
|
"""配置所有定时任务""" |
|
|
from task.scheduled_tasks import ( |
|
|
daily_task, |
|
|
weekly_report_task, |
|
|
data_cleanup_task |
|
|
) |
|
|
|
|
|
|
|
|
self.scheduler.add_job( |
|
|
daily_task, |
|
|
trigger=CronTrigger(hour='8', minute='13', timezone=pytz.timezone(get_settings().TIMEZONE)), |
|
|
id='daily_task' |
|
|
) |
|
|
|
|
|
self.scheduler.add_job( |
|
|
daily_task, |
|
|
trigger=CronTrigger(hour='9', minute='13', timezone=pytz.timezone(get_settings().TIMEZONE)), |
|
|
id='daily_task_1' |
|
|
) |
|
|
|
|
|
self.scheduler.add_job( |
|
|
weekly_report_task, |
|
|
trigger=CronTrigger(day_of_week='mon', hour='9', timezone=pytz.timezone(get_settings().TIMEZONE)), |
|
|
id='weekly_report' |
|
|
) |
|
|
|
|
|
self.scheduler.add_job( |
|
|
data_cleanup_task, |
|
|
trigger=CronTrigger(hour='2', timezone=pytz.timezone(get_settings().TIMEZONE)), |
|
|
id='data_cleanup' |
|
|
) |
|
|
|
|
|
async def start(self): |
|
|
"""异步启动调度器""" |
|
|
self.scheduler.start() |
|
|
|
|
|
async def shutdown(self): |
|
|
"""异步关闭调度器""" |
|
|
self.scheduler.shutdown() |
|
|
|
|
|
def get_jobs(self): |
|
|
"""获取所有任务""" |
|
|
return self.scheduler.get_jobs() |
|
|
|
|
|
def add_job(self, func, trigger, **kwargs): |
|
|
"""添加新任务""" |
|
|
return self.scheduler.add_job(func, trigger, **kwargs) |
|
|
|
|
|
def remove_job(self, job_id): |
|
|
"""移除任务""" |
|
|
self.scheduler.remove_job(job_id) |
|
|
|
|
|
|
|
|
|
