Trying to get autoretry logic to work.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>

app/celery/process_ses_receipts_tasks.py

@@ -27,9 +27,10 @@ from app.utils import utc_now
 @notify_celery.task(
     bind=True,
     name="process-ses-result",
+    throws=(Exception,),
+    autoretry_for=(Exception,),
     max_retries=5,
     default_retry_delay=300,
-    autoretry_for=(Exception,),
 )
 def process_ses_results(self, response):
     try:
@@ -58,9 +59,11 @@ def process_ses_results(self, response):
                 reference
             )
         except NoResultFound:
+            print("&"*80)
             message_time = iso8601.parse_date(ses_message["mail"]["timestamp"]).replace(
                 tzinfo=None
             )
+            print("&"*80)
             if utc_now() - message_time < timedelta(minutes=5):
                 current_app.logger.info(
                     f"Notification not found for reference: {reference}"
@@ -68,6 +71,7 @@ def process_ses_results(self, response):
                     f"Callback may have arrived before notification was"
                     f"persisted to the DB. Adding task to retry queue"
                 )
+                print("&"*80)
                 raise
             else:
                 current_app.logger.warning(
@@ -113,7 +117,11 @@ def process_ses_results(self, response):
 
         return True
 
-    except Exception:
+    except Exception as e:
+        print("^"*80)
+        print(type(e))
+        print(e)
+        print("^"*80)
         current_app.logger.exception("Error processing SES results")
         raise
 

app/celery/provider_tasks.py

@@ -107,9 +107,11 @@ def check_sms_delivery_receipt(self, message_id, notification_id, sent_at):
         )
 
 
-def _deliver_sms_task_handler(func):
+def _deliver_sms_task_handler(cls):
     """Handle the max retries exceeded error case for delivering sms notifications."""
 
+    func = cls.__call__
+
     @wraps(func)
     def deliver_sms_task_wrapper(self, notification_id):
         try:
@@ -127,7 +129,9 @@ def _deliver_sms_task_handler(func):
             )
             raise NotificationTechnicalFailureException(message)
 
-    return deliver_sms_task_wrapper
+    cls.__call__ = deliver_sms_task_wrapper
+
+    return cls
 
 
 @_deliver_sms_task_handler
@@ -189,9 +193,11 @@ def deliver_sms(self, notification_id):
         raise
 
 
-def _deliver_email_task_handler(func):
+def _deliver_email_task_handler(cls):
     """Handle the max retries exceeded error case for delivering email notifications."""
 
+    func = cls.__call__
+
     @wraps(func)
     def deliver_email_task_wrapper(self, notification_id):
         try:
@@ -208,7 +214,9 @@ def _deliver_email_task_handler(func):
             )
             raise NotificationTechnicalFailureException(message)
 
-    return deliver_email_task_wrapper
+    cls.__call__ = deliver_email_task_wrapper
+
+    return cls
 
 
 @_deliver_email_task_handler

app/celery/service_callback_tasks.py

@@ -9,7 +9,10 @@ from app import encryption, notify_celery
 from app.utils import DATETIME_FORMAT
 
 
-def _send_to_service_task_handler(func):
+def _send_to_service_task_handler(cls):
+
+    func = cls.__call__
+
     @wraps(func)
     def send_to_service_task_wrapper(*args, **kwargs):
         sig = signature(func)
@@ -48,7 +51,9 @@ def _send_to_service_task_handler(func):
             )
             raise
 
-    return send_to_service_task_wrapper
+    cls.__call__ = send_to_service_task_wrapper
+
+    return cls
 
 
 @_send_to_service_task_handler