【python学习】retry库用法大全！附示例代码

Retry 是一个用于 Python 的库，用于在函数调用失败时自动重试。它的目标是简化重试逻辑的编写，处理由于临时性问题（如网络故障、API 限制等）导致的失败。Retry 的主要特点包括：

1. 简单易用：只需使用装饰器或上下文管理器即可实现重试功能。
2. 灵活配置：可以配置重试次数、重试间隔、异常处理等。
3. 支持自定义重试逻辑：可以自定义哪些异常触发重试、重试的等待时间等。

下面通过代码演示 Retry 的几个最核心的功能：

安装 Retry

你可以通过 pip 安装 Retry：

pip install retry

基本使用

from retry import retry

# 装饰器实现重试
@retry()
def test_function():
    print("尝试执行函数...")
    raise Exception("函数执行失败")  # 输出: 尝试执行函数...（重试3次后，抛出异常）

test_function()

配置重试参数

from retry import retry

# 配置重试次数和重试间隔
@retry(tries=3, delay=2)
def test_function():
    print("尝试执行函数...")
    raise Exception("函数执行失败")  # 输出: 尝试执行函数...（每隔2秒重试一次，重试3次后，抛出异常）

test_function()

只对特定异常进行重试

from retry import retry

# 只对特定异常进行重试
@retry(exceptions=ValueError, tries=3)
def test_function():
    print("尝试执行函数...")
    raise ValueError("值错误")  # 输出: 尝试执行函数...（重试3次后，抛出异常）

test_function()

自定义重试间隔

from retry import retry
import random

# 自定义重试间隔
@retry(delay=2, backoff=2)
def test_function():
    print("尝试执行函数...")
    if random.random() < 0.5:
        raise Exception("函数执行失败")  # 输出: 尝试执行函数...（如果失败，重试时间隔会逐渐增加）

test_function()

捕获并处理异常

from retry import retry

# 捕获并处理异常
@retry(tries=3, delay=2)
def test_function():
    print("尝试执行函数...")
    raise Exception("函数执行失败")  # 输出: 尝试执行函数...（重试3次后，抛出异常）

try:
    test_function()
except Exception as e:
    print(f"最终失败，异常信息: {e}")  # 输出: 最终失败，异常信息: 函数执行失败

在上下文管理器中使用

from retry.api import retry_call

def test_function():
    print("尝试执行函数...")
    raise Exception("函数执行失败")  # 输出: 尝试执行函数...（重试3次后，抛出异常）

try:
    retry_call(test_function, tries=3, delay=2)
except Exception as e:
    print(f"最终失败，异常信息: {e}")  # 输出: 最终失败，异常信息: 函数执行失败