django retry decorator for tests

django_retry_decorator.py

import unittest
from django.utils.functional import wraps


def retry(tries=1, ignore_exception=None):
  """
  Decorator to retry a test.
  :param ignore_exception: Exception class to catch.
  :param tries: number of times to try running the wrapped function.
  Don't use if you need to do something with the return value.
  """
  def decorator(func):
    @wraps(func)
    def wrapped_func(*args, **kwargs):
      num_tries = max(1, tries)
      while num_tries > 0:
        try:
          return func(*args, **kwargs)
        except ignore_exception as e:
          num_tries -= 1
          if num_tries == 0:
            raise unittest.SkipTest('Skipping test; raised exception %s', e)
    return wrapped_func
  return decorator