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

python/cpython#100458

Author: taro-kayo

statelint/fields/common.py

@@ -1,9 +1,9 @@
 import json
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 
-class Comparison(str, Enum):
+class Comparison(StrEnum):
     STRING_EQUALS = "StringEquals"
     STRING_LESS_THAN = "StringLessThan"
     STRING_GREATER_THAN = "StringGreaterThan"
@@ -45,7 +45,7 @@ class Comparison(str, Enum):
     STRING_MATCHES = "StringMatches"
 
 
-class StateType(str, Enum):
+class StateType(StrEnum):
     PASS = "Pass"
     SUCCEED = "Succeed"
     FAIL = "Fail"

statelint/problem/problem_type.py

@@ -1,7 +1,7 @@
-from enum import Enum
+from enum import StrEnum
 
 
-class ProblemType(str, Enum):
+class ProblemType(StrEnum):
     # type check
     FLOAT = "a Float"
     STRING = "a String"