Python retry的一个AOP实现 (Python中级)

最新推荐文章于 2025-08-22 23:47:35 发布

weixin_30326515

最新推荐文章于 2025-08-22 23:47:35 发布

阅读量90

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/velly/p/4635093.html

本文介绍如何使用Python实现一个自定义的重试机制，包括条件判断、重试次数、等待间隔和错误忽略等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

def retry(condition, retry_times, interval=60, ignore_error=True):
    """
    @condition: condition variable or func.
    @retry_times: you know it means.
    @interval: wait time before retrying
    @param: ignore_error, retry if error happened.
    """
    def wrap(f):
        def wrapped_f(*args):
            __retry_times = retry_times - 1
            __interval = interval
            __ignore_error = ignore_error
            __does_error_happened = False
            if not hasattr(f, '__call__'):
                raise RetryException("func is not a callable.")
            if type(__retry_times) != int:
                raise RetryException("retry_times is not a number.")
            if __retry_times < 0:
                __retry_times = 0

            def get_condition():
                if __ignore_error and __does_error_happened:
                    return True
                elif hasattr(condition, '__call__'):
                    return condition()
                else:
                    return condition

            while get_condition() and __retry_times >= 0:
                print("Retry times remain -> ", __retry_times)
                try:
                    f(*args)
                    __does_error_happened = False
                except Exception as e:
                    print("##############################################################################")
                    print("                     >>>   Exception Happened   <<<                           ")
                    print(traceback.format_exc())
                    print("                     ^^^   Check What Happened  ^^^                           ")
                    print("##############################################################################")
                    __does_error_happened = True
                    if not ignore_error:
                        raise e
                __retry_times -= 1
                if get_condition():
                    print ("""
******************************************************************************
            Condition fail, retry will be performed in next %ss
******************************************************************************
""" % interval)
                    time.sleep(__interval)
        return wrapped_f
    return wrap

转载于:https://www.cnblogs.com/velly/p/4635093.html