52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from enum import Enum
|
|
from datetime import datetime, date
|
|
from typing import List, Literal
|
|
|
|
from pydantic import Field, BaseModel
|
|
from beanie import Document, Link
|
|
|
|
|
|
class EntityType(str, Enum):
|
|
individual = 'individual'
|
|
corporation = 'corporation'
|
|
institution = 'institution'
|
|
|
|
|
|
class Individual(BaseModel):
|
|
type: Literal['individual'] = 'individual'
|
|
firstname: str
|
|
middlenames: List[str] = Field(default=[])
|
|
lastname: str
|
|
surnames: List[str] = Field(default=[])
|
|
day_of_birth: date
|
|
job: str
|
|
employer: str
|
|
|
|
|
|
class Employee(BaseModel):
|
|
entity_id: str
|
|
role: str
|
|
|
|
|
|
class Corporation(BaseModel):
|
|
type: Literal['corporation'] = 'corporation'
|
|
title: str
|
|
activity: str
|
|
employees: List[Employee] = Field(default=[])
|
|
|
|
|
|
class Entity(Document):
|
|
_id: str
|
|
name: str
|
|
address: str
|
|
entity_data: Individual | Corporation = Field(..., discriminator='type')
|
|
|
|
created_at: datetime = Field(default=datetime.utcnow(), nullable=False)
|
|
updated_at: datetime = Field(default_factory=datetime.utcnow, nullable=False)
|
|
|
|
class Settings:
|
|
bson_encoders = {
|
|
date: lambda dt: datetime(year=dt.year, month=dt.month, day=dt.day, hour=0, minute=0, second=0) \
|
|
if not hasattr(dt, 'hour') else dt
|
|
}
|