utils.py

from exceptions import MaxRetriesExceededException
from config import MAX_RETRIES
import logging


def retry(exceptions, max_retries=MAX_RETRIES):
    """
    Retry decorator that retries the wrapped function a maximum of 'max_retries' times if 'exception' is raised

    :param exceptions: Tuple
    :param max_retries:
    :return:
    """

    def deco_retry(f):
        def f_retry(*args, **kwargs):
            n_retries = 0
            while n_retries < max_retries:
                logging.info(f'Attempt {n_retries}: {f.__name__} for {args[1].name}')
                try:
                    return f(*args, **kwargs)
                except exceptions as e:
                    logging.error(f'{args[1].name}: {e} when calling {f.__name__}')
                    n_retries += 1
                    continue
            logging.error(f'{args[1].name} reached maximum number of retries on {f.__name__}')
            raise MaxRetriesExceededException

        return f_retry

    return deco_retry