from enum import Enum
from datetime import datetime
from pydantic import Field

from beanie import Document


class EntityType(str, Enum):
    individual = 'individual'
    corporation = 'corporation'
    institution = 'institution'


class Entity(Document):
    _id: str
    type: EntityType
    name: str
    address: str
    created_at: datetime = Field(default=datetime.utcnow(), nullable=False)
    updated_at: datetime = Field(default_factory=datetime.utcnow, nullable=False)