Retry 是一个用于 Python 的库,用于在函数调用失败时自动重试。它的目标是简化重试逻辑的编写,处理由于临时性问题(如网络故障、API 限制等)导致的失败。Retry 的主要特点包括:
- 简单易用:只需使用装饰器或上下文管理器即可实现重试功能。
- 灵活配置:可以配置重试次数、重试间隔、异常处理等。
- 支持自定义重试逻辑:可以自定义哪些异常触发重试、重试的等待时间等。
下面通过代码演示 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}") # 输出: 最终失败,异常信息: 函数执行失败
标签:function,...,retry,函数,示例,python,重试,test
From: https://blog.csdn.net/m0_54007171/article/details/140741565