add max retry func

Author: Alexey Tsitkin

auto_tasks/exceptions.py

@@ -1,10 +1,11 @@
 class RetryableTaskException(Exception):
-    def __init__(self, inner, delay=None, count_retries=None):
-        # type: (Exception, int, bool) -> None
+    def __init__(self, inner, delay=None, count_retries=None, max_retries_func=None):
+        # type: (Exception, int, bool, Any) -> None
         self._inner = inner
 
         self.delay = delay
         self.count_retries = count_retries
+        self.max_retries_func = max_retries_func
 
     def __repr__(self):
         # type: () -> str

auto_tasks/service.py

@@ -39,7 +39,11 @@ def _auto_task_wrapper(module_name, class_name, func_name, *args, **kwargs):
 
             _auto_task_wrapper.retry(**retry_kwargs)
         except MaxRetriesReachedException:
-            logger.error('Reached max retries in auto task {}.{}.{} with error: {}'.format(module_name, class_name, func_name, repr(exc)))
+            if exc.max_retries_func:
+                exc.max_retries_func()
+            else:
+                # by default log an error
+                logger.error('Reached max retries in auto task {}.{}.{} with error: {}'.format(module_name, class_name, func_name, repr(exc)))
 
 
 class AutoTaskService(BaseAutoTaskService):

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name='django-eb-sqs',
-    version='1.31',
+    version='1.32',
     package_dir={'eb_sqs': 'eb_sqs'},
     include_package_data=True,
     packages=find_packages(),