編寫裝飾器實現python請求錯誤重試功能

在作接口自動化測試的時候，總會遇到，因鏈接超時等錯誤致使，接口腳本失敗。

官方給出的方法：

max_retries=5 出錯重試5次
注意的是，這個只對DNS，鏈接錯誤進行重試。

    from requests.adapters import HTTPAdapter
    s = requests.Session()
    s.mount('http://',HTTPAdapter(max_retries=5))
    s.mount('https://',HTTPAdapter(max_retries=5))
    s.get('https://www.baidu.com')

 

 

自編寫裝飾器一

from requests.exceptions import ConnectionError
import requests
def retry(**kw):
    def war(func):
        def w(*args,**kwargs):
            try:
                ret = func(*args,**kwargs)
            except ConnectionError:
                kw['reNum'] = int(kw['reNum']) - 1
                if kw['reNum'] >=0:
                    print kw['reNum']
                    ret = w(*args,**kwargs)
                else:
                    ret = ConnectionError
            return ret
        return w
    return war

 

自編寫裝飾器二

from requests.exceptions import ConnectionError

def retry(**kw):
    def wrapper(func):
        def _wrapper(*args,**kwargs):
            raise_ex = None
            for _ in range(kw['reNum']):
                print _
                try:
                    return func(*args,**kwargs)
                except ConnectionError as ex:
                    raise_ex = ex
            #raise raise_ex
        return _wrapper
    return wrapper

 

使用方法：reNum = 5 表明，出現ConnectionError時最多可重試5次。

@retry(reNum=5)
def demo():
    raise ConnectionError

 

總結：

1.編寫裝飾器，其實沒有那麼難，只要掌握方法。 這個能夠參考，我以前寫的關於裝飾器的文章

2.裝飾器的通熟解釋，就是在函數以前後以後作點什麼。經過這個咱們能夠作不少。

3.關於請求鏈接錯誤，重試，裝飾器；原理就是作一個循環，只要捕獲到有ConnectionError 錯誤，就進入下一次循環

調用；只要有正確的時候，直接返回函數。

 

qq技術交流羣，期待你的加入：

python｜測試｜技術交流羣：563227894

python｜測試｜技術交流羣：563227894

python｜測試｜技術交流羣：563227894