decorator for retrying function in specific times

retry_decorator.py

def retry(attempt, raise_on_fail=False):
    def decorator(func):
        def wrapper(*args, **kw):
            att = 0
            last_except = None
            while att < attempt:
                try:
                    return func(*args, **kw)
                except Exception as e:
                    att += 1
                    last_except = e
            else:
                if raise_on_fail:
                    raise Exception('Hit retry threshold, failed for {0}' % str(last_except))
                return None
        return wrapper
    return decorator