#69 - Overloading the method 'parse_message' for BoolEntityField class to provide a special behavior when parsing messages

Author: dotchetter

pyttman/core/entity_parsing/fields.py

@@ -2,6 +2,10 @@
 from abc import ABC
 from typing import Any, Sequence, Type
 
+from pyttman.core.entity_parsing.entity import Entity
+
+from pyttman.core.containers import MessageMixin, Message
+
 from pyttman.core.entity_parsing.identifiers import IntegerIdentifier, \
     Identifier
 from pyttman.core.entity_parsing.parsers import EntityFieldValueParser
@@ -155,6 +159,15 @@ def __init__(self, *args,
                  **kwargs):
         super().__init__(*args, valid_strings=message_contains, **kwargs)
 
+    def parse_message(self,
+                      message: MessageMixin,
+                      original_message_content: tuple[str],
+                      memoization: dict = None) -> None:
+        original_content = Message(original_message_content).lowered_content()
+        self.value = Entity(value=self.default, is_fallback_default=True)
+        if set(self.valid_strings).intersection(original_content):
+            self.value = Entity(value=True)
+
 
 if __name__ != "__main__":
     StringEntityField = TextEntityField