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

猜你喜欢

转载自blog.csdn.net/qq_31821675/article/details/88874547