【Python】重试库-tenacity【原创】

Python的重试库-tenacity

一、介绍

为了避免由于一些网络或等其他不可控因素，而引起的功能性问题。比如在发送请求时，会因为网络不稳定，往往会有请求超时的问题。

这种情况下，我们通常会在代码中加入重试的代码。重试的代码本身不难实现，但如何写得优雅、易用，是我们要考虑的问题。

这里有一个第三方库 - Tenacity ，它实现了几乎我们可以使用到的所有重试场景，比如：

1. 在什么情况下才进行重试？
2. 重试几次呢?
3. 重试多久后结束？
4. 每次重试的间隔多长呢？
5. 重试失败后的回调？

在使用它之前 ，先要安装它

$ pip install tenacity

二、用法

基本用法

简单的使用方法是给需要重试的方法加上@retry装饰器，代码抛出异常之后会被装饰器捕获并进行重试，异常抛出时会不断进行重试直到代码执行成功

例子：无条件重试直到执行成功，重试之间无间隔

from tenacity import retry

@retry
def test_retry():
    print("等待重试，重试无间隔执行...")
    raise Exception

test_retry()

可以使用@wait_fixed来指定重试之前的等待时间，也可使用@wait_random进行随机等待

例子：无条件重试直到成功，但在重试之前要等待 2 秒

from tenacity import retry, wait_fixed

@retry(wait=wait_fixed(2))
def test_retry():
    print("等待重试...")
    raise Exception

test_retry()

随机等待1-2秒再重试

from tenacity import retry, wait_random

@retry(wait=wait_random(min=1, max=2