utils/retry.py

import time
import functools

def retry(exceptions, tries=3, delay=2, backoff=2):
    """
    Retry decorator with exponential backoff.

    Args:
        exceptions (tuple): exceptions to catch and retry on.
        tries (int): number of attempts.
        delay (int): initial delay between retries.
        backoff (int): multiplier for delay.

    Usage:
        @retry((TimeoutError, SomeOtherError))
        def func(...):
            ...
    """
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            _tries, _delay = tries, delay
            while _tries > 1:
                try:
                    return func(*args, **kwargs)
                except exceptions as e:
                    print(f"Warning: {e}, retrying in {_delay} seconds...")
                    time.sleep(_delay)
                    _tries -= 1
                    _delay *= backoff
            return func(*args, **kwargs)
        return wrapper
    return decorator