Aurora-Core/app/schemas/company.py

"""Company response schemas."""

from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel

from app.schemas.common import CompanyBriefOut
from app.schemas.movie import MovieSummaryOut
from app.schemas.tv import SeriesSummaryOut


class CompanyDetailOut(BaseModel):
    id: str
    type: str = "company"
    name: str
    company_type: str
    description: str | None = None
    country: str | None = None
    headquarters: str | None = None
    homepage: str | None = None
    logo_path: str | None = None
    parent_company_id: str | None = None
    external_ids: dict[str, str] = {}

    movies: list[MovieSummaryOut] = []
    series: list[SeriesSummaryOut] = []

    created_at: datetime
    updated_at: datetime


class CompanyListOut(BaseModel):
    page: int
    page_size: int
    total_results: int
    total_pages: int
    results: list[CompanyBriefOut] = []