openrelik-api-client 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

openrelik_api_client/workflows.py

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2024-2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -25,7 +25,7 @@ class WorkflowsAPI:
         self.folders_url = f"{self.api_client.base_url}/folders"
 
     def create_workflow(
-        self, folder_id: int, file_ids: list, template_id: int = None
+        self, folder_id: int, file_ids: list, template_id: int = None, template_params: dict = {}
     ) -> int | None:
         """Creates a new workflow.
 
@@ -46,6 +46,7 @@ class WorkflowsAPI:
             "folder_id": folder_id,
             "file_ids": file_ids,
             "template_id": template_id,
+            "template_params": template_params,
         }
         response = self.api_client.session.post(endpoint, json=data)
         response.raise_for_status()
@@ -157,3 +158,21 @@ class WorkflowsAPI:
         if response.status_code == 200:
             workflow = response.json()
         return workflow
+
+    def get_workflow_report(self, workflow_id: int) -> dict[str, Any] | None:
+        """Retrieves a workflow report.
+
+        Args:
+            workflow_id: The ID of the workflow to retrieve the report from.
+
+        Returns:
+            The workflow report data.
+
+         Raises:
+            HTTPError: If the API request failed.
+        """
+        endpoint = f"{self.api_client.base_url}/workflows/{workflow_id}/report/"
+        response = self.api_client.session.get(endpoint)
+        response.raise_for_status()
+        if response.status_code == 200:
+            return response.json()

{openrelik_api_client-0.4.0.dist-info → openrelik_api_client-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openrelik-api-client
-Version: 0.4.0
+Version: 0.5.0
 Summary: API client that automatically handles token refresh
 Author: Johan Berggren
 Author-email: jberggren@gmail.com

{openrelik_api_client-0.4.0.dist-info → openrelik_api_client-0.5.0.dist-info}/RECORD

@@ -4,8 +4,8 @@ openrelik_api_client/configs.py,sha256=ns0AF6i9vZq8M-1EKEF4xZ-IpcvDPbTFxn1WMAe_P
 openrelik_api_client/files.py,sha256=0lWUsMIjUGsz9CGQJFchcfq3myBSaxSZCKPJ1oyMgWQ,8444
 openrelik_api_client/folders.py,sha256=6Hix2G4WoRPBMQu7HTeV_ZS_lZTj8pZW57NW80riZp4,8118
 openrelik_api_client/groups.py,sha256=zhR2oq9_PG0ZmFgVHf62T7KWeg7h6mJmXOABScOJGyU,6116
-openrelik_api_client/workflows.py,sha256=5DjpNuqGfbLz-glWPqqIZpL6wtHtt2DWsqG-EpFDeRw,5510
-openrelik_api_client-0.4.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-openrelik_api_client-0.4.0.dist-info/METADATA,sha256=65UwGhJtWFuEc_Hcfpf7ehjdUbAU6EPsfyxYc8eQQgU,2114
-openrelik_api_client-0.4.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-openrelik_api_client-0.4.0.dist-info/RECORD,,
+openrelik_api_client/workflows.py,sha256=BwIBSjS0krsJWglPiFdAAtcwRMqNmtWDyz9H7UDgHWE,6187
+openrelik_api_client-0.5.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+openrelik_api_client-0.5.0.dist-info/METADATA,sha256=pcSwYPyaRjtqxonmu6mMPr6YpnKdw4JDkerxXSnYgw8,2114
+openrelik_api_client-0.5.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+openrelik_api_client-0.5.0.dist-info/RECORD,,