feat: all models
This commit is contained in:
@@ -1,18 +1,73 @@
|
||||
from typing import Dict, Any
|
||||
from core import VorkNode
|
||||
from core.form_descriptors import get_form_descriptor
|
||||
from model_nodes.node_while_models import (
|
||||
WhileNodeData,
|
||||
WhileNodeLinks,
|
||||
WhileNodeCoreSchema,
|
||||
WhileNodeCoreSchemaData
|
||||
)
|
||||
|
||||
|
||||
class VorkNodeWhile(VorkNode):
|
||||
|
||||
def __init__(self, data: Dict[str, Any], links: Dict[str, Any] = None):
|
||||
"""
|
||||
Инициализация узла while
|
||||
"""
|
||||
super().__init__(data, links or {})
|
||||
|
||||
@property
|
||||
def id(self) -> str:
|
||||
return "node_while"
|
||||
return "WHILE"
|
||||
|
||||
@classmethod
|
||||
def form(cls):
|
||||
pass
|
||||
def form(cls) -> Dict[str, Any]:
|
||||
"""
|
||||
Возвращает статический дескриптор формы для узла While
|
||||
"""
|
||||
return get_form_descriptor("WHILE")
|
||||
|
||||
def validate(self) -> bool:
|
||||
return True
|
||||
def validate(self) -> WhileNodeCoreSchema:
|
||||
"""
|
||||
Валидирует данные узла while и возвращает схему
|
||||
"""
|
||||
try:
|
||||
# Валидируем данные узла
|
||||
validated_data = self.validate_data()
|
||||
|
||||
# Валидируем связи узла
|
||||
validated_links = self.validate_links()
|
||||
|
||||
# Создаем данные портов (аналогично if)
|
||||
node_data = WhileNodeCoreSchemaData(
|
||||
then_port_number=0,
|
||||
else_port_number=1,
|
||||
)
|
||||
|
||||
# Создаем схему с валидированными данными
|
||||
return WhileNodeCoreSchema(
|
||||
ps_id=validated_data.ps_id,
|
||||
node_type=validated_data.node_type,
|
||||
parent_id=validated_links.parent_id,
|
||||
parent_port_number=validated_links.parent_port_number,
|
||||
data=node_data
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"While node validation error: {e}")
|
||||
raise
|
||||
|
||||
def validate_data(self) -> WhileNodeData:
|
||||
"""
|
||||
Валидирует данные узла while
|
||||
"""
|
||||
return WhileNodeData(**self.data)
|
||||
|
||||
def validate_links(self) -> WhileNodeLinks:
|
||||
"""
|
||||
Валидирует связи узла while
|
||||
"""
|
||||
return WhileNodeLinks(**self.links)
|
||||
|
||||
def process(self, context):
|
||||
pass
|
||||
|
Reference in New Issue
Block a user