#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2017/8/18 上午9:50
# @Author : Matrix
# @Github : https://github.com/blackmatrix7/
# @Blog : http://www.cnblogs.com/blackmatrix/
# @File : retry.py
# @Software: PyCharm
import time
from functools import wraps

__author__ = 'blackmatrix'


"""
在函數(shù)執(zhí)行出現(xiàn)異常時自動重試的簡單裝飾器
"""


class StopRetry(Exception):

 def __repr__(self):
  return 'retry stop'


def retry(max_retries: int =5, delay: (int, float) =0, step: (int, float) =0,
   exceptions: (BaseException, tuple, list) =BaseException,
   sleep=time.sleep, callback=None, validate=None):
 """
 函數(shù)執(zhí)行出現(xiàn)異常時自動重試的簡單裝飾器。
 :param max_retries: 最多重試次數(shù)。
 :param delay: 每次重試的延遲，單位秒。
 :param step: 每次重試后延遲遞增，單位秒。
 :param exceptions: 觸發(fā)重試的異常類型，單個異常直接傳入異常類型，多個異常以tuple或list傳入。
 :param sleep: 實現(xiàn)延遲的方法，默認為time.sleep。
 在一些異步框架，如tornado中，使用time.sleep會導致阻塞，可以傳入自定義的方法來實現(xiàn)延遲。
 自定義方法函數(shù)簽名應與time.sleep相同，接收一個參數(shù)，為延遲執(zhí)行的時間。
 :param callback: 回調(diào)函數(shù)，函數(shù)簽名應接收一個參數(shù)，每次出現(xiàn)異常時，會將異常對象傳入。
 可用于記錄異常日志，中斷重試等。
 如回調(diào)函數(shù)正常執(zhí)行，并返回True，則表示告知重試裝飾器異常已經(jīng)處理，重試裝飾器終止重試，并且不會拋出任何異常。
 如回調(diào)函數(shù)正常執(zhí)行，沒有返回值或返回除True以外的結(jié)果，則繼續(xù)重試。
 如回調(diào)函數(shù)拋出異常，則終止重試，并將回調(diào)函數(shù)的異常拋出。
 :param validate: 驗證函數(shù)，用于驗證執(zhí)行結(jié)果，并確認是否繼續(xù)重試。
 函數(shù)簽名應接收一個參數(shù)，每次被裝飾的函數(shù)完成且未拋出任何異常時，調(diào)用驗證函數(shù)，將執(zhí)行的結(jié)果傳入。
 如驗證函數(shù)正常執(zhí)行，且返回False，則繼續(xù)重試，即使被裝飾的函數(shù)完成且未拋出任何異常。
 如回調(diào)函數(shù)正常執(zhí)行，沒有返回值或返回除False以外的結(jié)果，則終止重試，并將函數(shù)執(zhí)行結(jié)果返回。
 如驗證函數(shù)拋出異常，且異常屬于被重試裝飾器捕獲的類型，則繼續(xù)重試。
 如驗證函數(shù)拋出異常，且異常不屬于被重試裝飾器捕獲的類型，則將驗證函數(shù)的異常拋出。
 :return: 被裝飾函數(shù)的執(zhí)行結(jié)果。
 """
 def wrapper(func):
  @wraps(func)
  def _wrapper(*args, **kwargs):
   nonlocal delay, step, max_retries
   func_ex = StopRetry
   while max_retries > 0:
    try:
     result = func(*args, **kwargs)
     # 驗證函數(shù)返回False時，表示告知裝飾器驗證不通過，繼續(xù)重試
     if callable(validate) and validate(result) is False:
      continue
     else:
      return result
    except exceptions as ex:
     # 回調(diào)函數(shù)返回True時，表示告知裝飾器異常已經(jīng)處理，終止重試
     if callable(callback) and callback(ex) is True:
      return
     func_ex = ex
    finally:
     max_retries -= 1
     if delay > 0 or step > 0:
      sleep(delay)
      delay += step
   else:
    raise func_ex
  return _wrapper
 return wrapper


if __name__ == '__main__':
 pass