29 lines
633 B
Python
29 lines
633 B
Python
|
|
import sqlite3
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from celery import Celery, platforms
|
||
|
|
from celery.schedules import crontab
|
||
|
|
|
||
|
|
from config import Config
|
||
|
|
from main import Detector
|
||
|
|
|
||
|
|
platforms.C_FORCE_ROOT = True
|
||
|
|
celery_db = f"{Config.name}_scheduler.db"
|
||
|
|
sqlite3.connect(celery_db)
|
||
|
|
app_run = Celery(Config.name, broker=f"sqla+sqlite:///{celery_db}")
|
||
|
|
|
||
|
|
app_run.conf.update(
|
||
|
|
beat_schedule={
|
||
|
|
"定时检测": {
|
||
|
|
"task": "scheduler.run_check",
|
||
|
|
# "schedule": crontab(hour=9, minute=00),
|
||
|
|
"schedule": timedelta(seconds=60),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@app_run.task()
|
||
|
|
def run_check():
|
||
|
|
Detector().get_data()
|