1 、简介
我们在编写程序尤其是与网络请求相关的程序,如调用web接口、运行网络爬虫等任务时,经常会遇到一些偶然发生的请求失败的状况,这种时候如果我们仅仅简单的捕捉错误然后跳过对应任务,肯定是不严谨的,尤其是在网络爬虫中,会存在损失有价值数据的风险。
这类情况下我们就很有必要为我们的程序逻辑添加一些「错误重试」的策略,费老师我在几年前写过文章介绍过Python
中的retry
库,但它功能较为单一,只能应对基本的需求。
而今天我要给大家介绍的tenacity
库,可能是目前Python
生态中最好用的错误重试库,下面就让我们一睹其主要功能吧~
2 tenacity中的常用功能
作为一个第三方Python
库,我们可以使用pip install tenacity
对其进行安装,安装完成后,下面我们来学习一下tenacity
的主要使用方法和特性:
2.1 tenacity的基础使用
tenacity
的错误重试核心功能由其retry
装饰器来实现,默认不给retry
装饰器传参数时,它会在其所装饰的函数运行过程抛出错误时不停地重试下去,譬如下面这个简单的例子:
import random
from tenacity import retry
@retry
def demo_func1():
a = random.random()
print(a)
if a >= 0.1:
raise Exception
demo_func1()
可以看到,我们的函数体内每次生成0到1之间的随机数,当这个随机数不超过0.1时才会停止抛出错误,否则则会被tenacity
捕捉到每次的错误抛出行为并立即重试。
2.2 设置最大重试次数
有些时候我们对某段函数逻辑错误重试的忍耐是有限度的,譬如当我们调用某个网络接口时,如果连续n次都执行失败