我之前对装饰器的理解是:在不改变原有函数功能的情况下,提供新的功能,但是我还真没意识到原来函数的属性已经改变了。仔细想想也是,装饰器(或者说@语法糖)实际上就是把原函数当作参数传递给了装饰器函数,那么现在原函数的属性应该变成了装饰器函数的属性。而functools模块提供了一个wraps装饰器,它的作用就是复制原函数的函数名等属性到wrapper()函数中。下面是一个例子。
未使用wraps装饰器:
def decorator(f):
def wrapper(*args, **kwargs):
"""
11111111
"""
print('调用装饰器。。。')
return f(*args, **kwargs)
return wrapper
@decorator
def example():
"""
2222222
"""
print('调用example。。。')
example()
print(example.__name__)
print(example.__doc__)
# 结果:
#
# 调用装饰器。。。
# 调用example。。。
# wrapper
# 11111111
使用wraps装饰器:
from functools import wraps
def decorator(f):
@wraps(f)
def wrapper(*args, **kwargs):
"""
11111111
"""
print('调用装饰器。。。')
return f(*args, **kwargs)
return wrapper
@decorator
def example():
"""
2222222
"""
print('调用example。。。')
example()
print(example.__name__)
print(example.__doc__)
# 结果:
#
# 调用装饰器。。。
# 调用example。。。
# example
# 2222222
看来已学过的东西还是有很多值得挖掘的地方,还是得多想想为什么。。。