Add contains to enum

huridocs · Oct 20, 2023 · cfad32a · cfad32a
1 parent 89499a3
commit cfad32a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/pdf_token_type_labels/TaskMistakesType.py b/src/pdf_token_type_labels/TaskMistakesType.py
@@ -5,3 +5,7 @@ class TaskMistakesType(Enum):
     CORRECT = "CORRECT"
     WRONG = "WRONG"
     MISSING = "MISSING"
+
+    @staticmethod
+    def contains(key: str):
+        return key.upper() in [e.value for e in TaskMistakesType]