retry_with_backoff

The oracle · bundled tests

from retry_with_backoff import retry

def test_succeeds_after_failures():
    calls = {'n': 0}
    @retry(times=3, base=0)
    def f():
        calls['n'] += 1
        if calls['n'] < 2: raise ValueError()
        return 'ok'
    assert f() == 'ok' and calls['n'] == 2

The artifact

import time, functools

def retry(times: int = 3, base: float = 0.1):
    def deco(fn):
        @functools.wraps(fn)
        def wrap(*a, **k):
            for i in range(times):
                try:
                    return fn(*a, **k)
                except Exception:
                    if i == times - 1:
                        raise
                    time.sleep(base * (2 ** i))
        return wrap
    return deco