Python 装饰器 @wraps()

装饰器（Decorator）

在 Python 中，装饰器是一个特殊的函数，它可以修改或扩展另一个函数的行为。装饰器是一个小函数，它接受另一个函数作为参数，并返回一个新的函数，该新的函数“包装”了原始函数。

元数据丢失问题

当你创建一个装饰器时，你实际上是创建了一个新的函数，该函数调用原始函数。然而，这个新的函数不会继承原始函数的元数据（例如名称、文档字符串、参数）。这可能会导致问题，例如使用 help() 或 inspect 模块来检查函数的元数据时。

wraps 的作用

wraps 装饰器来自 functools 模块，它可以解决元数据丢失的问题。当你使用 wraps 来装饰一个装饰器时，它将确保原始函数的元数据被复制到新的函数中。

让我们来探讨如果不使用 @wraps() 装饰器时会发生什么情况。😊

不使用 @wraps() 的结果

如果不使用 @wraps() 装饰器，原始函数的元数据（例如函数名、文档字符串、参数签名）将不被保留。这可能会导致以下问题：

1. 函数名和文档字符串丢失：装饰后的函数将具有不同的函数名和文档字符串，这可能会使函数的目的和行为不明确。
2. 参数签名不正确：装饰后的函数的参数签名将不匹配原始函数的参数签名，这可能会导致调用函数时的错误。
3. 调试困难：当错误发生时，调试器将显示装饰后的函数名和签名，而不是原始函数的名称和签名，使得错误的来源难以确定。

代码示例

def MyDecorator(func):
    def wrapper(*args, **kwargs):
        print("函数调用前发生了什么。")
        result = func(*args, **kwargs)
        print("函数调用后发生了什么。")
        return result
    return wrapper

@MyDecorator
def add(a, b):
    """加两个数字"""
    return a + b

print(add.__name__)  # 输出：wrapper
print(add.__doc__)  # 输出：None
help(add)  # 输出：没有文档可用

如你所见，装饰后的函数 加 丢失了其原始函数名和文档字符串。这可能会导致代码维护和调试困难。

使用 @wraps() 的结果

相比之下，如果使用 @wraps() 装饰器，原始函数的元数据将被保留，避免了上述问题。

from functools import wraps

def MyDecorator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print("函数调用前发生了什么。")
        result = func(*args, **kwargs)
        print("函数调用后发生了什么。")
        return result
    return wrapper

@MyDecorator
def add(a, b):
    """加两个数字"""
    return a + b

print(add.__name__)  # 输出：add
print(add.__doc__)  # 输出：加两个数字
help(add)  # 输出：正确的文档

使用 @wraps() 后，装饰后的函数 add 保留了其原始函数名和文档字符串，避免了代码维护和调试困难。