首先请看如下代码

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

　　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

优质内容筛选与推荐>>
1、pandas DataFrame学习（1）
2、Creating a Physical Standby from Primary on Version 12c (Doc ID 1570958.1)
3、spring MVC的执行流程
4、线条与线条之间连接的样式
5、Scatter 散点图

赞赏

长按二维码向我转账

受苹果公司新规定影响，微信 iOS 版的赞赏功能被关闭，可通过二维码转账支持公众号。

阅读

已发送

发送中