Python的Tenacity库：通过装饰器（@retry）实现自动重试

Tenacity库

Python的Tenacity库是一个功能强大的重试库，它主要用于简化任务重试的逻辑，提升程序的健壮性和可靠性。Tenacity库的主要功能包括以下几个方面：

1. 自动重试机制
基本重试：通过Tenacity装饰器（@retry）实现自动重试。默认情况下，如果不给retry装饰器传递参数，它会在被装饰的函数抛出错误时不停地重试下去。
条件重试：可以设置重试的条件，如只针对特定类型的异常进行重试，或者根据函数的返回值判断是否重试。
2. 灵活的停止策略
最大重试次数：通过stop_after_attempt()函数可以指定最大重试次数，达到次数后停止重试。
最大超时时长：通过stop_after_delay()函数可以设置整个重试过程的最大耗时，超出时长后结束重试。
组合停止条件：可以使用|运算符组合不同的停止条件，如同时设置最大重试次数和最大超时时长。
3. 等待策略
固定等待时间：通过wait_fixed()函数，可以设置相邻重试之间的固定等待时间。
随机等待时间：wait_random()函数允许在指定的时间范围内随机选择等待时间，这有助于在爬虫等场景中避免被服务器识别为恶意访问。
指数退避等待：wait_exponential()函数实现了一种更为智能的等待策略，即每次重试之间的等待时间按照指数级增长，以减少对服务器的频繁请求。
4. 自定义重试逻辑
重试前处理：可以在重试前执行一些预处理操作，如记录日志、更新状态等。
重试后处理：在每次重试后执行一些后处理操作，如清理资源、更新重试次数等。
5. 易于集成
Tenacity库易于与现有的Python项目集成，只需通过pip安装即可使用。
它提供了简洁的API和丰富的配置选项，使得开发者可以轻松地根据自己的需求定制重试逻辑。
综上所述，Tenacity库通过提供自动重试、灵活的停止策略、等待策略以及自定义重试逻辑等功能，极大地简化了任务重试的实现过程，提高了程序的健壮性和可靠性。在编写需要处理网络请求、文件操作等可能失败的任务时，使用Tenacity库可以大大减少代码量并降低出错的可能性。

自动重试

以下是一个使用Python的Tenacity库来实现自动重试的简单示例。在这个示例中，我们将创建一个可能失败的函数（比如模拟网络请求或文件操作），并使用@retry装饰器来自动重试这个函数。

首先，你需要安装Tenacity库（如果你还没有安装的话）：

pip install tenacity

然后，编写如下的Python代码：

from tenacity import retry, stop_after_attempt, wait_fixed  
  
# 定义一个可能失败的函数  
# 这里我们简单地模拟一个总是失败的函数（总是抛出异常）  
# 但在实际应用中，这个函数可能会因为网络问题、文件不存在等原因失败  
def might_fail():  
    print("Attempting to do something...")  
    # 假设这里有一个可能导致失败的操作  
    # 为了示例，我们总是抛出异常  
    raise ValueError("Something went wrong!")  
  
# 使用@retry装饰器来包装might_fail函数  
# 设置重试停止条件为最多尝试3次  
# 设置重试等待策略为每次失败后等待1秒  
@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))  
def robust_might_fail():  
    return might_fail()  
  
# 调用被装饰的函数  
try:  
    result = robust_might_fail()  
    print("Success!")  
except Exception as e:  
    print(f"Failed after all retries: {e}")  
  
# 输出将显示三次尝试失败的过程，并在第三次失败后捕获异常  
# 尝试输出示例:  
# Attempting to do something...  
# Attempting to do something...  
# Attempting to do something...  
# Failed after all retries: Something went wrong!

在这个示例中，might_fail函数总是抛出ValueError异常，表示操作失败。我们使用@retry装饰器来包装might_fail函数，并将其重命名为robust_might_fail。通过装饰器，我们设置了重试的停止条件为最多尝试3次（stop=stop_after_attempt(3)），以及相邻重试之间的等待时间为1秒（wait=wait_fixed(1)）。

当调用robust_might_fail函数时，如果内部的might_fail函数抛出异常，Tenacity会自动捕获这个异常，等待1秒后再次尝试调用might_fail，直到达到最大重试次数或成功为止。在这个示例中，由于might_fail总是失败，所以最终会抛出异常，但我们可以通过try-except块来捕获这个异常，并给出相应的处理或提示。

注：装饰器（@retry）请见参考：Python中用@装饰器-CSDN博客