Python - 装饰器之 @functools.wraps(func)

本文主要是介绍Python - 装饰器之 @functools.wraps(func)，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

@functools.wraps(func)

用在装饰器函数之上，用于写装饰器时，保留原有函数的名称和函数属性。

　　# 定义一个最简单的装饰器
　　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

    # 输出结果：
　　　　num1
　　　　num2

这篇关于Python - 装饰器之 @functools.wraps(func)的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！