55import celery
66import celery_aio_pool
77
8- from backend .app .task .conf import task_settings
98from backend .core .conf import settings
109
1110__all__ = ['celery_app' ]
1211
1312
1413def get_broker_url () -> str :
1514 """获取消息代理 URL"""
16- if task_settings .CELERY_BROKER == 'redis' :
15+ if settings .CELERY_BROKER == 'redis' :
1716 return (
1817 f'redis://:{ settings .REDIS_PASSWORD } @{ settings .REDIS_HOST } :'
19- f'{ settings .REDIS_PORT } /{ task_settings .CELERY_BROKER_REDIS_DATABASE } '
18+ f'{ settings .REDIS_PORT } /{ settings .CELERY_BROKER_REDIS_DATABASE } '
2019 )
2120 return (
22- f'amqp://{ task_settings . RABBITMQ_USERNAME } :{ task_settings . RABBITMQ_PASSWORD } @'
23- f'{ task_settings . RABBITMQ_HOST } :{ task_settings . RABBITMQ_PORT } '
21+ f'amqp://{ settings . CELERY_RABBITMQ_USERNAME } :{ settings . CELERY_RABBITMQ_PASSWORD } @'
22+ f'{ settings . CELERY_RABBITMQ_HOST } :{ settings . CELERY_RABBITMQ_PORT } '
2423 )
2524
2625
2726def get_result_backend () -> str :
2827 """获取结果后端 URL"""
2928 return (
3029 f'redis://:{ settings .REDIS_PASSWORD } @{ settings .REDIS_HOST } :'
31- f'{ settings .REDIS_PORT } /{ task_settings .CELERY_BACKEND_REDIS_DATABASE } '
30+ f'{ settings .REDIS_PORT } /{ settings .CELERY_BACKEND_REDIS_DATABASE } '
3231 )
3332
3433
3534def get_result_backend_transport_options () -> dict [str , Any ]:
3635 """获取结果后端传输选项"""
3736 return {
38- 'global_keyprefix' : task_settings .CELERY_BACKEND_REDIS_PREFIX ,
37+ 'global_keyprefix' : settings .CELERY_BACKEND_REDIS_PREFIX ,
3938 'retry_policy' : {
40- 'timeout' : task_settings .CELERY_BACKEND_REDIS_TIMEOUT ,
39+ 'timeout' : settings .CELERY_BACKEND_REDIS_TIMEOUT ,
4140 },
4241 }
4342
@@ -55,7 +54,7 @@ def init_celery() -> celery.Celery:
5554 'fba_celery' ,
5655 enable_utc = False ,
5756 timezone = settings .DATETIME_TIMEZONE ,
58- beat_schedule = task_settings .CELERY_SCHEDULE ,
57+ beat_schedule = settings .CELERY_SCHEDULE ,
5958 broker_url = get_broker_url (),
6059 broker_connection_retry_on_startup = True ,
6160 result_backend = get_result_backend (),
@@ -65,7 +64,7 @@ def init_celery() -> celery.Celery:
6564 )
6665
6766 # 自动发现任务
68- app .autodiscover_tasks (task_settings .CELERY_TASK_PACKAGES )
67+ app .autodiscover_tasks (settings .CELERY_TASK_PACKAGES )
6968
7069 return app
7170
0 commit comments