当前位置:王中王鉄算盘开奖结果 > 操作系统 > 编写装饰器实现python请求错误重试功能

编写装饰器实现python请求错误重试功能

文章作者:操作系统 上传时间:2019-11-17

在做接口自动化测试的时候,总会遇到,因连接超时等错误导致,接口脚本失败。

官方给出的方法:

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

 

本文由王中王鉄算盘开奖结果发布于操作系统,转载请注明出处:编写装饰器实现python请求错误重试功能

关键词: