reformat

app/celery/service_callback_tasks.py

@@ -8,102 +8,106 @@ from app.config import QueueNames
 from app.utils import DATETIME_FORMAT
 
 
-@notify_celery.task(bind=True, name="send-delivery-status", max_retries=5, default_retry_delay=300)
-def send_delivery_status_to_service(
-    self, notification_id, encrypted_status_update
-):
+@notify_celery.task(
+    bind=True, name="send-delivery-status", max_retries=5, default_retry_delay=300
+)
+def send_delivery_status_to_service(self, notification_id, encrypted_status_update):
     status_update = encryption.decrypt(encrypted_status_update)
 
     data = {
         "id": str(notification_id),
-        "reference": status_update['notification_client_reference'],
-        "to": status_update['notification_to'],
-        "status": status_update['notification_status'],
-        "created_at": status_update['notification_created_at'],
-        "completed_at": status_update['notification_updated_at'],
-        "sent_at": status_update['notification_sent_at'],
-        "notification_type": status_update['notification_type'],
-        "template_id": status_update['template_id'],
-        "template_version": status_update['template_version']
+        "reference": status_update["notification_client_reference"],
+        "to": status_update["notification_to"],
+        "status": status_update["notification_status"],
+        "created_at": status_update["notification_created_at"],
+        "completed_at": status_update["notification_updated_at"],
+        "sent_at": status_update["notification_sent_at"],
+        "notification_type": status_update["notification_type"],
+        "template_id": status_update["template_id"],
+        "template_version": status_update["template_version"],
     }
 
     _send_data_to_service_callback_api(
         self,
         data,
-        status_update['service_callback_api_url'],
-        status_update['service_callback_api_bearer_token'],
-        'send_delivery_status_to_service'
+        status_update["service_callback_api_url"],
+        status_update["service_callback_api_bearer_token"],
+        "send_delivery_status_to_service",
     )
 
 
-@notify_celery.task(bind=True, name="send-complaint", max_retries=5, default_retry_delay=300)
+@notify_celery.task(
+    bind=True, name="send-complaint", max_retries=5, default_retry_delay=300
+)
 def send_complaint_to_service(self, complaint_data):
     complaint = encryption.decrypt(complaint_data)
 
     data = {
-        "notification_id": complaint['notification_id'],
-        "complaint_id": complaint['complaint_id'],
-        "reference": complaint['reference'],
-        "to": complaint['to'],
-        "complaint_date": complaint['complaint_date']
+        "notification_id": complaint["notification_id"],
+        "complaint_id": complaint["complaint_id"],
+        "reference": complaint["reference"],
+        "to": complaint["to"],
+        "complaint_date": complaint["complaint_date"],
     }
 
     _send_data_to_service_callback_api(
         self,
         data,
-        complaint['service_callback_api_url'],
-        complaint['service_callback_api_bearer_token'],
-        'send_complaint_to_service'
+        complaint["service_callback_api_url"],
+        complaint["service_callback_api_bearer_token"],
+        "send_complaint_to_service",
     )
 
 
-def _send_data_to_service_callback_api(self, data, service_callback_url, token, function_name):
-    notification_id = (data["notification_id"] if "notification_id" in data else data["id"])
+def _send_data_to_service_callback_api(
+    self, data, service_callback_url, token, function_name
+):
+    notification_id = (
+        data["notification_id"] if "notification_id" in data else data["id"]
+    )
     try:
         response = request(
             method="POST",
             url=service_callback_url,
             data=json.dumps(data),
             headers={
-                'Content-Type': 'application/json',
-                'Authorization': 'Bearer {}'.format(token)
+                "Content-Type": "application/json",
+                "Authorization": "Bearer {}".format(token),
             },
-            timeout=5
+            timeout=5,
+        )
+        current_app.logger.info(
+            "{} sending {} to {}, response {}".format(
+                function_name,
+                notification_id,
+                service_callback_url,
+                response.status_code,
+            )
         )
-        current_app.logger.info('{} sending {} to {}, response {}'.format(
-            function_name,
-            notification_id,
-            service_callback_url,
-            response.status_code
-        ))
         response.raise_for_status()
     except RequestException as e:
         current_app.logger.warning(
             "{} request failed for notification_id: {} and url: {}. exception: {}".format(
-                function_name,
-                notification_id,
-                service_callback_url,
-                e
+                function_name, notification_id, service_callback_url, e
             )
         )
-        if not isinstance(e, HTTPError) or e.response.status_code >= 500 or e.response.status_code == 429:
+        if (
+            not isinstance(e, HTTPError)
+            or e.response.status_code >= 500
+            or e.response.status_code == 429
+        ):
             try:
                 self.retry(queue=QueueNames.CALLBACKS_RETRY)
             except self.MaxRetriesExceededError:
                 current_app.logger.warning(
                     "Retry: {} has retried the max num of times for callback url {} and notification_id: {}".format(
-                        function_name,
-                        service_callback_url,
-                        notification_id
+                        function_name, service_callback_url, notification_id
                     )
                 )
         else:
             current_app.logger.warning(
                 "{} callback is not being retried for notification_id: {} and url: {}. exception: {}".format(
-                    function_name,
-                    notification_id,
-                    service_callback_url,
-                    e
+                    function_name, notification_id, service_callback_url, e
                 )
             )
 
@@ -116,9 +120,12 @@ def create_delivery_status_callback_data(notification, service_callback_api):
         "notification_status": notification.status,
         "notification_provider_response": notification.provider_response,  # TODO do we test for provider_response?
         "notification_created_at": notification.created_at.strftime(DATETIME_FORMAT),
-        "notification_updated_at":
-            notification.updated_at.strftime(DATETIME_FORMAT) if notification.updated_at else None,
-        "notification_sent_at": notification.sent_at.strftime(DATETIME_FORMAT) if notification.sent_at else None,
+        "notification_updated_at": notification.updated_at.strftime(DATETIME_FORMAT)
+        if notification.updated_at
+        else None,
+        "notification_sent_at": notification.sent_at.strftime(DATETIME_FORMAT)
+        if notification.sent_at
+        else None,
         "notification_type": notification.notification_type,
         "service_callback_api_url": service_callback_api.url,
         "service_callback_api_bearer_token": service_callback_api.bearer_token,
@@ -128,7 +135,9 @@ def create_delivery_status_callback_data(notification, service_callback_api):
     return encryption.encrypt(data)
 
 
-def create_complaint_callback_data(complaint, notification, service_callback_api, recipient):
+def create_complaint_callback_data(
+    complaint, notification, service_callback_api, recipient
+):
     data = {
         "complaint_id": str(complaint.id),
         "notification_id": str(notification.id),