toolbox/tenacity/retry.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import logging

from tenacity import before_sleep_log, retry, retry_if_exception_type, stop_after_attempt, wait_fixed

logger = logging.getLogger("toolbox")


def retry_call(fn, *args, **kwargs):
    @retry(
        wait=wait_fixed(10),
        stop=stop_after_attempt(3),
        before_sleep=before_sleep_log(logger, logging.ERROR),
    )
    def wrapped():
        return fn(*args, **kwargs)
    return wrapped()


if __name__ == "__main__":
    pass