python裝飾器中@wraps作用--修復被裝飾后的函數名等屬性的改變

Python裝飾器（decorator）在實現的時候，被裝飾后的函數其實已經是另外一個函數了（函數名等函數屬性會發生改變），為了不影響，Python的functools包中提供了一個叫wraps的decorator來消除這樣的副作用。寫一個decorator的時候，最好在實現之前加上functools的wrap，它能保留原有函數的名稱和docstring。
廢話不多說，上倆栗子就能搞明白！

 

實例一：

不加wraps

# -*- coding=utf-8 -*- 
from functools import wraps def my_decorator(func): def wrapper(*args, **kwargs): '''decorator'''
        print('Calling decorated function...') return func(*args, **kwargs) return wrapper @my_decorator def example(): """Docstring""" 
    print('Called example function') print(example.__name__, example.__doc__)

執行結果:

('wrapper', 'decorator')

　　

實例二：

加wraps

# -*- coding=utf-8 -*- 
from functools import wraps def my_decorator(func): @wraps(func) def wrapper(*args, **kwargs): '''decorator'''
        print('Calling decorated function...') return func(*args, **kwargs) return wrapper @my_decorator def example(): """Docstring""" 
    print('Called example function') print(example.__name__, example.__doc__)

執行結果：

('example', 'Docstring')

 

總結:

warps 作用: 消除(被裝飾后的函數名等屬性的改變)副作用