Tenacity是一个 Apache 2.0授权的通用重试库，用 Python 编写，用于简化向几乎所有内容添加重试行为的任务。它起源于一个重新尝试的分支，可惜这个分支已经不复存在了。
使用Tenacity可以用来进行测试用例的重跑，爬虫脚本的重跑，以及抢票的失败重抢等等。。。可以使用的场景也是比较多。
首先安装Tenacity
pip install Tenacity
无限重试
第一个重试案例,因为一直是抛出异常错误，所以无限进行重试执行
from tenacity import retry
@retry()
def test_retry():
print('失败重试中')
raise Exception
test_retry()
成功则停止
我们来优化成功一次后程序则终止，否则继续重试。
from tenacity import retry
import random
@retry()
def test_retry():
if random.randint(0,10) > 1:
print('失败重试中')
raise Exception
else:
print('成功')
test_retry()
重试次数
毕竟一直重试需要消耗很多资源，所以我们可以设置一些重试的次数，比如在失败多少次后停止重试,不管有没有成功。
from tenacity import retry,stop_after_attempt
import random
@retry(stop=stop_after_attempt(7))
def test_retry():
# if random.randint(0,10) > 1:
print('失败重试中')
raise Exception
# else:
# print('成功')
test_retry()
重试时间
也可以设置执行的时间
from tenacity import retry,stop_after_attempt,stop_after_delay
import random
from time import sleep
@retry(stop=stop_after_delay(3))
def test_retry():
# if random.randint(0,10) > 1:
sleep(1)
print('失败重试中')
raise Exception
# else:
# print('成功')
test_retry()
条件组合
甚至可以使用多个组合条件进行停止,哪个条件先触发则执行哪个
from tenacity import retry,stop_after_attempt,stop_after_delay
import random
from time import sleep
@retry(stop=stop_after_delay(3) | stop_after_attempt(2))
def test_retry():
# if random.randint(0,10) > 1:
sleep(1)
print('失败重试中')
raise Exception
# else:
# print('成功')
test_retry()
重试间隔
重试之间的间隔时间太短，所以让我们在重试之间等待2秒钟
from tenacity import retry,stop_after_attempt,stop_after_delay,wait_fixed
import random
import time
@retry(wait=wait_fixed(2))
def test_retry():
# if random.randint(0,10) > 1:
print('失败重试中')
print(time.ctime())
raise Exception
# else:
# print('成功')
test_retry()
重试随机间隔
我们还可以设置一些等待时间范围
from tenacity import retry,stop_after_attempt,stop_after_delay,wait_fixed,wait_random
import random
import time
@retry(wait=wait_random(min=1,max=2))
def test_retry():
# if random.randint(0,10) > 1:
print('失败重试中')
print(time.ctime())
raise Exception
# else:
# print('成功')
test_retry()
重试前日志
在执行之前打印日志
from tenacity import retry,stop_after_attempt,before_log
import logging
import sys
logging.basicConfig(stream=sys.stderr,level=logging.DEBUG)
logger = logging.getLogger(__name__)
@retry(stop=stop_after_attempt(3),before=before_log(logger,logging.DEBUG))
def test_retry():
print('失败重试中')
raise Exception('Fail')
test_retry()
重试后日志
那么相同的，可以在执行失败后打印日志
from tenacity import retry,stop_after_attempt,after_log
import logging
import sys
logging.basicConfig(stream=sys.stderr,level=logging.DEBUG)
logger = logging.getLogger(__name__)
@retry(stop=stop_after_attempt(3),after=after_log(logger,logging.DEBUG))
def test_retry():
print('失败重试中')
raise Exception('Fail')
test_retry()
基本常用的功能就这些了，如果有需要深入了解的可以访问github地址：https://github.com/jd/tenacity
js