python-snippets

retry.py

"""
Retry a function after `sleep` secionds for a max of `maxtries`
"""

def retry(sleep=0.1, maxtries=10):
    # wrapper for sleep-retrying a function call. use with `@retry()`
    from time import sleep as _sleep
    def wrap1(f):
        def wrap2(*a, __count=0, **kw):
            try:
                return f(*a, **kw)
            except Exception as e: 
                if __count > maxtries:
                    raise e from None # ignores stack trace
                _sleep(sleep)
                wrap2(*a, __count=__count+1, **kw)
        return wrap2
    return wrap1