Initial Aurora Core 0.1 implementation
CI / lint (push) Canceled after 0s
CI / typecheck (push) Canceled after 0s
CI / unit-tests (push) Canceled after 0s
CI / integration-tests (push) Canceled after 0s
CI / container-build (push) Canceled after 0s
CI / dependency-audit (push) Canceled after 0s
CI / lint (push) Canceled after 0s
CI / typecheck (push) Canceled after 0s
CI / unit-tests (push) Canceled after 0s
CI / integration-tests (push) Canceled after 0s
CI / container-build (push) Canceled after 0s
CI / dependency-audit (push) Canceled after 0s
Production-grade media metadata backend per docs/aurora-core-spec.md: Aurora-owned UUIDv7 identity system, normalized movie/TV/people/credits schema, provider abstraction with a TMDb adapter, an idempotent async ingestion pipeline (normalize -> match -> upsert -> provenance), a Postgres FTS/trigram search layer, a stateless FastAPI v1 API with Redis caching/rate-limiting, arq workers, dev/prod Docker Compose topologies, unit + integration + load tests, CI, and architecture/ database/API/ERD documentation. Co-Authored-By: Claude Sonnet 5 <[email protected]>
This commit is contained in:
@@ -0,0 +1,90 @@
|
||||
"""Shared data access for the polymorphic satellite tables — credits,
|
||||
artwork, videos, external IDs — that attach to more than one entity type via
|
||||
an (entity_type, entity_id) pair rather than a typed foreign key.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import uuid
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from sqlalchemy.orm import selectinload
|
||||
|
||||
from app.models.artwork import Artwork
|
||||
from app.models.credit import Credit
|
||||
from app.models.enums import ArtworkType
|
||||
from app.models.external_id import ExternalID
|
||||
from app.models.provider import Provider
|
||||
from app.models.video import Video
|
||||
|
||||
|
||||
async def get_external_ids(
|
||||
session: AsyncSession, entity_type: str, entity_id: uuid.UUID
|
||||
) -> dict[str, str]:
|
||||
stmt = (
|
||||
select(Provider.slug, ExternalID.external_id)
|
||||
.join(Provider, Provider.id == ExternalID.provider_id)
|
||||
.where(ExternalID.entity_type == entity_type, ExternalID.entity_id == entity_id)
|
||||
)
|
||||
result = await session.execute(stmt)
|
||||
return {slug: external_id for slug, external_id in result.all()}
|
||||
|
||||
|
||||
async def get_external_ids_bulk(
|
||||
session: AsyncSession, entity_type: str, entity_ids: list[uuid.UUID]
|
||||
) -> dict[uuid.UUID, dict[str, str]]:
|
||||
if not entity_ids:
|
||||
return {}
|
||||
stmt = (
|
||||
select(ExternalID.entity_id, Provider.slug, ExternalID.external_id)
|
||||
.join(Provider, Provider.id == ExternalID.provider_id)
|
||||
.where(ExternalID.entity_type == entity_type, ExternalID.entity_id.in_(entity_ids))
|
||||
)
|
||||
result = await session.execute(stmt)
|
||||
out: dict[uuid.UUID, dict[str, str]] = {eid: {} for eid in entity_ids}
|
||||
for entity_id, slug, external_id in result.all():
|
||||
out.setdefault(entity_id, {})[slug] = external_id
|
||||
return out
|
||||
|
||||
|
||||
async def get_artwork(
|
||||
session: AsyncSession,
|
||||
entity_type: str,
|
||||
entity_id: uuid.UUID,
|
||||
artwork_type: ArtworkType | None = None,
|
||||
) -> list[Artwork]:
|
||||
stmt = (
|
||||
select(Artwork)
|
||||
.options(selectinload(Artwork.provider))
|
||||
.where(Artwork.entity_type == entity_type, Artwork.entity_id == entity_id)
|
||||
.order_by(Artwork.is_primary.desc(), Artwork.vote_average.desc().nulls_last())
|
||||
)
|
||||
if artwork_type is not None:
|
||||
stmt = stmt.where(Artwork.artwork_type == artwork_type)
|
||||
result = await session.execute(stmt)
|
||||
return list(result.scalars().all())
|
||||
|
||||
|
||||
async def get_videos(session: AsyncSession, entity_type: str, entity_id: uuid.UUID) -> list[Video]:
|
||||
stmt = (
|
||||
select(Video)
|
||||
.options(selectinload(Video.provider))
|
||||
.where(Video.entity_type == entity_type, Video.entity_id == entity_id)
|
||||
.order_by(Video.is_official.desc(), Video.published_at.desc().nulls_last())
|
||||
)
|
||||
result = await session.execute(stmt)
|
||||
return list(result.scalars().all())
|
||||
|
||||
|
||||
async def get_credits(
|
||||
session: AsyncSession, entity_type: str, entity_id: uuid.UUID
|
||||
) -> list[Credit]:
|
||||
stmt = (
|
||||
select(Credit)
|
||||
.options(selectinload(Credit.person))
|
||||
.where(Credit.entity_type == entity_type, Credit.entity_id == entity_id)
|
||||
.order_by(Credit.billing_order.asc().nulls_last())
|
||||
)
|
||||
result = await session.execute(stmt)
|
||||
return list(result.scalars().all())
|
||||
Reference in New Issue
Block a user