这里使用两段代码比较加入wraps装饰器后,函数打印的结果对比:
代码1:不加wraps装饰器
# coding=utf-8
from functools import wraps
def my_decorator(func):
def wrapper(*args, **kwargs):
'''decorator'''
print('Decorated function...')
return func(*args, **kwargs)
return wrapper
@my_decorator
def test():
"""Testword"""
print('Test function')
print(test.__name__, test.__doc__)
打印结果:
wrapper decorator
[Finished in 0.1s]
代码2:加入@wraps(func)
装饰器后后
from functools import wraps
def my_decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
'''decorator'''
print('Decorated function...')
return func(*args, **kwargs)
return wrapper
@my_decorator
def test():
"""Testword"""
print('Test function')
print(test.__name__, test.__doc__)
打印结果:
test Testword
[Finished in 0.1s]
总结:
因为当使用装饰器装饰一个函数时,函数本身就已经是一个新的函数;即函数名称或属性产生了变化。所以在python的functools
模块中提供了wraps
装饰函数来确保原函数在使用装饰器时不改变自身的函数名及应有属性。
所以在装饰器的编写中建议加入wraps
确保被装饰的函数不会因装饰器带来异常情况。