测试派 Give testers a home.

python 错误重试用它就够了

一、安装

pip install tenacity

二、使用

• 使用规则

  • 同一个参数，多个值用|(或)，+(与)进行组合使用
  • 不同参数之间，只有组合使用，通过关键字参数传参即可

• @retry()【常用】

  • 【无条件重试】，只要抛出异常就会重试，直到执行不抛异常

  from tenacity import *
  一直重试
  @retry()
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()
  一直执行test_demo函数

• @retry(stop=stop_after_attempt(3))【常用】

  • 指定【重试的次数】，如3次

  from tenacity import *
  重试3次后停止
  @retry(stop=stop_after_attempt(3))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()
  执行三次test_demo函数

• @retry(stop=stop_after_delay(5))【常用】

  • 指定【重试多长时候后停止】，如5秒

  from tenacity import *
  重试5秒后停止
  @retry(stop=stop_after_delay(5))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()

• @retry(stop=(stop_after_delay(1) | stop_after_attempt(5)))

  • stop_after_delay()和stop_after_attempt()组合使用，只要其中一个条件满足，任务就停止

  from tenacity import *
  重试2秒或者重试50次停止
  @retry(stop=(stop_after_delay(2) | stop_after_attempt(50)))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()

• @retry(wait=wait_fixed(3))【常用】

  • 指定每一次【重试时等待时间】，如3秒，每一次重试前都要等待3秒钟

  from tenacity import *
  每次重试的等待时间5秒
  @retry(wait=wait_fixed(5))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()

• @retry(wait=wait_random(min=1, max=5))

  • 【重试时等待时间】在min,max之间随机取值

  from tenacity import *
  重试间隔时间1-5秒随机
  @retry(wait=wait_random(min=1, max=5))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()

• @retry(wait=wait_fixed(3) + wait_random(0, 2))

  • wait_fixed(3) 与wait_random(0, 2)组合使用，两个条件都满足
  • 比如每次等待3秒

  from tenacity import *
  随机等待0-2秒和每次等待3秒重试都满足才会重试
  @retry(wait=wait_fixed(3) + wait_random(0, 2))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()

• @retry(retry=retry_if_exception_type(IOError))

  • 指定特定类型的异常出现时，任务才重试，会一直重试

  from tenacity import *
  指定TypeError才会重复
  @retry(retry=retry_if_exception_type(TypeError))
  def test_demo():
  print('执行test_demo')
  print('a'+1)
  test_demo()

• 组合使用

  @retry(wait=wait_random(min=1, max=5),stop=stop_after_attempt(3))
  def test_demo():
  print('执行test_demo')
  raise Exception('手动抛出异常')
  test_demo()