from pydantic import BaseModel, Field

from schemas.CredencialSchema import CredencialBaseSchema
from api.api_v2.modules.pedimentos.schemas import PedimentoBaseSchema

class PartidaBaseSchema(BaseModel):
    id: int
    numero: int
    
class PartidaRequestSchema(BaseModel):
    partida: PartidaBaseSchema
    pedimento: PedimentoBaseSchema
    credencial: CredencialBaseSchema

class PartidaListSchema(BaseModel):
    partidas: list[PartidaBaseSchema] = Field(..., description="Lista de partidas")
    pedimento: PedimentoBaseSchema
    credencial: CredencialBaseSchema