如何在装饰器中捕获异常

Question

I have a function my cause exception and i want it to be a decorator. The code is as follow:

def des(i):
    def new_func(func):
        if i == 1:
            raise Exception
        else:
            return func
    return new_func


@des(1)
def func():
    print "!!"


if __name__ == '__main__':
    try:
        func()
    except Exception:
        print 'error'

but the output is:

Traceback (most recent call last):
  File "D:/des.py", line 10, in <module>
    @des(1)
  File "D:/des.py", line 4, in new_func
    raise Exception
Exception

so, how can I catch this exception?

Answer 1

The exception is raised when you define the function. The only way to catch this exception would be:

try:
    @des(1)
    def func():
        print '!!'
except:
    print 'error'

If it's confusing why this is causing an exception, remember your code is equivalent to:

def func():
    print '!!'
func = des(1)(func)
# des(1) = new_func, so des(1)(func) is new_func(func)

Answer 2

So, right now your code basically boils down to this:

_des = des(1)

def _func();
    print '!!'

func = _des(func)

You're using the return value of des as the decorator, and I think that is causing the problem.

I think you might want to nest that returned function one more time:

def des(i): # container func.
    def new_func(func):
        def ret_func(*args, **kwargs):
            if i == 1:
                raise Exception
            else:
                return func(*args, **kwargs)

        return ret_func # return the func with the bound variable
    return new_func # return the func which creates the function w/ the bound var.


@des(1)
def func():
    print "!!"

Answer 3

I am missing one function level here.

ITYM

import functools
def des(i): # this is the "decorator creator", called with des(1)
    def deco(func): # this is returned and is the real decorator, called at function definition time
        @functools.wraps(func) # sugar
        def new_func(*a, **k): # and this is the function called on execution.
            if i == 1:
                raise Exception # I hope this is just for testing... better create a new exception for this
            else:
                return func(*a, **k)
        return new_func
    return deco

@des(1)
def func():
    print "!!"

if __name__ == '__main__':
    try:
        func()
    except Exception:
        print 'error'

Answer 4

As the other answers have explained, your current issue is that you're getting the exception raised when the decorator is applied to the function, not when the function is called.

To fix this, you need to make the decorator return a function that does the exception raising. Here's how that could work:

import functools

def des(i):
    def decorator(func):
        if i != 1:
            return func # no wrapper needed

        @functools.wraps(func)
        def raiser(*args, **kwargs):
            raise Exception

        return raiser

    return decorator

The des function is a "decorator factory". It doesn't really doesn't do anything other than providing a scope to hold the i parameter for the decorator that it returns.

The decorator function does the check to see if anything special needs to be done. If not, it returns the decorated function unmodified. If i==1 , it returns a custom function.

The raiser function is the decorator's return value if i==1 . It always raises an exception when it is called. The functools.wraps decorator applied to it is not strictly necessary, but it makes it look more like the original function (same __name__ , __doc__ , etc).