Trying to get autoretry logic to work.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-12-03 11:24:09 -05:00
parent 00908593dd
commit 8edc8b04a7
4 changed files with 47 additions and 11 deletions

View File

@@ -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