from typing import Any
from uuid import UUID

from litestar import Litestar, post
from litestar.dto import DTOConfig, DTOData
from litestar.plugins.pydantic import PydanticDTO
from pydantic import BaseModel, Field


class Address(BaseModel):
    street: str = ""
    city: str = ""
    state: str = ""
    zip: str = ""


class User(BaseModel):
    id: UUID
    name: str

    nullable_password: str | None
    nullable_address: Address | None
    optional_password: str | None = None
    optional_address: Address | None = None
    passwords: list[str] = Field(default_factory=list)
    addresses: list[Address] = Field(default_factory=list)


class UserDTO(PydanticDTO[User]):
    config = DTOConfig(exclude={"id"})


@post("/users", dto=UserDTO, sync_to_thread=False)
def create_user(data: DTOData[User]) -> dict[str, Any]:
    # data.create_instance(id=uuid4())
    print(data.as_builtins())
    return data.as_builtins()


app = Litestar(route_handlers=[create_user])