python中wraps的作用

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yuyexiaohan/article/details/82860807

这里使用两段代码比较加入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:加入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/yuyexiaohan/article/details/82860807