From 075b0a544cdd8ac43fe5696672d72b6aef6d917e Mon Sep 17 00:00:00 2001
From: Felix Stupp <felix.stupp@outlook.com>
Date: Sun, 7 Nov 2021 01:58:08 +0100
Subject: [PATCH] extractors.SuitableLevel: Added methods for fancy tenary's

---
 server/entertainment_decider/extractors/generic.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/server/entertainment_decider/extractors/generic.py b/server/entertainment_decider/extractors/generic.py
index 32952ee..996278a 100644
--- a/server/entertainment_decider/extractors/generic.py
+++ b/server/entertainment_decider/extractors/generic.py
@@ -27,6 +27,18 @@ class SuitableLevel(Enum):
     def accept_immediately(self):
         return self.value[1]
 
+    @staticmethod
+    def always_or_no(value: bool) -> SuitableLevel:
+        return SuitableLevel.ALWAYS if value else SuitableLevel.NO
+
+    @staticmethod
+    def always_or_fallback(value: bool) -> SuitableLevel:
+        return SuitableLevel.ALWAYS if value else SuitableLevel.FALLBACK
+
+    @staticmethod
+    def fallback_or_no(value: bool) -> SuitableLevel:
+        return SuitableLevel.FALLBACK if value else SuitableLevel.NO
+
 
 class ExtractionError(Exception):
     pass