revamp how authentication code is displayed as per Steven

app/celery/provider_tasks.py

@@ -1,3 +1,4 @@
+import os
 from datetime import datetime, timedelta
 from time import time
 
@@ -23,7 +24,12 @@ from app.models import (
 )
 
 
-@notify_celery.task(bind=True, name="check_sms_delivery_receipt", max_retries=48, default_retry_delay=300)
+@notify_celery.task(
+    bind=True,
+    name="check_sms_delivery_receipt",
+    max_retries=48,
+    default_retry_delay=300,
+)
 def check_sms_delivery_receipt(self, message_id, notification_id, sent_at):
     """
     This is called after deliver_sms to check the status of the message. This uses the same number of
@@ -36,48 +42,69 @@ def check_sms_delivery_receipt(self, message_id, notification_id, sent_at):
     """
     # TODO the localstack cloudwatch doesn't currently have our log groups.  Possibly create them with awslocal?
     if aws_cloudwatch_client.is_localstack():
-        status = 'success'
-        provider_response = 'this is a fake successful localstack sms message'
+        status = "success"
+        provider_response = "this is a fake successful localstack sms message"
     else:
-        status, provider_response = aws_cloudwatch_client.check_sms(message_id, notification_id, sent_at)
+        status, provider_response = aws_cloudwatch_client.check_sms(
+            message_id, notification_id, sent_at
+        )
 
-    if status == 'success':
+    if status == "success":
         status = NOTIFICATION_DELIVERED
-    elif status == 'failure':
+    elif status == "failure":
         status = NOTIFICATION_FAILED
     # if status is not success or failure the client raised an exception and this method will retry
 
     if status == NOTIFICATION_DELIVERED:
         sanitize_successful_notification_by_id(notification_id)
-        current_app.logger.info(f"Sanitized notification {notification_id} that was successfully delivered")
+        current_app.logger.info(
+            f"Sanitized notification {notification_id} that was successfully delivered"
+        )
     else:
-        update_notification_status_by_id(notification_id, status, provider_response=provider_response)
-        current_app.logger.info(f"Updated notification {notification_id} with response '{provider_response}'")
+        update_notification_status_by_id(
+            notification_id, status, provider_response=provider_response
+        )
+        current_app.logger.info(
+            f"Updated notification {notification_id} with response '{provider_response}'"
+        )
 
 
-@notify_celery.task(bind=True, name="deliver_sms", max_retries=48, default_retry_delay=300)
+@notify_celery.task(
+    bind=True, name="deliver_sms", max_retries=48, default_retry_delay=300
+)
 def deliver_sms(self, notification_id):
     try:
         # Get the time we are doing the sending, to minimize the time period we need to check over for receipt
         now = round(time() * 1000)
-        current_app.logger.info("Start sending SMS for notification id: {}".format(notification_id))
+        current_app.logger.info(
+            "Start sending SMS for notification id: {}".format(notification_id)
+        )
         notification = notifications_dao.get_notification_by_id(notification_id)
+        ansi_green = "\033[32m"
+        ansi_reset = "\033[0m"
+
         if not notification:
             raise NoResultFound()
+        if (
+            os.getenv("NOTIFY_ENVIRONMENT") == "development"
+            and "authentication code" in notification.content
+        ):
+            current_app.logger.warning(
+                ansi_green + f"AUTHENTICATION CODE: {notification.content}" + ansi_reset
+            )
+
         message_id = send_to_providers.send_sms_to_provider(notification)
         # We have to put it in UTC.  For other timezones, the delay
         # will be ignored and it will fire immediately (although this probably only affects developer testing)
         my_eta = datetime.utcnow() + timedelta(seconds=300)
         check_sms_delivery_receipt.apply_async(
-            [message_id, notification_id, now],
-            eta=my_eta,
-            queue=QueueNames.CHECK_SMS
+            [message_id, notification_id, now], eta=my_eta, queue=QueueNames.CHECK_SMS
         )
     except Exception as e:
         if isinstance(e, SmsClientResponseException):
             current_app.logger.warning(
                 "SMS notification delivery for id: {} failed".format(notification_id),
-                exc_info=True
+                exc_info=True,
             )
         else:
             current_app.logger.exception(
@@ -90,16 +117,26 @@ def deliver_sms(self, notification_id):
             else:
                 self.retry(queue=QueueNames.RETRY)
         except self.MaxRetriesExceededError:
-            message = "RETRY FAILED: Max retries reached. The task send_sms_to_provider failed for notification {}. " \
-                      "Notification has been updated to technical-failure".format(notification_id)
-            update_notification_status_by_id(notification_id, NOTIFICATION_TECHNICAL_FAILURE)
+            message = (
+                "RETRY FAILED: Max retries reached. The task send_sms_to_provider failed for notification {}. "
+                "Notification has been updated to technical-failure".format(
+                    notification_id
+                )
+            )
+            update_notification_status_by_id(
+                notification_id, NOTIFICATION_TECHNICAL_FAILURE
+            )
             raise NotificationTechnicalFailureException(message)
 
 
-@notify_celery.task(bind=True, name="deliver_email", max_retries=48, default_retry_delay=300)
+@notify_celery.task(
+    bind=True, name="deliver_email", max_retries=48, default_retry_delay=300
+)
 def deliver_email(self, notification_id):
     try:
-        current_app.logger.info("Start sending email for notification id: {}".format(notification_id))
+        current_app.logger.info(
+            "Start sending email for notification id: {}".format(notification_id)
+        )
         notification = notifications_dao.get_notification_by_id(notification_id)
         if not notification:
             raise NoResultFound()
@@ -108,7 +145,7 @@ def deliver_email(self, notification_id):
         current_app.logger.exception(
             f"Email notification {notification_id} failed: {e}"
         )
-        update_notification_status_by_id(notification_id, 'technical-failure')
+        update_notification_status_by_id(notification_id, "technical-failure")
     except Exception as e:
         try:
             if isinstance(e, AwsSesClientThrottlingSendRateException):
@@ -122,8 +159,14 @@ def deliver_email(self, notification_id):
 
             self.retry(queue=QueueNames.RETRY)
         except self.MaxRetriesExceededError:
-            message = "RETRY FAILED: Max retries reached. " \
-                      "The task send_email_to_provider failed for notification {}. " \
-                      "Notification has been updated to technical-failure".format(notification_id)
-            update_notification_status_by_id(notification_id, NOTIFICATION_TECHNICAL_FAILURE)
+            message = (
+                "RETRY FAILED: Max retries reached. "
+                "The task send_email_to_provider failed for notification {}. "
+                "Notification has been updated to technical-failure".format(
+                    notification_id
+                )
+            )
+            update_notification_status_by_id(
+                notification_id, NOTIFICATION_TECHNICAL_FAILURE
+            )
             raise NotificationTechnicalFailureException(message)