通过Django中的装饰器处理异常

时间:2017-04-23 16:00:29

标签: django python-3.x python-decorators

我正在尝试在django中创建一个错误处理装饰器,如果出现异常,则会在装饰器中发生日志记录,并将异常提升回装饰函数,然后发送错误的http响应。

但是当我尝试这样做时,如果我在装饰函数中添加了一个Except块,而装饰函数的异常被提升回来,那么只有装饰函数的except块被执行并且装饰器的except块被留下了未执行的。

MyDecorator.py

def log_error(message):

    def decorator(target_function):

        @functools.wraps(target_function)
        def wrapper(*args, **kwargs):
            try:
                return target_function(*args, **kwargs)
            except Exception as e:
                print('except block of the decorator')
                exceptiontype, exc_obj, exc_tb = sys.exc_info()
                filename = traceback.extract_tb(exc_tb)[-2][0]
                linenumber = traceback.extract_tb(exc_tb)[-2][1]
                mylogger(message=str(e),description='Related to Registration',fileName=filename,lineNumber=linenumber, exceptionType = type(e).__name__)

                raise

        return wrapper

    return decorator

装饰功能

    @log_error('message')
    def action(self, serializer):
        try:
            ..................
            ..................
            ..................
            ..................
            return Response(status=status.HTTP_200_OK, data={
                "message":'link sent successfully'})
        except Exception as e:
            print('except block of the decorated function')
            return Response(status=status.HTTP_500_INTERNAL_SERVER_ERROR, 
                    data={"message" : 'error sending link'})

这是打印行

  

除了装饰函数的块

而不是

  

除了装饰者的块

如果我从装饰函数中删除了Except块,那么装饰器的except块就会被执行。

了解如何在装饰器中处理此异常跟踪的任何帮助。

1 个答案:

答案 0 :(得分:0)

这是一种正常的行为:你从装饰器运行target_function并在其中捕获异常,因此装饰器看不到它。我认为你必须raise target_function中的异常,就像你在装饰器中所做的那样。但是由于你必须在装饰函数之外返回Response,问题似乎很难解决...