from typing import Optional
from pydantic import BaseModel, Field


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


class TestNodeLinks(BaseModel):
    """
    Pydantic модель для валидации связей узла test
    """
    # Test узел не имеет родительских связей, только исходящие


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


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