用Python装饰器给函数自动加上try except

最新推荐文章于 2023-12-26 16:00:00 发布

morein2008

最新推荐文章于 2023-12-26 16:00:00 发布

阅读量496

点赞数

分类专栏： Python 文章标签： python logging lambda delegate yii

本文链接：https://blog.csdn.net/hoo1990/article/details/117377469

版权

Python 专栏收录该内容

21 篇文章 2 订阅

订阅专栏

该博客介绍了一个Python装饰器的实现，用于自动捕获并记录函数执行过程中的异常。通过定义`error_handler`装饰器，可以将try...except结构包裹在函数调用周围，当发生异常时，会将异常信息写入日志文件并返回一个错误提示。示例中展示了如何用这个装饰器处理可能会抛出ZeroDivisionError的`test()`函数。

摘要由CSDN通过智能技术生成

用装饰器实现把函数用try...except包起来：

import logging
import traceback
from functools import wraps

logging.basicConfig(filename='error.log',
                    level=logging.INFO,
                    format='%(levelname)s::%(name)s::%(asctime)s::%(filename)s,line %(lineno)s, in %(funcName)s(),msg:%(message)s'
                    )
#定义装饰器error_handler
def error_handler(func):
    @wraps(func) #关键行
    def wrapper(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except:
            msg=traceback.format_exc()
            logging.error(msg)
            return 'error'
    return wrapper #返回函数名

#用法示例：自动处理test()的异常
@error_handler
def test():
    a=1/0
    return a

if __name__=='__main__':
    test()