python wraps模块_python中 wraps 的作用

这里使用两段代码比较加入wraps装饰器后，函数打印的结果对比：

新建文件名：Testword

代码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：加入wrapper函数后

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确保被装饰的函数不会因装饰器带来异常情况。