### 创建包装器（Wrapper）的基础概念 在 Python 中，包装器是一种设计模式，常用于扩展或修改函数或类的功能而不改变其原始定义。`@wraps` 是 `functools` 模块中的一个装饰器，主要用于保留被装饰函数的元数据[^1]。 以下是构建函数和类的包装器的具体实现方式： --- ### 函数包装器示例 #### 定义包装器 下面是一个简单的函数包装器示例，展示了如何使用 `@wraps` 来保持原函数的元数据： ```python from functools import wraps def my_decorator(func): @wraps(func) # 复制 func 的元数据到 wrapper 函数中 def wrapper(*args, **kwargs): print(f"Before calling {func.__name__}") result = func(*args, **kwargs) print(f"After calling {func.__name__}") return result return wrapper @my_decorator def greet(name): """A simple greeting function.""" print(f"Hello, {name}!") greet("Alice") # 输出 Before/After 和 Hello Alice! print(greet.__doc__) # 输出 A simple greeting function. ``` 上述代码中，`@wraps(func)` 确保了 `greet` 函数的文档字符串 (`__doc__`) 不会被覆盖为 `wrapper` 的默认值。 --- ### 类包装器示例 对于类而言，可以通过装饰器来增强类的行为。例如，可以记录类的实例化次数或者提供额外的方法。 #### 记录类实例化的次数 以下是一个基于类方法的计数器示例，展示如何通过装饰器实现功能扩展： ```python class CounterDecorator: def __init__(self, cls): self.cls = cls self.count = 0 def __call__(self, *args, **kwargs): self.count += 1 print(f"{self.cls.__name__} has been instantiated {self.count} times.") return self.cls(*args, **kwargs) @CounterDecorator class MyClass: def __init__(self, value): self.value = value obj1 = MyClass(10) # 输出 MyClass has been instantiated 1 times. obj2 = MyClass(20) # 输出 MyClass has been instantiated 2 times. ``` 在此示例中，`CounterDecorator` 被用作类的包装器，每次实例化 `MyClass` 都会更新并打印计数器的值[^3]。 --- ### 利用装饰器实现函数重载的效果 虽然 Python 并不支持传统意义上的函数重载，但可以通过装饰器动态调整函数行为以模拟这一特性。以下是一个简单示例： ```python from functools import wraps def overload(func): registry = {} @wraps(func) def wrapper(*args, **kwargs): key = (len(args), frozenset(kwargs.items())) specific_func = registry.get(key, func) return specific_func(*args, **kwargs) def register(new_func): nonlocal registry signature_key = (new_func.__code__.co_argcount, new_func.__defaults__) registry[signature_key] = new_func return new_func wrapper.register = register return wrapper @overload def add(a, b): return a + b @add.register def _(a, b, c): return a + b + c print(add(1, 2)) # 输出 3 print(add(1, 2, 3)) # 输出 6 ``` 此代码片段演示了如何通过注册不同签名的函数版本来实现类似函数重载的效果[^4]。 --- ### 总结 - 对于函数包装器，推荐使用 `@wraps` 来确保元数据一致性。 - 对于类包装器，可以通过自定义装饰器实现更复杂的功能，比如统计实例化次数。 - 可以利用装饰器机制模拟函数重载效果，从而灵活应对不同的输入场景。 ---