vectorvein 0.2.34__tar.gz → 0.2.35__tar.gz

{vectorvein-0.2.34 → vectorvein-0.2.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.2.34
+Version: 0.2.35
 Summary: VectorVein Python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.2.34 → vectorvein-0.2.35}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein Python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.34"
+version = "0.2.35"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.34 → vectorvein-0.2.35}/src/vectorvein/workflow/graph/node.py

@@ -4,6 +4,32 @@ from typing import Dict, Any, Optional, List, Union
 from .port import PortType, Port, InputPort, OutputPort
 
 
+class PortsDict(Dict[str, Port]):
+    """自定义字典类，用于管理节点的端口，并强制执行端口添加权限。"""
+
+    def __init__(self, owner_node: "Node", *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._owner_node = owner_node
+
+    def __setitem__(self, key: str, value: Port) -> None:
+        # 如果端口已存在，允许更新
+        if key in self:
+            super().__setitem__(key, value)
+            return
+
+        # 对于新端口，检查添加权限
+        if isinstance(value, OutputPort) and not self._owner_node.can_add_output_ports:
+            raise ValueError(
+                f"Node<{self._owner_node.id}> '{self._owner_node.type}' does not allow adding output ports"
+            )
+        elif isinstance(value, InputPort) and not self._owner_node.can_add_input_ports:
+            raise ValueError(
+                f"Node<{self._owner_node.id}> '{self._owner_node.type}' does not allow adding input ports"
+            )
+
+        super().__setitem__(key, value)
+
+
 class Node:
     def __init__(
         self,
@@ -25,7 +51,14 @@ class Node:
         self.category: str = category
         self.task_name: str = task_name
         self.description: str = description
-        self.ports: Dict[str, Port] = ports or {}
+        self.can_add_input_ports: bool = can_add_input_ports
+        self.can_add_output_ports: bool = can_add_output_ports
+        # 初始化自定义PortsDict
+        self.ports = PortsDict(self)
+        # 如果提供了初始端口，将它们添加到字典中
+        if ports:
+            for name, port in ports.items():
+                self.ports[name] = port
         self.position: Dict[str, float] = position or {"x": 0, "y": 0}
         self.seleted_workflow_title: str = seleted_workflow_title
         self.is_template: bool = is_template
@@ -33,8 +66,6 @@ class Node:
         self.ignored: bool = False
         self.lock: bool = False
         self.shadow: bool = False
-        self.can_add_input_ports: bool = can_add_input_ports
-        self.can_add_output_ports: bool = can_add_output_ports
 
     def add_port(
         self,