kailash 0.1.5__py3-none-any.whl → 0.2.0__py3-none-any.whl

kailash/nodes/data/writers.py

@@ -34,6 +34,7 @@ import json
 from typing import Any, Dict
 
 from kailash.nodes.base import Node, NodeParameter, register_node
+from kailash.security import safe_open, validate_file_path
 
 
 @register_node()
@@ -190,7 +191,7 @@ class CSVWriterNode(Node):
             - External tools can process output
             - Metrics available for monitoring
         """
-        file_path = kwargs["file_path"]
+        file_path = kwargs.get("file_path") or self.config.get("file_path")
         data = kwargs["data"]
         headers = kwargs.get("headers")
         delimiter = kwargs.get("delimiter", ",")
@@ -198,7 +199,10 @@ class CSVWriterNode(Node):
         if not data:
             return {"rows_written": 0}
 
-        with open(file_path, "w", newline="", encoding="utf-8") as f:
+        # Validate file path for security
+        validated_path = validate_file_path(file_path, operation="CSV write")
+
+        with safe_open(validated_path, "w", newline="", encoding="utf-8") as f:
             if isinstance(data[0], dict):
                 # Writing dictionaries
                 if not headers:
@@ -357,11 +361,14 @@ class JSONWriterNode(Node):
             - Version control can track
             - APIs can import data
         """
-        file_path = kwargs["file_path"]
+        file_path = kwargs.get("file_path") or self.config.get("file_path")
         data = kwargs["data"]
         indent = kwargs.get("indent", 2)
 
-        with open(file_path, "w", encoding="utf-8") as f:
+        # Validate file path for security
+        validated_path = validate_file_path(file_path, operation="JSON write")
+
+        with safe_open(validated_path, "w", encoding="utf-8") as f:
             json.dump(data, f, indent=indent, ensure_ascii=False)
 
         return {"file_path": file_path}
@@ -517,13 +524,16 @@ class TextWriterNode(Node):
             - Log analyzers can process
             - Metrics available for monitoring
         """
-        file_path = kwargs["file_path"]
+        file_path = kwargs.get("file_path") or self.config.get("file_path")
         text = kwargs["text"]
         encoding = kwargs.get("encoding", "utf-8")
         append = kwargs.get("append", False)
 
         mode = "a" if append else "w"
-        with open(file_path, mode, encoding=encoding) as f:
+        # Validate file path for security
+        validated_path = validate_file_path(file_path, operation="text write")
+
+        with safe_open(validated_path, mode, encoding=encoding) as f:
             f.write(text)
 
         return {"file_path": file_path, "bytes_written": len(text.encode(encoding))}

kailash/nodes/logic/__init__.py

@@ -1,6 +1,11 @@
 """Logic operation nodes for the Kailash SDK."""
 
 from kailash.nodes.logic.async_operations import AsyncMergeNode, AsyncSwitchNode
+from kailash.nodes.logic.convergence import (
+    ConvergenceCheckerNode,
+    MultiCriteriaConvergenceNode,
+)
+from kailash.nodes.logic.loop import LoopNode
 from kailash.nodes.logic.operations import MergeNode, SwitchNode
 from kailash.nodes.logic.workflow import WorkflowNode
 
@@ -10,4 +15,7 @@ __all__ = [
     "AsyncSwitchNode",
     "AsyncMergeNode",
     "WorkflowNode",
+    "LoopNode",
+    "ConvergenceCheckerNode",
+    "MultiCriteriaConvergenceNode",
 ]