测试派 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()