80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
from datetime import date
|
|
from typing import List, Literal, Optional
|
|
|
|
from pydantic import Field, BaseModel, ConfigDict
|
|
from beanie import PydanticObjectId
|
|
|
|
from firm.core.models import CrudDocument, ForeignKey
|
|
from firm.core.filter import Filter, FilterSchema
|
|
|
|
|
|
class EntityType(BaseModel):
|
|
@property
|
|
def label(self) -> str:
|
|
return self.title
|
|
|
|
|
|
class Individual(EntityType):
|
|
model_config = ConfigDict(title='Particulier')
|
|
|
|
type: Literal['individual'] = 'individual'
|
|
firstname: str = Field(title='Prénom')
|
|
middlename: str = Field(default="", title='Autres prénoms')
|
|
lastname: str = Field(title='Nom de famille')
|
|
surnames: List[str] = Field(
|
|
default=[],
|
|
props={"items_per_row": "4", "numbered": True},
|
|
title="Surnoms"
|
|
)
|
|
day_of_birth: Optional[date] = Field(default=None, title='Date de naissance')
|
|
place_of_birth: Optional[str] = Field(default="", title='Lieu de naissance')
|
|
|
|
@property
|
|
def label(self) -> str:
|
|
# if len(self.surnames) > 0:
|
|
# return f"{self.firstname} \"{self.surnames[0]}\" {self.lastname}"
|
|
return f"{self.firstname} {self.lastname}"
|
|
|
|
|
|
class Employee(BaseModel):
|
|
model_config = ConfigDict(title='Fiche Employé')
|
|
|
|
position: str = Field(title='Poste')
|
|
entity_id: PydanticObjectId = ForeignKey("entities", "Entity", title='Employé')
|
|
|
|
|
|
class Corporation(EntityType):
|
|
model_config = ConfigDict(title='Entreprise')
|
|
|
|
type: Literal['corporation'] = 'corporation'
|
|
title: str = Field(title='Dénomination sociale')
|
|
activity: str = Field(title='Activité')
|
|
employees: List[Employee] = Field(default=[], title='Employés')
|
|
|
|
|
|
class Institution(Corporation):
|
|
model_config = ConfigDict(title='Institution')
|
|
|
|
type: Literal['institution'] = 'institution'
|
|
|
|
|
|
class Entity(CrudDocument):
|
|
"""
|
|
Fiche d'un client
|
|
"""
|
|
model_config = ConfigDict(title='Client')
|
|
|
|
entity_data: Individual | Corporation | Institution = Field(..., discriminator='type')
|
|
address: str = Field(default="", title='Adresse')
|
|
|
|
def compute_label(self) -> str:
|
|
if not self.entity_data:
|
|
return ""
|
|
return self.entity_data.label
|
|
|
|
|
|
class EntityFilters(FilterSchema):
|
|
class Constants(Filter.Constants):
|
|
model = Entity
|
|
search_model_fields = ["label"]
|