Merge pull request #310 from sartography/feature/add_task_not_found_e…

…rror added an exception for task not found w/ burnettk
sartography · Apr 5, 2023 · 37d7ae6 · 37d7ae6
2 parents e1add83 + b98efbd
commit 37d7ae6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/SpiffWorkflow/exceptions.py b/SpiffWorkflow/exceptions.py
@@ -129,3 +129,7 @@ def __init__(self, error_msg, task=None, exception=None,
 
 class StorageException(SpiffWorkflowException):
     pass
+
+
+class TaskNotFoundException(WorkflowException):
+    pass
diff --git a/SpiffWorkflow/workflow.py b/SpiffWorkflow/workflow.py
@@ -23,10 +23,11 @@
 from .task import Task, TaskState
 from .util.compat import mutex
 from .util.event import Event
-from .exceptions import WorkflowException
+from .exceptions import TaskNotFoundException, WorkflowException
 
 logger = logging.getLogger('spiff')
 
+
 class Workflow(object):
 
     """
@@ -246,7 +247,7 @@ def get_task_from_id(self, task_id, tasklist=None):
             if task.id == task_id:
                 return task
         msg = 'A task with the given task_id (%s) was not found' % task_id
-        raise WorkflowException(msg, task_spec=self.spec)
+        raise TaskNotFoundException(msg, task_spec=self.spec)
 
     def run_task_from_id(self, task_id):
         """