fix: Enum with str or int Mixin Breaking Change in Python 3.11

python/cpython#100458
taro-kayo · May 27, 2023 · 8183604 · 8183604
1 parent 2010ba0
commit 8183604
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/statelint/fields/common.py b/statelint/fields/common.py
@@ -4,6 +4,9 @@
 
 
 class Comparison(str, Enum):
+    def __str__(self) -> str:
+        return self.value
+
     STRING_EQUALS = "StringEquals"
     STRING_LESS_THAN = "StringLessThan"
     STRING_GREATER_THAN = "StringGreaterThan"
@@ -46,6 +49,9 @@ class Comparison(str, Enum):
 
 
 class StateType(str, Enum):
+    def __str__(self) -> str:
+        return self.value
+
     PASS = "Pass"
     SUCCEED = "Succeed"
     FAIL = "Fail"

diff --git a/statelint/problem/problem_type.py b/statelint/problem/problem_type.py
@@ -2,6 +2,9 @@
 
 
 class ProblemType(str, Enum):
+    def __str__(self) -> str:
+        return self.value
+
     # type check
     FLOAT = "a Float"
     STRING = "a String"