FlowerPower 0.31.1__tar.gz → 0.31.2__tar.gz

{flowerpower-0.31.1/src/FlowerPower.egg-info → flowerpower-0.31.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: FlowerPower
-Version: 0.31.1
+Version: 0.31.2
 Summary: A simple workflow framework for building and managing data processing pipelines
 Author-email: "Volker L." <ligno.blades@gmail.com>
 Project-URL: Homepage, https://github.com/legout/flowerpower

{flowerpower-0.31.1 → flowerpower-0.31.2}/pyproject.toml

@@ -4,7 +4,7 @@ description = "A simple workflow framework for building and managing data proces
 authors = [{ name = "Volker L.", email = "ligno.blades@gmail.com" }]
 readme = "README.md"
 requires-python = ">= 3.11"
-version = "0.31.1"
+version = "0.31.2"
 keywords = ["hamilton", "workflow", "pipeline", "scheduler", "dask", "ray"]
 
 dependencies = [
@@ -76,6 +76,7 @@ dev-dependencies = [
   "numpy>=2.3.3",
   "matplotlib>=3.10.6",
   "seaborn>=0.13.2",
+  "pyyaml>=6.0.3",
 ]
 package = true
 

{flowerpower-0.31.1 → flowerpower-0.31.2/src/FlowerPower.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: FlowerPower
-Version: 0.31.1
+Version: 0.31.2
 Summary: A simple workflow framework for building and managing data processing pipelines
 Author-email: "Volker L." <ligno.blades@gmail.com>
 Project-URL: Homepage, https://github.com/legout/flowerpower

{flowerpower-0.31.1 → flowerpower-0.31.2}/src/flowerpower/pipeline/manager.py

@@ -580,11 +580,32 @@ class PipelineManager:
         """
         return self._lifecycle_manager.list_pipelines()
 
-    def show_pipelines(self) -> None:
-        """Display all available pipelines in a formatted table.
+    def show_pipelines(self, format: str = "table") -> None:
+        """Display all available pipelines in a selected format.
 
-        Uses rich formatting for terminal display.
+        Args:
+            format: One of "table", "json", or "yaml". Defaults to "table".
         """
+        fmt = (format or "table").lower()
+        if fmt == "table":
+            return self.registry.show_pipelines()
+        names = self._lifecycle_manager.list_pipelines()
+        try:
+            if fmt == "json":
+                import json
+                print(json.dumps(names))
+                return None
+            if fmt == "yaml":
+                import yaml  # type: ignore
+                print(yaml.safe_dump(names, sort_keys=False))
+                return None
+        except Exception as e:
+            warnings.warn(
+                f"Failed to format pipelines as {fmt}: {e}. Falling back to table.",
+                RuntimeWarning,
+            )
+            return self.registry.show_pipelines()
+        warnings.warn(f"Unknown format '{format}', using table.", RuntimeWarning)
         return self.registry.show_pipelines()
 
     @property
@@ -985,7 +1006,13 @@ class PipelineManager:
         )
 
     # Visualizer Delegations
-    def save_dag(self, name: str, format: str = "png", reload: bool = False) -> None:
+    def save_dag(
+        self,
+        name: str,
+        format: str = "png",
+        reload: bool = False,
+        output_path: str | None = None,
+    ) -> str:
         """Save pipeline DAG visualization to a file.
 
         Creates a visual representation of the pipeline's directed acyclic graph (DAG)
@@ -1020,7 +1047,9 @@ class PipelineManager:
             ...     reload=True
             ... )
         """
-        self.visualizer.save_dag(name=name, format=format, reload=reload)
+        return self.visualizer.save_dag(
+            name=name, format=format, reload=reload, output_path=output_path
+        )
 
     def show_dag(
         self, name: str, format: str = "png", reload: bool = False, raw: bool = False

{flowerpower-0.31.1 → flowerpower-0.31.2}/src/flowerpower/pipeline/visualizer.py

@@ -67,7 +67,8 @@ class PipelineVisualizer:
         name: str,
         format: str = "png",
         reload: bool = False,
-    ):
+        output_path: str | None = None,
+    ) -> str:
         """
         Save an image of the graph of functions for a given pipeline name.
 
@@ -86,25 +87,38 @@ class PipelineVisualizer:
         """
         dag = self._get_dag_object(name=name, reload=reload)
 
-        # Use project_cfg attributes for path and filesystem access
-        graph_dir = posixpath.join(self.project_cfg.base_dir, "graphs")
-        self._fs.makedirs(graph_dir, exist_ok=True)
-
-        output_path = posixpath.join(
-            graph_dir, name
-        )  # Output filename is just the pipeline name
-        output_path_with_ext = f"{output_path}.{format}"
+        # Determine final output path
+        if output_path is None:
+            graph_dir = posixpath.join(self.project_cfg.base_dir, "graphs")
+            self._fs.makedirs(graph_dir, exist_ok=True)
+            base = posixpath.join(graph_dir, name)
+            final_path = f"{base}.{format}"
+            render_path = base
+        else:
+            # If output_path already has an extension, use as-is; otherwise append format
+            if "." in posixpath.basename(output_path):
+                final_path = output_path
+                # Remove extension for graphviz render base path
+                render_path = final_path.rsplit(".", 1)[0]
+                fmt = final_path.rsplit(".", 1)[1]
+                if fmt != format:
+                    # Honor explicit extension if it differs from format argument
+                    format = fmt
+            else:
+                final_path = f"{output_path}.{format}"
+                render_path = output_path
 
         # Render the DAG using the graphviz object returned by display_all_functions
         dag.render(
-            output_path,  # graphviz appends the format automatically
+            render_path,  # graphviz appends the format automatically
             format=format,
             cleanup=True,
             view=False,
         )
         print(
-            f"📊 Saved graph for [bold blue]{self.project_cfg.name}.{name}[/bold blue] to [green]{output_path_with_ext}[/green]"
+            f"📊 Saved graph for [bold blue]{self.project_cfg.name}.{name}[/bold blue] to [green]{final_path}[/green]"
         )
+        return final_path
 
     def show_dag(
         self,