from typing import Optional
from pydantic import BaseModel, Field


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


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


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


class SetNodeCoreSchema(BaseModel):
    """
    Pydantic модель для схемы узла set
    """
    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[SetNodeCoreSchemaData] = Field(default=None, description="Данные узла")