Aurora-Core/app/workers/tasks.py

"""arq task functions. Thin wrappers around the ingestion runner so retry/
concurrency/observability behavior lives in arq's worker loop, not here.

Each task receives the `IngestionJob` row's id (created synchronously by the
enqueuing endpoint via `get_or_create_ingestion_job`, *not* generated here)
so status polling has a real, immediately-valid id to look up from the
moment the HTTP response comes back.
"""

from __future__ import annotations

import uuid
from typing import Any

from app.core.logging import get_logger
from app.services.ingestion.runner import (
    run_movie_ingestion,
    run_person_ingestion,
    run_season_ingestion,
    run_series_ingestion,
)

logger = get_logger(__name__)


async def ingest_movie_task(ctx: dict[str, Any], job_id: str, external_id: str) -> str:
    movie_id = await run_movie_ingestion(uuid.UUID(job_id), external_id)
    return str(movie_id)


async def ingest_person_task(ctx: dict[str, Any], job_id: str, external_id: str) -> str:
    person_id = await run_person_ingestion(uuid.UUID(job_id), external_id)
    return str(person_id)


async def ingest_series_task(ctx: dict[str, Any], job_id: str, external_id: str) -> str:
    series_id = await run_series_ingestion(uuid.UUID(job_id), external_id)
    return str(series_id)


async def ingest_season_task(
    ctx: dict[str, Any], job_id: str, series_external_id: str, season_number: int
) -> str:
    season_id = await run_season_ingestion(uuid.UUID(job_id), series_external_id, season_number)
    return str(season_id)