fix: Fix enum str representation

canonical · Jan 13, 2025 · 97181a9 · 97181a9
1 parent 68fa4c6
commit 97181a9
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/spark8t/cli/service_account_registry.py b/spark8t/cli/service_account_registry.py
@@ -53,6 +53,13 @@ class Actions(str, Enum):
     PRIMARY = "get-primary"
     LIST = "list"
 
+    def __str__(self) -> str:
+        """Define string representation.
+
+        TODO(py310): replace inheritance with StrEnum once we drop py310
+        """
+        return str.__str__(self)
+
 
 def create_namespace_if_missing(kube_interface: AbstractKubeInterface, namespace: str):
     """Create namespace if does not exist."""

diff --git a/spark8t/domain.py b/spark8t/domain.py
@@ -352,3 +352,10 @@ class KubernetesResourceType(str, Enum):
     SECRET = "secret"
     SECRET_GENERIC = "secret generic"
     NAMESPACE = "namespace"
+
+    def __str__(self) -> str:
+        """Define string representation.
+
+        TODO(py310): replace inheritance with StrEnum once we drop py310
+        """
+        return str.__str__(self)
diff --git a/spark8t/services.py b/spark8t/services.py
@@ -1596,6 +1596,13 @@ class SparkDeployMode(str, Enum):
     CLIENT = "client"
     CLUSTER = "cluster"
 
+    def __str__(self) -> str:
+        """Define string representation.
+
+        TODO(py310): replace inheritance with StrEnum once we drop py310
+        """
+        return str.__str__(self)
+
 
 class SparkInterface(WithLogging):
     """Class for providing interfaces for spark commands."""