标准库functools.wraps的使用方法

Posted liurenli

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了标准库functools.wraps的使用方法相关的知识,希望对你有一定的参考价值。

首先请看如下代码

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

  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

以上是关于标准库functools.wraps的使用方法的主要内容,如果未能解决你的问题,请参考以下文章

Python 中实现装饰器时使用 @functools.wraps 的理由

Python functools.wraps 等价于类

python functools.wraps 实例

functools.wraps 等价于类装饰器

functools.wraps

python functools.wraps