fix: add FUNCTION_STATUS_UNKOWN

Signed-off-by: Guilhem Barthes <[email protected]>
Substra · Sep 7, 2023 · 1a2c9b8 · 1a2c9b8
1 parent 469fb90
commit 1a2c9b8
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 15 deletions.
diff --git a/backend/api/migrations/0053_function_status.py b/backend/api/migrations/0053_function_status.py
@@ -15,13 +15,14 @@ class Migration(migrations.Migration):
             name="status",
             field=models.CharField(
                 choices=[
+                    ("FUNCTION_STATUS_UNKONWN", "Function Status Unkonwn"),
                     ("FUNCTION_STATUS_CREATED", "Function Status Created"),
                     ("FUNCTION_STATUS_BUILDING", "Function Status Building"),
                     ("FUNCTION_STATUS_READY", "Function Status Ready"),
                     ("FUNCTION_STATUS_CANCELED", "Function Status Canceled"),
                     ("FUNCTION_STATUS_FAILED", "Function Status Failed"),
                 ],
-                default="FUNCTION_STATUS_READY",
+                default="FUNCTION_STATUS_UNKONWN",
                 max_length=64,
             ),
             preserve_default=False,

diff --git a/backend/orchestrator/function_pb2.py b/backend/orchestrator/function_pb2.py
diff --git a/backend/orchestrator/function_pb2.pyi b/backend/orchestrator/function_pb2.pyi
@@ -47,19 +47,21 @@ class _FunctionStatus:
 
 class _FunctionStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FunctionStatus.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    FUNCTION_STATUS_CREATED: _FunctionStatus.ValueType  # 0
-    FUNCTION_STATUS_BUILDING: _FunctionStatus.ValueType  # 1
-    FUNCTION_STATUS_READY: _FunctionStatus.ValueType  # 2
-    FUNCTION_STATUS_CANCELED: _FunctionStatus.ValueType  # 3
-    FUNCTION_STATUS_FAILED: _FunctionStatus.ValueType  # 4
+    FUNCTION_STATUS_UNKNOWN: _FunctionStatus.ValueType  # 0
+    FUNCTION_STATUS_CREATED: _FunctionStatus.ValueType  # 1
+    FUNCTION_STATUS_BUILDING: _FunctionStatus.ValueType  # 2
+    FUNCTION_STATUS_READY: _FunctionStatus.ValueType  # 3
+    FUNCTION_STATUS_CANCELED: _FunctionStatus.ValueType  # 4
+    FUNCTION_STATUS_FAILED: _FunctionStatus.ValueType  # 5
 
 class FunctionStatus(_FunctionStatus, metaclass=_FunctionStatusEnumTypeWrapper): ...
 
-FUNCTION_STATUS_CREATED: FunctionStatus.ValueType  # 0
-FUNCTION_STATUS_BUILDING: FunctionStatus.ValueType  # 1
-FUNCTION_STATUS_READY: FunctionStatus.ValueType  # 2
-FUNCTION_STATUS_CANCELED: FunctionStatus.ValueType  # 3
-FUNCTION_STATUS_FAILED: FunctionStatus.ValueType  # 4
+FUNCTION_STATUS_UNKNOWN: FunctionStatus.ValueType  # 0
+FUNCTION_STATUS_CREATED: FunctionStatus.ValueType  # 1
+FUNCTION_STATUS_BUILDING: FunctionStatus.ValueType  # 2
+FUNCTION_STATUS_READY: FunctionStatus.ValueType  # 3
+FUNCTION_STATUS_CANCELED: FunctionStatus.ValueType  # 4
+FUNCTION_STATUS_FAILED: FunctionStatus.ValueType  # 5
 global___FunctionStatus = FunctionStatus
 
 @typing_extensions.final

diff --git a/backend/orchestrator/resources.py b/backend/orchestrator/resources.py
@@ -139,6 +139,7 @@ def from_grpc(cls, o: function_pb2.FunctionOutput) -> FunctionOutput:
 
 
 class FunctionStatus(AutoNameEnum):
+    FUNCTION_STATUS_UNKNOWN = enum.auto()
     FUNCTION_STATUS_CREATED = enum.auto()
     FUNCTION_STATUS_BUILDING = enum.auto()
     FUNCTION_STATUS_READY = enum.auto()