Python 的重试机制

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

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

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

　　· 在什么情况下才进行重试?

　　· 重试几次呢?

　　· 重试多久后结束?

　　· 每次重试的间隔多长呢?

　　· 重试失败后的回调?

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

　　$ 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

　　以上内容为大家介绍了Python 的重试机制，希望对大家有所帮助，如果想要了解更多Python相关知识，请关注多测师。https://www.e70w.com/xwzx/