functools中的wraps装饰器

我之前对装饰器的理解是：在不改变原有函数功能的情况下，提供新的功能，但是我还真没意识到原来函数的属性已经改变了。仔细想想也是，装饰器（或者说@语法糖）实际上就是把原函数当作参数传递给了装饰器函数，那么现在原函数的属性应该变成了装饰器函数的属性。而functools模块提供了一个wraps装饰器，它的作用就是复制原函数的函数名等属性到wrapper()函数中。下面是一个例子。

未使用wraps装饰器：

def decorator(f):
    def wrapper(*args, **kwargs):
        """
        11111111
        """
        print('调用装饰器。。。')
        return f(*args, **kwargs)

    return wrapper


@decorator
def example():
    """
    2222222
    """
    print('调用example。。。')


example()
print(example.__name__)
print(example.__doc__)
# 结果：
#
# 调用装饰器。。。
# 调用example。。。
# wrapper
# 11111111

使用wraps装饰器：

from functools import wraps


def decorator(f):
    @wraps(f)
    def wrapper(*args, **kwargs):
        """
        11111111
        """
        print('调用装饰器。。。')
        return f(*args, **kwargs)

    return wrapper


@decorator
def example():
    """
    2222222
    """
    print('调用example。。。')


example()
print(example.__name__)
print(example.__doc__)
# 结果：
#
# 调用装饰器。。。
# 调用example。。。
# example
# 2222222

看来已学过的东西还是有很多值得挖掘的地方，还是得多想想为什么。。。