29 lines
803 B
Python
29 lines
803 B
Python
# myproject/scheduler.py
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
from apscheduler.triggers.cron import CronTrigger
|
|
import logging
|
|
from subscription_app.tasks import generate_billing
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 初始化调度器
|
|
scheduler = BackgroundScheduler()
|
|
|
|
def start():
|
|
if not scheduler.running:
|
|
# 添加任务（例如每隔 5 分钟运行一次任务）
|
|
scheduler.add_job(
|
|
generate_billing,
|
|
trigger=CronTrigger(second='*/5'), # 每 5 分钟执行一次
|
|
id='generate_billing task', # 给任务一个唯一 ID
|
|
replace_existing=True,
|
|
)
|
|
logger.info("Scheduler started")
|
|
scheduler.start()
|
|
|
|
def stop():
|
|
logger.info("Scheduler stopped")
|
|
scheduler.shutdown()
|
|
|
|
|