Files
core/model_nodes/node_while_models.py
2025-10-06 19:13:55 +05:00

66 lines
2.5 KiB
Python

from typing import Optional
from pydantic import BaseModel, Field
from .form_base_descriptor_models import FormDescriptor, RowElement, AreaElement, LinkPort
class WhileNodeData(BaseModel):
"""
Pydantic модель для валидации данных узла while
"""
ps_id: Optional[int] = Field(default=None, description="ID процесса")
node_type: Optional[str] = Field(default=None, description="Тип узла")
class WhileNodeLinks(BaseModel):
"""
Pydantic модель для валидации связей узла while
"""
parent_id: Optional[int] = Field(default=None, description="ID родительского узла")
parent_port_number: Optional[int] = Field(default=None, description="Номер порта родительского узла")
class WhileNodeCoreSchemaData(BaseModel):
"""
Pydantic модель для данных портов узла while
"""
then_port_number: Optional[int] = Field(default=0, description="Номер порта для перехода по Связи Then (LINK)")
else_port_number: Optional[int] = Field(default=1, description="Номер порта для перехода по Связи Else (LINK)")
class WhileNodeCoreSchema(BaseModel):
"""
Pydantic модель для схемы узла while
"""
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[WhileNodeCoreSchemaData] = Field(default=None, description="Данные узла")
# Дескриптор формы узла WHILE
WHILE_FORM_DESCRIPTOR = FormDescriptor(
elements=[
RowElement(
name="condition_row",
type="row",
label="Parameter",
link_port=LinkPort(id="then_output", label="Then"),
elements=[
AreaElement(
name="condition",
type="area",
label="Parameter",
placeholder="P(condition) == true",
)
],
),
RowElement(
name="else_row",
type="row",
label="Alternative path",
link_port=LinkPort(id="else_output", label="Else"),
),
]
)