feldera 0.119.0__tar.gz → 0.121.0__tar.gz

{feldera-0.119.0 → feldera-0.121.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: feldera
-Version: 0.119.0
+Version: 0.121.0
 Summary: The feldera python client
 Author-email: Feldera Team <dev@feldera.com>
 License: MIT

{feldera-0.119.0 → feldera-0.121.0}/feldera/pipeline.py

@@ -1,6 +1,7 @@
 import logging
 import time
 from datetime import datetime
+import pathlib
 
 import pandas
 from uuid import UUID
@@ -1059,3 +1060,33 @@ pipeline '{self.name}' to sync checkpoint '{uuid}'"""
         if derr:
             errors.append(derr)
         return errors
+
+    def support_bundle(self, output_path: Optional[str] = None) -> bytes:
+        """
+        Generate a support bundle containing diagnostic information from this pipeline.
+
+        This method collects various diagnostic data from the pipeline including
+        circuit profile, heap profile, metrics, logs, stats, and connector statistics,
+        and packages them into a single ZIP file for support purposes.
+
+        :param output_path: Optional path to save the support bundle file. If None,
+            the support bundle is only returned as bytes.
+        :return: The support bundle as bytes (ZIP archive)
+        :raises FelderaAPIError: If the pipeline does not exist or if there's an error
+        """
+
+        support_bundle_bytes = self.client.get_pipeline_support_bundle(self.name)
+
+        if output_path is not None:
+            path = pathlib.Path(output_path)
+
+            # Ensure the file has .zip extension
+            if path.suffix != ".zip":
+                path = path.with_suffix(".zip")
+
+            with open(path, "wb") as f:
+                f.write(support_bundle_bytes)
+
+            print(f"Support bundle written to {path}")
+
+        return support_bundle_bytes

{feldera-0.119.0 → feldera-0.121.0}/feldera/rest/feldera_client.py

@@ -951,3 +951,28 @@ Reason: The pipeline is in a STOPPED state due to the following error:
         resp = self.http.get(path="/config")
 
         return FelderaConfig(resp)
+
+    def get_pipeline_support_bundle(self, pipeline_name: str) -> bytes:
+        """
+        Generate a support bundle containing diagnostic information from a pipeline.
+
+        This endpoint collects various diagnostic data from the pipeline including
+        circuit profile, heap profile, metrics, logs, stats, and connector statistics,
+        and packages them into a single ZIP file for support purposes.
+
+        :param pipeline_name: The name of the pipeline
+        :return: The support bundle as bytes (ZIP file)
+        :raises FelderaAPIError: If the pipeline does not exist or if there's an error
+        """
+
+        resp = self.http.get(
+            path=f"/pipelines/{pipeline_name}/support_bundle",
+            stream=True,
+        )
+
+        buffer = b""
+        for chunk in resp.iter_content(chunk_size=1024):
+            if chunk:
+                buffer += chunk
+
+        return buffer

{feldera-0.119.0 → feldera-0.121.0}/feldera.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: feldera
-Version: 0.119.0
+Version: 0.121.0
 Summary: The feldera python client
 Author-email: Feldera Team <dev@feldera.com>
 License: MIT

{feldera-0.119.0 → feldera-0.121.0}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 name = "feldera"
 readme = "README.md"
 description = "The feldera python client"
-version = "0.119.0"
+version = "0.121.0"
 license = { text = "MIT" }
 requires-python = ">=3.10"
 authors = [

{feldera-0.119.0 → feldera-0.121.0}/tests/test_shared_pipeline.py

@@ -3,6 +3,9 @@ import pathlib
 import pandas as pd
 import time
 import unittest
+import io
+import tempfile
+import zipfile
 
 from tests.shared_test_pipeline import SharedTestPipeline
 from tests import TEST_CLIENT, enterprise_only
@@ -549,6 +552,41 @@ class TestPipeline(SharedTestPipeline):
         got = TEST_CLIENT.get_pipeline(self.pipeline.name).runtime_config["resources"]
         assert got == config
 
+    def test_support_bundle(self):
+        self.pipeline.start()
+
+        # Test getting support bundle as bytes
+        support_bundle_bytes = self.pipeline.support_bundle()
+        assert isinstance(support_bundle_bytes, bytes)
+        assert len(support_bundle_bytes) > 0
+
+        # Test that it's a valid ZIP file
+        try:
+            with zipfile.ZipFile(io.BytesIO(support_bundle_bytes), "r") as zip_file:
+                # Check that the ZIP file contains some files
+                file_list = zip_file.namelist()
+                assert len(file_list) > 0
+        except zipfile.BadZipFile:
+            self.fail("Support bundle is not a valid ZIP file")
+
+        # Test saving to file
+        with tempfile.NamedTemporaryFile(suffix=".zip", delete=False) as temp_file:
+            temp_path = temp_file.name
+
+        try:
+            # Test saving with explicit path
+            self.pipeline.support_bundle(output_path=temp_path)
+
+            # Verify the file was created and is a valid ZIP
+            assert os.path.exists(temp_path)
+            with zipfile.ZipFile(temp_path, "r") as zip_file:
+                file_list = zip_file.namelist()
+                assert len(file_list) > 0
+        finally:
+            # Clean up
+            if os.path.exists(temp_path):
+                os.unlink(temp_path)
+
 
 if __name__ == "__main__":
     unittest.main()