Added whatsapp as a medium to send OTPs

hasgeek · Nov 10, 2023 · e52c7cc · e52c7cc
1 parent c7bcdf3
commit e52c7cc
Showing 1 changed file with 49 additions and 0 deletions.
diff --git a/funnel/views/otp.py b/funnel/views/otp.py
@@ -34,6 +34,7 @@
     TransportRecipientError,
     TransportTransactionError,
     sms,
+    whatsapp,
 )
 from ..transports.email import jsonld_view_action, send_email
 from ..utils import blake2b160_hex, mask_email, mask_phone
@@ -272,6 +273,30 @@ def send_sms(
             return msg
         return None
 
+    # Send whatsapp message
+    def send_whatsapp(
+        self, flash_success: bool = True, flash_failure: bool = True
+    ) -> str | None:
+        """Send an OTP via WhatsApp to a phone number."""
+        if not self.phone:
+            return None
+        message = whatsapp.OTPTemplate(self.otp)
+        try:
+            whatsapp.send_via_meta(self.phone, message)
+        except TransportRecipientError as exc:
+            if flash_failure:
+                flash(str(exc), 'error')
+            else:
+                raise
+        if flash_success:
+            flash(
+                _("An OTP has been sent to your phone number {number}").format(
+                    number=self.display_phone
+                ),
+                'success',
+            )
+        return message
+
     def send_email(
         self, flash_success: bool = True, flash_failure: bool = True
     ) -> str | None:
@@ -388,6 +413,30 @@ def send_sms(
             return msg
         return None
 
+    # Send whatsapp message
+    def send_whatsapp(
+        self, flash_success: bool = True, flash_failure: bool = True
+    ) -> str | None:
+        """Send an OTP via WhatsApp to a phone number."""
+        if not self.phone:
+            return None
+        message = whatsapp.OTPTemplate(self.otp)
+        try:
+            whatsapp.send_via_meta(self.phone, message)
+        except TransportRecipientError as exc:
+            if flash_failure:
+                flash(str(exc), 'error')
+            else:
+                raise
+        if flash_success:
+            flash(
+                _("An OTP has been sent to your phone number {number}").format(
+                    number=self.display_phone
+                ),
+                'success',
+            )
+        return message
+
     def send_email(
         self, flash_success: bool = True, flash_failure: bool = True
     ) -> str | None: