from dataclasses import dataclass

from litestar import Litestar, get
from litestar.dto import DTOConfig, DataclassDTO


@dataclass
class Dog:
    id: int
    nickname: str


@dataclass
class Cat:
    id: int
    color: str


@dataclass
class Person:
    id: int
    name: str
    pets: tuple[Dog, Cat]


class ReadDTO(DataclassDTO[Person]):
    config = DTOConfig(
        include={"name", "pets.0.name", "pets.1.color"}
    )


@get("/", return_dto=ReadDTO)
async def index() -> Person:
    person = Person(id=1, name="John", pets=(
        Dog(id=2, nickname="Dog"),
        Cat(id=3, color="White"),
    ))
    return person


app = Litestar(route_handlers=[index], debug=True)