52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from celery import Celery
|
|
|
|
from ..core.config import settings
|
|
|
|
|
|
def _broker_url() -> str:
|
|
return settings.celery.broker_url or settings.redis.url
|
|
|
|
|
|
def _result_backend() -> str:
|
|
return settings.celery.result_backend or settings.redis.url
|
|
|
|
|
|
celery_app = Celery(
|
|
"iaai_scraper",
|
|
broker=_broker_url(),
|
|
backend=_result_backend(),
|
|
)
|
|
|
|
celery_app.conf.update(
|
|
task_serializer="json",
|
|
accept_content=["json"],
|
|
result_serializer="json",
|
|
timezone="UTC",
|
|
enable_utc=True,
|
|
|
|
task_soft_time_limit=settings.celery.task_soft_time_limit,
|
|
task_time_limit=settings.celery.task_time_limit,
|
|
worker_concurrency=settings.celery.worker_concurrency,
|
|
|
|
# prefork + 1 процесс, тк playwright sync api
|
|
worker_pool="prefork",
|
|
worker_prefetch_multiplier=1,
|
|
|
|
broker_connection_retry_on_startup=True,
|
|
|
|
beat_schedule={
|
|
"periodic-sync-listing": {
|
|
"task": "iaai_scraper.worker.tasks.sync_listing_task",
|
|
"schedule": settings.celery.beat_sync_interval_minutes * 60.0,
|
|
"kwargs": {"limit": settings.celery.beat_sync_limit},
|
|
"options": {"queue": "scraping"},
|
|
},
|
|
},
|
|
|
|
task_routes={
|
|
"iaai_scraper.worker.tasks.*": {"queue": "scraping"},
|
|
},
|
|
)
|
|
|
|
celery_app.autodiscover_tasks(["iaai_scraper.worker"])
|