【Python】functools.wraps定義函數裝飾器

　　第一次見到functools.wraps是在 Flask Web開發 中，一直不明白怎么回事。

　　裝飾器（decorator）是干嘛的？對於受到封裝的原函數來說，裝飾器能夠在那個函數執行前或者執行后分別運行一些代碼，使得可以再裝飾器里面訪問並修改原函數的參數以及返回值，以實現約束定義、調試程序、注冊函數等目標。裝飾器一般返回一個包裝器（wrapper），而functools.wraps就是裝飾包裝器的裝飾器。

先來看一個不使用functools.wraps的裝飾器例子。

def tracer(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print('%s(%r,%r)->%r'%(func.__name__,args,kwargs,result))
        return result
    return wrapper

@tracer
def fibonacci(n):
    if n in (0,1):
        return n
    return (fibonacci(n-1)+fibonacci(n-2))


fibonacci(3)
print(fibonacci)
print('help:')
help(fibonacci)

 

輸出結果：

 

 

可以看到，裝飾器完全可以正常工作。。。

但是，函數的名字變成裝飾器中的包裝器了！！！help內置函數也失效了

也就是說，原函數的屬性失效了

如果想要保留原函數的屬性，就可以用到functools.wraps了

from functools import wraps
def tracer(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print('%s(%r,%r)->%r'%(func.__name__,args,kwargs,result))
        return result
    return wrapper

@tracer
def fibonacci(n):
    if n in (0,1):
        return n
    return (fibonacci(n-1)+fibonacci(n-2))


fibonacci(3)
print(fibonacci)
print('help:')
help(fibonacci)

輸出結果：

 

 

參考資料：Effective Python,第6章 內置模塊