feat: add Waiting for dependencies and paused statuses

aneoconsulting · Sep 2, 2024 · 4a1ebb1 · 4a1ebb1
1 parent f49315d
commit 4a1ebb1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Protos/V1/task_status.proto b/Protos/V1/task_status.proto
@@ -23,4 +23,6 @@ enum TaskStatus {
   TASK_STATUS_PROCESSING = 9; /** Task is being processed. */
   TASK_STATUS_PROCESSED = 10; /** Task is processed. */
   TASK_STATUS_RETRIED = 11; /** Task is retried. */
+  TASK_STATUS_WAITING_FOR_DEPENDENCIES = 12; /** Task is waiting for its dependencies before becoming executable. */
+  TASK_STATUS_PAUSED = 11; /** Task is paused and will not be executed until session is resumed. */
 }
diff --git a/packages/python/src/armonik/common/enumwrapper.py b/packages/python/src/armonik/common/enumwrapper.py
@@ -14,6 +14,8 @@
     TASK_STATUS_PROCESSING,
     TASK_STATUS_PROCESSED,
     TASK_STATUS_RETRIED,
+    TASK_STATUS_WAITING_FOR_DEPENDENCIES,
+    TASK_STATUS_PAUSED,
 )
 from ..protogen.common.events_common_pb2 import (
     EVENTS_ENUM_UNSPECIFIED,
@@ -69,7 +71,8 @@ class TaskStatus(IntEnum):
     PROCESSING = TASK_STATUS_PROCESSING
     PROCESSED = TASK_STATUS_PROCESSED
     RETRIED = TASK_STATUS_RETRIED
-
+    WAITING_FOR_DEPENDENCIES = TASK_STATUS_WAITING_FOR_DEPENDENCIES
+    PAUSED = TASK_STATUS_PAUSED
 
 class EventTypes:
     UNSPECIFIED = EVENTS_ENUM_UNSPECIFIED