Change how the exception handling will work (#2435)

cds-snc · Jan 30, 2025 · 83573c5 · 83573c5
1 parent 9f04693
commit 83573c5
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/app/celery/process_ses_receipts_tasks.py b/app/celery/process_ses_receipts_tasks.py
@@ -43,12 +43,12 @@ def process_ses_results(self, response):  # noqa: C901
         ses_message = json.loads(response["Message"])
         notification_type = ses_message["notificationType"]
 
-        if notification_type == "Complaint":
-            _check_and_queue_complaint_callback_task(*handle_complaint(ses_message))
-            return True
-
-        reference = ses_message["mail"]["messageId"]
         try:
+            if notification_type == "Complaint":
+                _check_and_queue_complaint_callback_task(*handle_complaint(ses_message))
+                return True
+
+            reference = ses_message["mail"]["messageId"]
             notification = notifications_dao.dao_get_notification_by_reference(reference)
         except NoResultFound:
             try: