vectorvein 0.2.34__py3-none-any.whl → 0.2.36__py3-none-any.whl

vectorvein/workflow/graph/node.py

@@ -4,6 +4,37 @@ 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
+        self._initializing = True  # 初始化标志
+
+    def __setitem__(self, key: str, value: Port) -> None:
+        # 初始化阶段或端口已存在时，允许直接添加/更新
+        if self._initializing or 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)
+
+    def finish_initialization(self):
+        """结束初始化阶段"""
+        self._initializing = False
+
+
 class Node:
     def __init__(
         self,
@@ -25,7 +56,17 @@ 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.ports.finish_initialization()
+
         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 +74,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,

vectorvein/workflow/graph/workflow.py

@@ -48,28 +48,28 @@ class Workflow:
         # 检查源节点是否存在
         source_node_exists = any(node.id == source_node_id for node in self.nodes)
         if not source_node_exists:
-            raise ValueError(f"源节点不存在: {source_node_id}")
+            raise ValueError(f"Source node not found: {source_node_id}")
 
         # 检查目标节点是否存在
         target_node_exists = any(node.id == target_node_id for node in self.nodes)
         if not target_node_exists:
-            raise ValueError(f"目标节点不存在: {target_node_id}")
+            raise ValueError(f"Target node not found: {target_node_id}")
 
         # 检查源节点的端口是否存在
         source_node_obj = next(node for node in self.nodes if node.id == source_node_id)
         if not source_node_obj.has_output_port(source_port):
-            raise ValueError(f"源节点 {source_node_id} 不存在输出端口: {source_port}")
+            raise ValueError(f"Source node {source_node_id} has no output port: {source_port}")
 
         # 检查目标节点的端口是否存在
         target_node_obj = next(node for node in self.nodes if node.id == target_node_id)
         if not target_node_obj.has_input_port(target_port):
-            raise ValueError(f"目标节点 {target_node_id} 不存在输入端口: {target_port}")
+            raise ValueError(f"Target node {target_node_id} has no input port: {target_port}")
 
         # 检查目标端口是否已有被连接的线
         for edge in self.edges:
             if edge.target == target_node_id and edge.target_handle == target_port:
                 raise ValueError(
-                    f"目标节点 {target_node_id} 的输入端口 {target_port} 已经被连接: {edge.source}({edge.source_handle}) → {edge.target}({edge.target_handle})"
+                    f"The input port {target_port} of the target node {target_node_id} is already connected: {edge.source}({edge.source_handle}) → {edge.target}({edge.target_handle})"
                 )
 
         # 创建并添加边

{vectorvein-0.2.34.dist-info → vectorvein-0.2.36.dist-info}/METADATA

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

{vectorvein-0.2.34.dist-info → vectorvein-0.2.36.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.34.dist-info/METADATA,sha256=WO8O8F9QaASuHqAVYkEkBwabFpV974hXTnzZ3CSggSc,4570
-vectorvein-0.2.34.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.34.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.36.dist-info/METADATA,sha256=gSasEHecgJcAzKS-5l6V8Wk2yQMYZlNKUd4XkiUCYUo,4570
+vectorvein-0.2.36.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.36.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/api/__init__.py,sha256=lfY-XA46fgD2iIZTU0VYP8i07AwA03Egj4Qua0vUKrQ,738
 vectorvein/api/client.py,sha256=xF-leKDQzVyyy9FnIRaz0k4eElYW1XbbzeRLcpnyk90,33047
@@ -42,9 +42,9 @@ vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9k
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
 vectorvein/workflow/graph/edge.py,sha256=1ckyyjCue_PLm7P1ItUfKOy6AKkemOpZ9m1WJ8UXIHQ,1072
-vectorvein/workflow/graph/node.py,sha256=3txou3HlxniY9aSPoS4JsRRiEbseJ3LJnAcYaMWXGVI,3769
+vectorvein/workflow/graph/node.py,sha256=U3LAq01wOaDANVUuYWc2BhIq-Pl2-33FirJZkkNL_wA,5329
 vectorvein/workflow/graph/port.py,sha256=Q6HmI2cUi6viJ98ec6-MmMPMRtKS1-OgaudP3LMwVLA,6054
-vectorvein/workflow/graph/workflow.py,sha256=l0rSDZeSd6OtHMwxzhIENfOKS0fJKKB4JUow-dV-LUI,8610
+vectorvein/workflow/graph/workflow.py,sha256=K-1-FMfHO49Ww3sHKDOr_sD8GH-i36ylS7qtyJ-Vivg,8617
 vectorvein/workflow/nodes/__init__.py,sha256=a1fsitA0mf1UlDOSEz6QjSGjiFsvsQcuDE-p2q9SlhM,3181
 vectorvein/workflow/nodes/audio_generation.py,sha256=ZRFZ_ycMTSJ2LKmekctagQdJYTl-3q4TNOIKETpS9AM,5870
 vectorvein/workflow/nodes/control_flows.py,sha256=l8CjFQlsGV3fNGM6SVzS1Kz361K1xDv1fGT7acuDXuU,6613
@@ -62,4 +62,4 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=BhJBX1AZH7-22Gu95Ox4qJqmH5DU-m4dbUb5N5DTA-M,5559
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.2.34.dist-info/RECORD,,
+vectorvein-0.2.36.dist-info/RECORD,,