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
上一篇:functools.wraps装饰器的作用


下一篇:2021-04-16