擴展functools.wraps裝飾器的功能

Question

我想創建一個新的裝飾器來代替@wraps(f)，它可以做任何其他的魔術@wraps(f)。我會怎麼做？

具體來說，我有幾個裝飾器的形式爲：

def decorator(f): 
    @wraps(f) 
    def wrapper(*args, **kwargs): 
     # does some stuff in here 
     return f(*args, **kwargs) 
    if not hasattr(wrapper, '_first_line'): 
     wrapper._first_line = inspect.getsourcelines(f)[1] 
    return wrapper 

好像我應該能夠創造這樣一個@wraps_with_first_line(f)裝飾，將盡一切@wraps(f)是因爲這樣做以及if not hasattr(wrapper, '_first_line'): wrapper._first_line = inspect.getsourcelines(f)[1] 。

Answer 1

如果你想添加究竟是不是已經被包裝對象的屬性，那麼你可以使用這個：

def wraps_with_first_line(f): 
    def wrap(wrapper): 
     wrapper = wraps(f)(wrapper) 
     if not hasattr(wrapper, '_first_line'): 
      wrapper._first_line = inspect.getsourcelines(f)[1] 
     return wrapper 
    return wrap 

如果是已經被包裝的對象的屬性，使用斯文的方法。

Answer 2

您應該更好地添加指向包裝函數的__wrapped__屬性，而不是添加該包裝函數的單個屬性。的functools.wraps()New versions自動做到這一點，但如果你使用Python的是舊版本比3.2，你可以ALS輕鬆地擴展wraps()添加__wrapped__：

def my_wraps(wrapped, **kwargs): 
    def decorator(wrapper): 
     functools.update_wrapper(wrapper, wrapped, **kwargs) 
     wrapper.__wrapped__ = wrapped 
    return decorator 

編輯：這裏有一個函數提取原始函數從一個可能的裝飾乘法功能：

def orig_function(f): 
    try: 
     while True: 
      f = f.__wrapped__ 
    except AttributeError: 
     return f