1. 程式人生 > >python裝飾器中@wraps作用--修復被裝飾後的函式名等屬性的改變

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 作用: 消除(被裝飾後的函式名等屬性的改變)副作用