vectorvein 0.2.33__py3-none-any.whl → 0.2.35__py3-none-any.whl

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,

vectorvein/workflow/nodes/text_processing.py

@@ -97,7 +97,7 @@ class TextSplitters(Node):
                     value="\\n",
                     condition="return fieldsData.split_method.value == 'delimiter'",
                 ),
-                "output": OutputPort(),
+                "output": OutputPort(list=True),
             },
         )
 

{vectorvein-0.2.33.dist-info → vectorvein-0.2.35.dist-info}/METADATA

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

{vectorvein-0.2.33.dist-info → vectorvein-0.2.35.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.33.dist-info/METADATA,sha256=NdnNIjJN4LboNo5a8s8eQNtBOEhC7q83EMvKL1_3kkk,4570
-vectorvein-0.2.33.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.33.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.35.dist-info/METADATA,sha256=cIifHIbhWkmoBq64isEQVHa0BoZKKCp1aXuR3IAyKCw,4570
+vectorvein-0.2.35.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.35.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,7 +42,7 @@ 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=9eBS4hOjpiIOdO9ZmG6ctbRtsZeDO29SJqsu_3ELFFc,5033
 vectorvein/workflow/graph/port.py,sha256=Q6HmI2cUi6viJ98ec6-MmMPMRtKS1-OgaudP3LMwVLA,6054
 vectorvein/workflow/graph/workflow.py,sha256=l0rSDZeSd6OtHMwxzhIENfOKS0fJKKB4JUow-dV-LUI,8610
 vectorvein/workflow/nodes/__init__.py,sha256=a1fsitA0mf1UlDOSEz6QjSGjiFsvsQcuDE-p2q9SlhM,3181
@@ -55,11 +55,11 @@ vectorvein/workflow/nodes/media_editing.py,sha256=qJImXwBJt59Ud0tzsRVKItPvrVZws3
 vectorvein/workflow/nodes/media_processing.py,sha256=jkSP287gyhvF7hqY8Nohdc1Rdn-br3zAj9sklPs6PAs,19778
 vectorvein/workflow/nodes/output.py,sha256=_UQxiddHtGv2rkjhUFE-KDgrjnh0AGJQJyq9-4Aji5A,12567
 vectorvein/workflow/nodes/relational_db.py,sha256=zfzUhV25TpZGhkIzO18PmAT5xhcsJC4AXKy0zyA05w8,5408
-vectorvein/workflow/nodes/text_processing.py,sha256=uhVmPDpbtlsqdg2ZlsM2Vi9L32Or_HWFR7wRFrX0q9c,8333
+vectorvein/workflow/nodes/text_processing.py,sha256=MRo_-oaC65hbzMxm7TYoeiS3rgvqh9y_Rny5RCfoATE,8342
 vectorvein/workflow/nodes/tools.py,sha256=uMlqJQI9oel3pG4WYrZdpSmWWy2pDQkAIzqOv5z3tB4,13811
 vectorvein/workflow/nodes/triggers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs076z-7dlQQ,5712
 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.33.dist-info/RECORD,,
+vectorvein-0.2.35.dist-info/RECORD,,