A simple Python retrying decorator

python_retry.md

A simple Python retrying decorator

Define:

import functools
import logging

logger = logging.getLogger(__name__)


def retry(round=3):
    """A simple retrying decorator"""
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            err = None
            for i in range(round):
                logger.debug(f'trying {func.__name__} round {i+1}/{round} ...')
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    err = e
                    continue
            raise err
        return wrapper
    return decorator

Usage:

import random

@retry(3)
def foo():
    assert random.random() > 0.5