[Python] 添加重试机制来处理某些可能出现异常的函数

在 Python 中，如果你想要添加重试机制来处理某些可能出现异常的函数，可以使用 retry 装饰器。虽然 Python 标准库中没有提供内置的 @retry 装饰器，但你可以使用第三方库 tenacity 来实现这个功能。

tenacity 是一个流行的 Python 库，提供了灵活而强大的重试逻辑，可以帮助你处理函数调用时可能出现的异常情况。

以下是一个简单的示例，展示如何使用 @retry 装饰器来实现重试功能：

首先，确保你已经安装了 tenacity 库，如果没有安装，可以通过以下命令进行安装：

pip install tenacity

然后，使用 @retry 装饰器来为函数添加重试逻辑：

from tenacity import retry, stop_after_attempt, wait_fixed

# 使用 @retry 装饰器指定重试逻辑和参数
@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), reraise=True)
def do_something():
    print("Doing something...")
    # 模拟一个失败的操作
    raise ValueError("Something went wrong")

try:
    do_something()
except Exception as e:
    print(f"Exception caught: {e}")

在上面的代码中，我们定义了一个名为 do_something 的函数，并使用 @retry 装饰器来指定重试逻辑。在装饰器中，我们设置了

• stop 参数为 stop_after_attempt(3)，表示最多重试 3 次；

• wait 参数为 wait_fixed(2)，表示每次重试之间等待 2 秒。

• reraise=True 参数。这意味着当重试次数用尽后，会重新引发最后一次异常。这在某些情况下可能有用，可以让程序在重试多次后仍然能够向上层传递异常信息。

---

解释每个参数的含义：

1. sleep: t.Callable[[t.Union[int, float]], None] = sleep

   • 这个参数表示用于引入延迟的函数，其类型为可调用对象，接受一个整数或浮点数作为参数，并返回 None。默认值为 sleep，即内置的 Python time.sleep 函数。

2. stop: "StopBaseT" = stop_never

   • 用于确定何时停止重试的策略，类型为 StopBaseT。默认值为 stop_never，表示永远不停止重试。

3. wait: "WaitBaseT" = wait_none()

   • 用于确定每次重试之间的等待时间的策略，类型为 WaitBaseT。默认值为 wait_none()，表示没有等待时间。

4. retry: "RetryBaseT" = retry_if_exception_type()

   • 用于确定是否应该进行重试的策略，类型为 RetryBaseT。默认值为 retry_if_exception_type()，表示根据异常类型来决定是否重试。

5. before: t.Callable[["RetryCallState"], None] = before_nothing

   • 在每次重试之前执行的函数，类型为接受 RetryCallState 对象并返回 None 的可调用对象。默认值为 before_nothing，表示什么都不做。

6. after: t.Callable[["RetryCallState"], None] = after_nothing

   • 在每次重试之后执行的函数，类型为接受 RetryCallState 对象并返回 None 的可调用对象。默认值为 after_nothing，表示什么都不做。

7. before_sleep: t.Optional[t.Callable[["RetryCallState"], None]] = None

   • 在每次重试之前执行延迟前的函数，类型为可选的可调用对象，接受 RetryCallState 对象并返回 None。默认值为 None，表示没有相关操作。

8. reraise: bool = False

   • 控制是否重新引发最新的异常。默认值为 False，表示不重新引发异常。

9. retry_error_cls: t.Type[RetryError] = RetryError

   • 重试错误的类，用于创建重试错误实例。默认值为 RetryError，表示使用内置的 RetryError 类。

10. retry_error_callback: t.Optional[t.Callable[["RetryCallState"], t.Any]] = None

    • 在发生重试错误时执行的回调函数，类型为可选的可调用对象，接受 RetryCallState 对象并返回任意类型。默认值为 None，表示没有回调函数。