少有人知的 Python "重试机制"

栏目: IT技术 · 发布时间: 4年前

内容简介:为了避免由于一些网络或等其他不可控因素,而引起的功能性问题。比如在发送请求时,会因为网络不稳定,往往会有请求超时的问题。这种情况下,我们通常会在代码中加入重试的代码。重试的代码本身不难实现,但如何写得优雅、易用,是我们要考虑的问题。

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

少有人知的 Python

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

这里要给大家介绍的是一个第三方库 - Tenacity (标题中的重试机制并并不准确,它不是 Python 的内置模块,因此并不能称之为机制),它实现了几乎我们可以使用到的所有重试场景,比如:

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

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

$ pip install tenacity 

1. 最基本的重试

无条件重试,重试之间无间隔

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

无条件重试,但是在重试之前要等待 2 秒

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

2. 设置停止基本条件

只重试7 次

from tenacity import retry, stop_after_attempt 
 
@retry(stop=stop_after_attempt(7)) 
def test_retry(): 
    print("等待重试...") 
    raise Exception 
 
test_retry() 

重试 10 秒后不再重试

from tenacity import retry, stop_after_delay 
 
@retry(stop=stop_after_delay(10)) 
def test_retry(): 
    print("等待重试...") 
    raise Exception 
 
test_retry() 

或者上面两个条件满足一个就结束重试

from tenacity import retry, stop_after_delay, stop_after_attempt 
 
@retry(stop=(stop_after_delay(10) | stop_after_attempt(7))) 
def test_retry(): 
    print("等待重试...") 
    raise Exception 
 
test_retry() 

3. 设置何时进行重试

在出现特定错误/异常(比如请求超时)的情况下,再进行重试

from requests import exceptions 
from tenacity import retry, retry_if_exception_type 
 
@retry(retry=retry_if_exception_type(exceptions.Timeout)) 
def test_retry(): 
    print("等待重试...") 
    raise exceptions.Timeout 
 
test_retry() 

在满足自定义条件时,再进行重试。

如下示例,当 test_retry 函数返回值为 False 时,再进行重试

from tenacity import retry, stop_after_attempt, retry_if_result 
 
def is_false(value): 
    return value is False 
 
@retry(stop=stop_after_attempt(3), 
       retry=retry_if_result(is_false)) 
def test_retry(): 
    return False 
 
test_retry() 

4. 重试后错误重新抛出

当出现异常后,tenacity 会进行重试,若重试后还是失败,默认情况下,往上抛出的异常会变成 RetryError,而不是最根本的原因。

因此可以加一个参数(reraise=True),使得当重试失败后,往外抛出的异常还是原来的那个。

from tenacity import retry, stop_after_attempt 
 
@retry(stop=stop_after_attempt(7), reraise=True) 
def test_retry(): 
    print("等待重试...") 
    raise Exception 
 
test_retry() 

5. 设置回调函数

当最后一次重试失败后,可以执行一个回调函数

from tenacity import * 
 
def return_last_value(retry_state): 
    print("执行回调函数") 
    return retry_state.outcome.result()  # 表示返回原函数的返回值 
 
def is_false(value): 
    return value is False 
 
@retry(stop=stop_after_attempt(3), 
       retry_error_callback=return_last_value, 
       retry=retry_if_result(is_false)) 
def test_retry(): 
    print("等待重试中...") 
    return False 
 
print(test_retry()) 

输出如下

等待重试中... 
等待重试中... 
等待重试中... 
执行回调函数 
False 

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

参与的胜利

参与的胜利

亨利·詹金斯 / 高芳芳 / 浙江大学出版社 / 2017-9-30 / CNY 42.00

《参与的胜利:网络时代的参与文化》是一场学者之间的对话,三位学者(亨利·詹金斯、伊藤瑞子和丹娜·博伊德)虽然来自不同的代际、不同的学科背景,但他们在相同的参与文化项目中展开合作,并试图解决相似的问题。 希望《参与的胜利:网络时代的参与文化》能够进一步激发团体内部及团体之间的对话,这些团体包括教育者、政策制定者、学者、关注参与文化的公民、业内人士、粉丝及其他任何关心我们文化的未来的人。理想的参......一起来看看 《参与的胜利》 这本书的介绍吧!

URL 编码/解码
URL 编码/解码

URL 编码/解码

XML 在线格式化
XML 在线格式化

在线 XML 格式化压缩工具