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