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

发表于:2020-7-21 10:46

字体: | 上一篇 | 下一篇 | 我要投稿

 作者:Python编程时光    来源:写代码的明哥

#
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

本文内容不用于商业目的,如涉及知识产权问题,请权利人联系51Testing小编(021-64471599-8017),我们将立即处理。
《2023软件测试行业现状调查报告》独家发布~

关注51Testing

联系我们

快捷面板 站点地图 联系我们 广告服务 关于我们 站长统计 发展历程

法律顾问:上海兰迪律师事务所 项棋律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2024
投诉及意见反馈:webmaster@51testing.com; 业务联系:service@51testing.com 021-64471599-8017

沪ICP备05003035号

沪公网安备 31010102002173号