models.py

import typing as tp
from uuid import uuid4

from pydantic import BaseModel, Field


class ProductionSchemaStage(BaseModel):
    name: str
    stage_id: str = Field(default_factory=lambda: uuid4().hex)
    type: tp.Optional[str] = None
    description: tp.Optional[str] = None
    equipment: tp.Optional[tp.List[str]] = None
    workplace: tp.Optional[str] = None
    duration_seconds: tp.Optional[int] = None


class ProductionSchema(BaseModel):
    schema_id: str = Field(default_factory=lambda: uuid4().hex)
    unit_name: str
    unit_short_name: str = None
    schema_type: str
    production_stages: tp.Optional[tp.List[ProductionSchemaStage]] = None
    required_components_schema_ids: tp.Optional[tp.List[str]] = None
    parent_schema_id: tp.Optional[str] = None

    @property
    def is_composite(self) -> bool:
        return self.required_components_schema_ids is not None

    @property
    def is_a_component(self) -> bool:
        return self.parent_schema_id is not None