Merge pull request #14 from openimis/enhancement/pythonHandlerError

Added exception for python worflow handler

workflow/exceptions.py

@@ -0,0 +1,17 @@
+
+class PythonWorkflowHandlerException(Exception):
+    """
+    Custom exception class for handling errors within Python Workflows.
+    """
+
+    def __init__(self, message, *args, **kwargs):
+        if args or kwargs:
+            message = message.format(*args, **kwargs)
+        super().__init__(message)
+        self.message = message
+
+    def __str__(self):
+        """
+        Return the string representation of the exception.
+        """
+        return f"PythonWorkflowHandlerException: {self.message}"

workflow/systems/python/handler.py

@@ -1,6 +1,7 @@
 import logging
 from typing import Dict
 
+from workflow.exceptions import PythonWorkflowHandlerException
 from workflow.systems.base import WorkflowHandler
 from workflow.util import result
 
@@ -19,6 +20,14 @@ def run(self, data: Dict):
                 success=True,
                 data=output
             )
+        # Dedicated exception handling for workflow exceptions that should have custom message
+        except PythonWorkflowHandlerException as e:
+            logging.error("Error while executing workflow %s-%s: ", self.system, self.name, exc_info=e)
+            return result(
+                success=False,
+                message=e.message,
+                details=str(e)
+            )
         except Exception as e:
             logging.error("Error while executing workflow %s-%s: ", self.system, self.name, exc_info=e)
             return result(