create WhatsAppTextOnlyFreetext

vaccine/states.py

@@ -216,7 +216,6 @@ def __init__(
         override_answer_name: Optional[str] = None,
         footer: Optional[str] = None,
         header: Optional[str] = None,
-        text_only_error: Optional[str] = None,
     ):
         self.app = app
         self.question = question
@@ -226,12 +225,8 @@ def __init__(
         self.override_answer_name = override_answer_name
         self.footer = footer
         self.header = header
-        self.text_only_error = text_only_error
 
     async def process_message(self, message: Message):
-        if self.text_only_error is not None and message.transport_metadata:
-            if message.transport_metadata.get("message", {}).get("type") != "text":
-                return self.app.send_message(self.text_only_error)
         if self.check is not None:
             if not isinstance(self.check, list):
                 self.check = [self.check]
@@ -259,6 +254,23 @@ async def display(self, message: Message):
         return self.app.send_message(text, helper_metadata=helper_metadata)
 
 
+class WhatsAppTextOnlyFreetext(FreeText):
+    def __init__(
+        self,
+        *args,
+        text_only_error: Optional[str] = None,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+        self.text_only_error = text_only_error
+
+    async def process_message(self, message: Message):
+        if self.text_only_error is not None and message.transport_metadata:
+            if message.transport_metadata.get("message", {}).get("type") != "text":
+                return self.app.send_message(self.text_only_error)
+        return await super().process_message(message)
+
+
 class BaseWhatsAppChoiceState(ChoiceState):
     """
     Base class for all whatsapp choice states.

yal/surveys/location.py

@@ -4,9 +4,9 @@
 from vaccine.states import (
     Choice,
     EndState,
-    FreeText,
     WhatsAppButtonState,
     WhatsAppListState,
+    WhatsAppTextOnlyFreetext,
 )
 from vaccine.validators import nonempty_validator
 from yal import config, contentrepo, rapidpro
@@ -258,7 +258,7 @@ async def state_location_name_city(self):
                 ]
             )
         )
-        return FreeText(
+        return WhatsAppTextOnlyFreetext(
             self,
             question=question,
             next="state_location_area_type",