NixReaperandClaude Sonnet 5
c20df1c503
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]>
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
"""Assembles Company API responses from repository data."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.ids import aurora_id, parse_aurora_id
|
|
from app.models.enums import ArtworkType
|
|
from app.repositories import company_repository, polymorphic
|
|
from app.schemas.common import CompanyBriefOut
|
|
from app.schemas.company import CompanyDetailOut
|
|
from app.services.common import primary_artwork_path
|
|
|
|
|
|
class CompanyNotFoundError(Exception):
|
|
pass
|
|
|
|
|
|
def _resolve_company_id(aid: str) -> uuid.UUID:
|
|
try:
|
|
return uuid.UUID(parse_aurora_id(aid, expected_type="company"))
|
|
except ValueError as exc:
|
|
raise CompanyNotFoundError(str(exc)) from exc
|
|
|
|
|
|
async def get_company_detail(session: AsyncSession, company_aurora_id: str) -> CompanyDetailOut:
|
|
company_id = _resolve_company_id(company_aurora_id)
|
|
company = await company_repository.get_company_by_id(session, company_id)
|
|
if company is None:
|
|
raise CompanyNotFoundError(company_aurora_id)
|
|
|
|
logos = await polymorphic.get_artwork(session, "company", company_id, ArtworkType.LOGO)
|
|
external_ids = await polymorphic.get_external_ids(session, "company", company_id)
|
|
|
|
return CompanyDetailOut(
|
|
id=aurora_id("company", str(company.id)),
|
|
name=company.name,
|
|
company_type=company.company_type.value,
|
|
description=company.description,
|
|
country=company.country_code,
|
|
headquarters=company.headquarters,
|
|
homepage=company.homepage,
|
|
logo_path=primary_artwork_path(logos, ArtworkType.LOGO),
|
|
parent_company_id=(
|
|
aurora_id("company", str(company.parent_company_id))
|
|
if company.parent_company_id
|
|
else None
|
|
),
|
|
external_ids=external_ids,
|
|
created_at=company.created_at,
|
|
updated_at=company.updated_at,
|
|
)
|
|
|
|
|
|
async def build_company_summary(
|
|
session: AsyncSession, company_id: uuid.UUID
|
|
) -> CompanyBriefOut | None:
|
|
company = await company_repository.get_company_by_id(session, company_id)
|
|
if company is None:
|
|
return None
|
|
logos = await polymorphic.get_artwork(session, "company", company_id, ArtworkType.LOGO)
|
|
return CompanyBriefOut(
|
|
id=aurora_id("company", str(company.id)),
|
|
name=company.name,
|
|
company_type=company.company_type.value,
|
|
logo_path=primary_artwork_path(logos, ArtworkType.LOGO),
|
|
)
|
|
|
|
|
|
async def build_company_summaries(
|
|
session: AsyncSession, company_ids: list[uuid.UUID]
|
|
) -> list[CompanyBriefOut]:
|
|
out: list[CompanyBriefOut] = []
|
|
for company_id in company_ids:
|
|
summary = await build_company_summary(session, company_id)
|
|
if summary is not None:
|
|
out.append(summary)
|
|
return out
|