import uuid

from datetime import datetime
from pydantic import BaseModel, Field

from .models import Entity, EntityType, Individual, Corporation
from ..core.schemas import Writer


class EntityRead(Entity):
    pass


class EntityCreate(Writer):
    name: str
    address: str
    entity_data: Individual | Corporation = Field(..., discriminator='type')


class EntityUpdate(BaseModel):
    name: str
    address: str
    entity_data: Individual | Corporation = Field(..., discriminator='type')