retry.py

retry.py

import functools
import time


def retry(exception, max_retry, sleep, err_handler):
    """ Catching exception and retrying
    """
    def dec(f):
        @functools.wraps(f)
        def _wrapped(*args, **kwargs):
            for t in range(1, max_retry + 1):
                try:
                    return f(*args, **kwargs)
                except exception as e:
                    err_handler(e, t, max_retry)
                    time.sleep(sleep)
            raise MaxRetriedError
        return _wrapped
    return dec