Allowing re-execution of tasks based on a reprocessing the message

wayfair-incubator · Oct 25, 2023 · 2f23f76 · 2f23f76
1 parent e61d726
commit 2f23f76
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/dagger/tasks/task.py b/dagger/tasks/task.py
@@ -322,15 +322,21 @@ async def start(
         ignore_status: bool = False,
     ) -> None:
         # pre-execute
-        if self.status.code in [
-            TaskStatusEnum.COMPLETED.name,
-            TaskStatusEnum.SKIPPED.name,
-        ] and not self.reprocess_on_message:
+        if (
+            self.status.code
+            in [
+                TaskStatusEnum.COMPLETED.name,
+                TaskStatusEnum.SKIPPED.name,
+            ]
+            and not self.reprocess_on_message
+        ):
             return await self.on_complete(
                 status=self.status, workflow_instance=workflow_instance
             )
         if (
-            ignore_status or self.status.code == TaskStatusEnum.NOT_STARTED.name or self.reprocess_on_message
+            ignore_status
+            or self.status.code == TaskStatusEnum.NOT_STARTED.name
+            or self.reprocess_on_message
         ) and workflow_instance:
             self.status = TaskStatus(
                 code=TaskStatusEnum.EXECUTING.name, value=TaskStatusEnum.EXECUTING.value
@@ -964,10 +970,14 @@ async def stop(
         )
 
     async def start(self, workflow_instance: Optional[ITemplateDAGInstance]) -> None:
-        if self.status.code in [
-            TaskStatusEnum.COMPLETED.name,
-            TaskStatusEnum.SKIPPED.name,
-        ] and not self.reprocess_on_message:
+        if (
+            self.status.code
+            in [
+                TaskStatusEnum.COMPLETED.name,
+                TaskStatusEnum.SKIPPED.name,
+            ]
+            and not self.reprocess_on_message
+        ):
             return await self.on_complete(
                 status=self.status, workflow_instance=workflow_instance
             )