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

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

清水湾的水

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

阅读量325

点赞数 1

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/cezyzhao/article/details/128937654

版权

python 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

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()