From 054bd2d7f1fc79311c57cf84ed004ff207fb5bd1 Mon Sep 17 00:00:00 2001 From: Cliff Hill Date: Fri, 22 Nov 2024 16:35:56 -0500 Subject: [PATCH] More fixes. Signed-off-by: Cliff Hill --- app/celery/tasks.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/app/celery/tasks.py b/app/celery/tasks.py index e533d6815..c84889a0a 100644 --- a/app/celery/tasks.py +++ b/app/celery/tasks.py @@ -358,13 +358,14 @@ def _save_api_task_handler(func): current_app.logger.exception( f"Max retry failed Failed to persist notification {notification['id']}", ) + raise return save_api_task_wrapper @_save_api_task_handler @notify_celery.task( - bind=True, name="save-api-email", max_retries=5, default_retry_delay=300 + bind=True, name="save-api-email", max_retries=5, default_retry_delay=300, autoretry_for=SQLAlchemyError, ) def save_api_email(self, encrypted_notification): save_api_email_or_sms(self, encrypted_notification) @@ -372,7 +373,7 @@ def save_api_email(self, encrypted_notification): @_save_api_task_handler @notify_celery.task( - bind=True, name="save-api-sms", max_retries=5, default_retry_delay=300 + bind=True, name="save-api-sms", max_retries=5, default_retry_delay=300, autoretry_for=SQLAlchemyError, ) def save_api_sms(self, encrypted_notification): save_api_email_or_sms(self, encrypted_notification) @@ -440,7 +441,7 @@ def _send_inbound_sms_to_service_handler(func): @_send_inbound_sms_to_service_handler @notify_celery.task( - bind=True, name="send-inbound-sms", max_retries=5, default_retry_delay=300 + bind=True, name="send-inbound-sms", max_retries=5, default_retry_delay=300, autoretry_for=RequestException, ) def send_inbound_sms_to_service(self, inbound_sms_id, service_id): inbound_api = get_service_inbound_api_for_service(service_id=service_id)