python装饰器中functools.wraps的作用详解

发布时间：2020-12-20 10:44:33 所属栏目：Python 来源：网络整理

导读：# 定义一个最简单的装饰器 def user_login_data(f): def wrapper(*args,**kwargs): return f(*args,**kwargs) return wrapper # 用装饰器装饰以下两个函数 @user_login_data def num1(): print("aaa") @user_login_data def num2(): print("bbbb") if __name

　　# 定义一个最简单的装饰器

　　def user_login_data(f):
　　　　def wrapper(*args,**kwargs):
　　　　　　return f(*args,**kwargs)

　　　　return wrapper

　　# 用装饰器装饰以下两个函数
　　
　　@user_login_data
　　def num1():
　　　　print("aaa")



　　@user_login_data
　　def num2():
　　　　print("bbbb")

　　if __name__ == ‘__main__‘:
　　　　print(num1.__name__)
　　　　print(num2.__name__)

　以上代码的输出结果为:

　　　　wrapper

　　　　wrapper

由此函数使用装饰器时,函数的函数名即 __name__已经被装饰器改变.

一般定义装饰器的话可以不用考虑这点,但是如果多个函数被两个装饰器装饰时就报错,因为两个函数名一样,第二个函数再去装饰的话就报错.

解决方案就是引入? functools.wraps?,以上代码的解决如下:?

　　　　

　　　　def user_login_data(f):
　　　　@functools.wraps(f)
　　　　　　def wrapper(*args,**kwargs):
　　　　　　　　return f(*args,**kwargs)

　　　　　　return wrapper

增加@functools.wraps(f),?可以保持当前装饰器去装饰的函数的 __name__ 的值不变

以上输出结果就是:?

　　　　num1

　　　　num2

Python装饰器（decorator）在实现的时候，有一些细节需要被注意。例如，被装饰后的函数其实已经是另外一个函数了（函数名等函数属性会发生改变）。这样有时候会对程序造成一些不便，例如笔者想对flask框架中的一些函数添加自定义的decorator，添加后由于函数名和函数的doc发生了改变，对测试结果有一些影响。

所以，Python的functools包中提供了一个叫wraps的decorator来消除这样的副作用。写一个decorator的时候，最好在实现之前加上functools的wrap，它能保留原有函数的名称和docstring。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!