Python错误重试方法

1|0前言

Tenacity是一个 Apache 2.0授权的通用重试库，用 Python 编写，用于简化向几乎所有内容添加重试行为的任务。它起源于一个重新尝试的分支，可惜这个分支已经不复存在了。
使用Tenacity可以用来进行测试用例的重跑，爬虫脚本的重跑，以及抢票的失败重抢等等。。。可以使用的场景也是比较多。

2|0使用

首先安装Tenacity

pip install Tenacity

2|1无限重试

第一个重试案例,因为一直是抛出异常错误，所以无限进行重试执行

from tenacity import retry
 
@retry()
def test_retry():
	print('失败重试中')
    raise Exception
    
test_retry()

2|2成功则停止

我们来优化成功一次后程序则终止，否则继续重试。

from tenacity import retry
import random
 
@retry()
def test_retry():
    if random.randint(0,10) > 1:
        print('失败重试中')
        raise Exception
    else:
        print('成功')
 
test_retry()

2|3重试次数

毕竟一直重试需要消耗很多资源，所以我们可以设置一些重试的次数，比如在失败多少次后停止重试,不管有没有成功。

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()

2|4重试时间

也可以设置执行的时间

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()

2|5条件组合

甚至可以使用多个组合条件进行停止,哪个条件先触发则执行哪个

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|6重试间隔

重试之间的间隔时间太短，所以让我们在重试之间等待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()

2|7重试随机间隔

我们还可以设置一些等待时间范围

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()

2|8重试前日志

在执行之前打印日志

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()

2|9重试后日志

那么相同的，可以在执行失败后打印日志

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

__EOF__

本文作者：Harry
本文链接：https://www.cnblogs.com/harry66/p/14223952.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！