from typing import Optional
from pydantic import BaseModel, Field


class WaitNodeData(BaseModel):
    """
    Pydantic модель для валидации данных узла wait
    """
    ps_id: Optional[int] = Field(default=None, description="ID процесса")
    node_type: Optional[str] = Field(default=None, description="Тип узла")


class WaitNodeLinks(BaseModel):
    """
    Pydantic модель для валидации связей узла wait
    """
    parent_id: Optional[int] = Field(default=None, description="ID родительского узла")
    parent_port_number: Optional[int] = Field(default=None, description="Номер порта родительского узла")


class WaitNodeCoreSchemaData(BaseModel):
    """
    Pydantic модель для данных портов узла wait
    """
    then_port_number: Optional[int] = Field(default=0, description="Номер порта для перехода по Связи then (LINK)")


class WaitNodeCoreSchema(BaseModel):
    """
    Pydantic модель для схемы узла wait
    """
    ps_id: Optional[int] = Field(default=None, description="ID процесса")
    node_type: Optional[str] = Field(default=None, description="Тип узла")
    parent_id: Optional[int] = Field(default=None, description="ID родительского узла")
    parent_port_number: Optional[int] = Field(default=None, description="Номер порта родительского узла")
    data: Optional[WaitNodeCoreSchemaData] = Field(default=None, description="Данные узла")