您的位置:首页 > 编程语言 > Python开发

python装饰器中@wraps作用--修复被装饰后的函数名等属性的改变

2019-01-01 22:58 218 查看

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 作用: 消除(被装饰后的函数名等属性的改变)副作用

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: