如何捕捉裝飾器中的異常

Question

我有一個函數，我的原因異常，我希望它是一個裝飾器。其代碼如下：

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' 

但輸出是：

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 

這樣，我怎麼能捕獲此異常？

Answer 1

至於其他的答案已經解釋，當前的問題是，當裝飾被應用到該函數，而不是函數被調用你得到拋出的異常。

爲了解決這個問題，你需要讓裝飾器返回一個提升異常的函數。以下是如何可以工作：

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 

的des功能是「裝飾工廠」。除了提供一個範圍來保存它返回的修飾器的參數i之外，它並沒有做任何其他的事情。

decorator函數檢查是否有任何特殊的需要完成。如果不是，它會返回修改後的功能。如果i==1，它返回一個自定義函數。

raiser函數是裝飾器的返回值，如果i==1。它在被調用時總會引發異常。應用於此的functools.wraps修飾器並不是嚴格必要的，但它使它看起來更像原始功能（相同的__name__,__doc__等）。

Answer 2

當您的定義了函數時會引發異常。捕獲這個異常的唯一方法是：

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

如果這是令人困惑，爲什麼這是造成異常，請記住你的代碼就相當於：

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

Answer 3

所以，現在你的代碼基本上可以歸結爲此：

_des = des(1) 

def _func(); 
    print '!!' 

func = _des(func) 

您使用的des作爲裝飾的返回值，我認爲這是造成問題的原因。

我想你可能需要在該函數返回一個更多的時間窩：

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 4

我在這裏缺少一個功能水平。

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'