Java自学者论坛

 找回密码
 立即注册

手机号码,快捷登录

恭喜Java自学者论坛(https://www.javazxz.com)已经为数万Java学习者服务超过8年了!积累会员资料超过10000G+
成为本站VIP会员,下载本站10000G+会员资源,会员资料板块,购买链接:点击进入购买VIP会员

JAVA高级面试进阶训练营视频教程

Java架构师系统进阶VIP课程

分布式高可用全栈开发微服务教程Go语言视频零基础入门到精通Java架构师3期(课件+源码)
Java开发全终端实战租房项目视频教程SpringBoot2.X入门到高级使用教程大数据培训第六期全套视频教程深度学习(CNN RNN GAN)算法原理Java亿级流量电商系统视频教程
互联网架构师视频教程年薪50万Spark2.0从入门到精通年薪50万!人工智能学习路线教程年薪50万大数据入门到精通学习路线年薪50万机器学习入门到精通教程
仿小米商城类app和小程序视频教程深度学习数据分析基础到实战最新黑马javaEE2.1就业课程从 0到JVM实战高手教程MySQL入门到精通教程
查看: 549|回复: 0

python异常装饰器--比较全的版本了

[复制链接]
  • TA的每日心情
    奋斗
    2024-11-24 15:47
  • 签到天数: 804 天

    [LV.10]以坛为家III

    2053

    主题

    2111

    帖子

    72万

    积分

    管理员

    Rank: 9Rank: 9Rank: 9

    积分
    726782
    发表于 2021-6-18 21:34:47 | 显示全部楼层 |阅读模式
    # 异常捕获装饰器(亦可用于类方法)
    def try_except_log(f=None, max_retries: int = 5, delay: (int, float) = 1, step: (int, float) = 0,
                       exceptions: (BaseException, tuple, list) = BaseException, sleep=time.sleep,
                       process=None, validate=None, callback=None, default=None):
        """
        函数执行出现异常时自动重试的简单装饰器
        :param f: function 执行的函数。
        :param max_retries: int 最多重试次数。
        :param delay: int/float 每次重试的延迟,单位秒。
        :param step: int/float 每次重试后延迟递增,单位秒。
        :param exceptions: BaseException/tuple/list 触发重试的异常类型,单个异常直接传入异常类型,多个异常以tuple或list传入。
        :param sleep: 实现延迟的方法,默认为time.sleep。
        在一些异步框架,如tornado中,使用time.sleep会导致阻塞,可以传入自定义的方法来实现延迟。
        自定义方法函数签名应与time.sleep相同,接收一个参数,为延迟执行的时间。
        :param process: 处理函数,函数签名应接收一个参数,每次出现异常时,会将异常对象传入。
        可用于记录异常日志,中断重试等。
        如处理函数正常执行,并返回True,则表示告知重试装饰器异常已经处理,重试装饰器终止重试,并且不会抛出任何异常。
        如处理函数正常执行,没有返回值或返回除True以外的结果,则继续重试。
        如处理函数抛出异常,则终止重试,并将处理函数的异常抛出。
        :param validate: 验证函数,用于验证执行结果,并确认是否继续重试。
        函数签名应接收一个参数,每次被装饰的函数完成且未抛出任何异常时,调用验证函数,将执行的结果传入。
        如验证函数正常执行,且返回False,则继续重试,即使被装饰的函数完成且未抛出任何异常。
        如验证函数正常执行,没有返回值或返回除False以外的结果,则终止重试,并将函数执行结果返回。
        如验证函数抛出异常,且异常属于被重试装饰器捕获的类型,则继续重试。
        如验证函数抛出异常,且异常不属于被重试装饰器捕获的类型,则将验证函数的异常抛出。
        :param callback: 回调函数,函数签名应接收一个参数,异常无法处理时,会将异常对象传入。
        可用于记录异常日志,发送异常日志等。
        :param default: 默认值/默认值生成函数
        :return: 被装饰函数的执行结果。
        """
    
        # 带参数的装饰器
        def decorator(func):
            @functools.wraps(func)
            def wrapper(*args, **kwargs):
                # nonlocal delay, step, max_retries
                i = 0
                func_exc, exc_traceback = StopRetry, None
                while i < max_retries:
                    try:
                        result = func(*args, **kwargs)
                        # 验证函数返回False时,表示告知装饰器验证不通过,继续重试
                        if callable(validate) and validate(result) is False:
                            continue
                        else:
                            return result
                    except exceptions as ex:
                        func_exc, exc_traceback = ex, traceback.format_exc()
                        # 处理函数返回True时,表示告知装饰器异常已经处理,终止重试
                        if callable(process):
                            try:
                                if process(ex) is True:
                                    return default() if callable(default) else default
                            except Exception as e:
                                func_exc, exc_traceback = e, traceback.format_exc()
                                break
                    finally:
                        i += 1
                        sleep(delay + step * i)
                else:
                    # 回调函数,处理自动无法处理的异常
                    if callable(callback):
                        callback(func_exc, exc_traceback)
                    return default() if callable(default) else default
                pass
    
            return wrapper
    
        if callable(f):
            return decorator(f)
        return decorator

     

    哎...今天够累的,签到来了1...
    回复

    使用道具 举报

    您需要登录后才可以回帖 登录 | 立即注册

    本版积分规则

    QQ|手机版|小黑屋|Java自学者论坛 ( 声明:本站文章及资料整理自互联网,用于Java自学者交流学习使用,对资料版权不负任何法律责任,若有侵权请及时联系客服屏蔽删除 )

    GMT+8, 2024-12-23 02:06 , Processed in 0.060186 second(s), 29 queries .

    Powered by Discuz! X3.4

    Copyright © 2001-2021, Tencent Cloud.

    快速回复 返回顶部 返回列表