from pydantic import Field

from firm.entity.models import Entity, Institution, Individual, Corporation
from firm.core.schemas import Writer, Reader

class EntityRead(Reader, Entity):
    pass

class EntityCreate(Writer):
    entity_data: Individual | Corporation | Institution = Field(..., discriminator='type')
    address: str = Field(default="", title='Adresse')

    class Config:
        title = "Création d'un client"

class EntityUpdate(EntityCreate):
    pass