0
点赞
收藏
分享

微信扫一扫

【python 装饰器 - 重试】做一个简易重试装饰器,如果函数执行错误则会自动重新执行,可设置重试次数,对爬虫比较友好

时光已翩然轻擦 03-24 18:00 阅读 3

上面的图片案例,里面的意思简单来说就是用 【传参】和【随机的整数】进行除法运算,而随机的整数不能为0,如果为0则表示错误,则会重新运行此函数,重新获取随机数,直到成功或者重试的次数用完
当然,这只是用来演示如何使用的,你们可以随意更改,也可以把参数改成爬虫请求,然后检测状态码,当出现错误的时候,会自动重试的
[retry]:默认为重试3次,可设置

【附上源码】
class retry_lane(object):
    def __init__(self, retry: int = 3):
        '''
        :param retry: int类型:重试次数
        使用方法:@retry_lane(retry=3)
        '''
        self.retry = retry
    def __call__(self, func):
        def wrapper(*args, **kwargs):
            l = 0
            while self.retry > 0:
                l += 1
                try:
                    print(f'运行第{l}次')
                    return func(*args, **kwargs)
                except Exception as e:
                    self.retry -= 1
                    if self.retry == 0:
                        return e
        return wrapper
举报

相关推荐

0 条评论