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
|