alien_py_cloud/alien_util/celery_app.py

from celery import Celery
from celery.schedules import crontab
from alien_gateway.config import settings

# 创建 Celery 应用
celery_app = Celery(
    "alien_cloud",
    broker=settings.REDIS_SENTINEL_URL,
    backend=settings.REDIS_SENTINEL_URL,
    include=["alien_util.tasks.contract_tasks"]
)

# Celery 配置
celery_app.conf.update(
    task_serializer="json",
    accept_content=["json"],
    result_serializer="json",
    timezone="Asia/Shanghai",
    enable_utc=True,
    broker_transport_options={
        "master_name": settings.REDIS_MASTER_NAME,
        "sentinel_kwargs": {
            **({"username": settings.REDIS_SENTINEL_USERNAME} if settings.REDIS_SENTINEL_USERNAME else {}),
            **({"password": settings.REDIS_SENTINEL_PASSWORD} if settings.REDIS_SENTINEL_PASSWORD else {}),
        },
        "password": settings.REDIS_PASSWORD,
        "db": settings.REDIS_DB,
    },
    result_backend_transport_options={
        "master_name": settings.REDIS_MASTER_NAME,
        "sentinel_kwargs": {
            **({"username": settings.REDIS_SENTINEL_USERNAME} if settings.REDIS_SENTINEL_USERNAME else {}),
            **({"password": settings.REDIS_SENTINEL_PASSWORD} if settings.REDIS_SENTINEL_PASSWORD else {}),
        },
        "password": settings.REDIS_PASSWORD,
        "db": settings.REDIS_DB,
    },
    # 定时任务配置
    beat_schedule={
        "check-contract-expiry": {
            "task": "alien_util.tasks.contract_tasks.check_contract_expiry",
            "schedule": crontab(hour=0, minute=1),  # 每天凌晨0点1分执行
        },
    },
)