feat: turn enumwrappers into real python enumerators (#570)

aneoconsulting · Nov 21, 2024 · 334b694 · 334b694
2 parents 90f0b51 + 12a884e
commit 334b694
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/packages/python/src/armonik/common/enumwrapper.py b/packages/python/src/armonik/common/enumwrapper.py
@@ -75,7 +75,7 @@ class TaskStatus(IntEnum):
     PAUSED = TASK_STATUS_PAUSED
 
 
-class EventTypes:
+class EventTypes(IntEnum):
     UNSPECIFIED = EVENTS_ENUM_UNSPECIFIED
     NEW_TASK = EVENTS_ENUM_NEW_TASK
     TASK_STATUS_UPDATE = EVENTS_ENUM_TASK_STATUS_UPDATE
@@ -88,7 +88,7 @@ def from_string(cls, name: str):
         return getattr(cls, name.upper())
 
 
-class SessionStatus:
+class SessionStatus(IntEnum):
     @staticmethod
     def name_from_value(status: RawSessionStatus) -> str:
         return _SESSIONSTATUS.values_by_number[status].name
@@ -102,7 +102,7 @@ def name_from_value(status: RawSessionStatus) -> str:
     DELETED = SESSION_STATUS_DELETED
 
 
-class ResultStatus:
+class ResultStatus(IntEnum):
     @staticmethod
     def name_from_value(status: RawResultStatus) -> str:
         return _RESULTSTATUS.values_by_number[status].name
@@ -115,19 +115,19 @@ def name_from_value(status: RawResultStatus) -> str:
     NOTFOUND = RESULT_STATUS_NOTFOUND
 
 
-class ServiceHealthCheckStatus:
+class ServiceHealthCheckStatus(IntEnum):
     UNSPECIFIED = HEALTH_STATUS_ENUM_UNSPECIFIED
     HEALTHY = HEALTH_STATUS_ENUM_HEALTHY
     DEGRADED = HEALTH_STATUS_ENUM_DEGRADED
     UNHEALTHY = HEALTH_STATUS_ENUM_UNHEALTHY
 
 
-class HealthCheckStatus:
+class HealthCheckStatus(IntEnum):
     UNKNOWN = HealthCheckReply.UNKNOWN
     SERVING = HealthCheckReply.SERVING
     NOT_SERVING = HealthCheckReply.NOT_SERVING
 
 
-class Direction:
+class Direction(IntEnum):
     ASC = SORT_DIRECTION_ASC
     DESC = SORT_DIRECTION_DESC
diff --git a/packages/python/src/armonik/common/objects.py b/packages/python/src/armonik/common/objects.py
@@ -64,7 +64,7 @@ def from_message(cls, task_options):
             application_namespace=task_options.application_namespace,
             application_service=task_options.application_service,
             engine_type=task_options.engine_type,
-            options=task_options.options,
+            options={k: v for k, v in task_options.options.items()},
         )
 
     def to_message(self) -> RawTaskOptions: