在Python中，函数装饰器是一种强大且优雅的语法结构，它允许你在不修改原始函数定义的情况下，**动态地增加或修改函数的行为**。理解装饰器的关键在于掌握其核心机制：**闭包** 和 **函数作为一等对象**。它本质上是一个“包装器”，在目标函数执行前后插入额外的逻辑。 ### 一、装饰器的核心：理解闭包 装饰器的实现依赖于**闭包**。闭包是指在一个内部函数中引用了其外部函数的局部变量，并且这个内部函数被返回，使得这些局部变量即使在外层函数执行完毕后依然能被访问和保持状态 [ref_2]。 ```python def outer_func(msg): # 外部函数的局部变量 message = msg def inner_func(): # 内部函数引用了外部函数的局部变量 message print(f"Message: {message}") # [ref_2] 闭包形成的关键：内函数使用外函数的变量 # 返回内部函数，形成闭包 return inner_func # 创建闭包实例 my_closure = outer_func("Hello, Closure!") my_closure() # 输出: Message: Hello, Closure! ``` ### 二、从闭包到装饰器：基本结构 一个最简单的装饰器，就是一个接收函数作为参数，并返回一个新函数的闭包。 ```python def my_decorator(func): """ 一个简单的装饰器，在函数执行前后打印日志。 """ def wrapper(): print(f"准备执行函数: {func.__name__}") # [ref_3] 在被装饰函数执行前添加逻辑 func() # 执行原始函数 print(f"函数 {func.__name__} 执行完毕。") # [ref_3] 在被装饰函数执行后添加逻辑 return wrapper # [ref_4] 返回包装后的函数 @my_decorator # 这是装饰器的语法糖，等价于 say_hello = my_decorator(say_hello) def say_hello(): print("Hello, World!") say_hello() # 输出: # 准备执行函数: say_hello # Hello, World! # 函数 say_hello 执行完毕。 ``` ### 三、装饰器的分类与高级用法 根据功能需求，装饰器有多种形态。 #### 1. 通用装饰器：处理带参数的函数 为了使装饰器能装饰任意参数的函数，需要使用 `*args` 和 `**kwargs` 来接收可变参数 [ref_4]。 ```python def universal_decorator(func): """ 一个通用的装饰器，可以装饰任何参数的函数。 """ def wrapper(*args, **kwargs): # [ref_4] 使用可变参数接收所有传入参数 print(f"调用函数: {func.__name__}， 参数: {args}, {kwargs}") result = func(*args, **kwargs) # [ref_4] 将参数原样传递给原始函数 print(f"函数 {func.__name__} 返回: {result}") return result return wrapper @universal_decorator def add(a, b): return a + b print(add(5, 3)) # 输出调用和返回信息，并返回 8 ``` #### 2. 带参数的装饰器 有时装饰器本身也需要参数来配置其行为。这需要再嵌套一层函数 [ref_5]。 ```python def repeat(times): """ 一个带参数的装饰器，用于重复执行函数指定次数。 """ def decorator(func): # 这一层接收被装饰的函数 def wrapper(*args, **kwargs): results = [] for i in range(times): print(f"第 {i+1} 次执行") result = func(*args, **kwargs) results.append(result) return results return wrapper return decorator # [ref_5] 带参装饰器返回的是真正的装饰器函数 @repeat(times=3) def greet(name): return f"Hello, {name}!" print(greet("Alice")) # 输出: # 第 1 次执行 # 第 2 次执行 # 第 3 次执行 # ['Hello, Alice!', 'Hello, Alice!', 'Hello, Alice!'] ``` #### 3. 类装饰器 除了函数，类也可以作为装饰器使用，这通常用于需要维护状态的场景 [ref_5]。 ```python class CallCounter: """ 一个类装饰器，用于记录函数被调用的次数。 """ def __init__(self, func): self.func = func self.count = 0 def __call__(self, *args, **kwargs): # [ref_5] 使实例可以像函数一样被调用 self.count += 1 print(f"{self.func.__name__} 已被调用第 {self.count} 次") return self.func(*args, **kwargs) @CallCounter def say_hi(): print("Hi!") say_hi() # 输出: say_hi 已被调用第 1 次 \n Hi! say_hi() # 输出: say_hi 已被调用第 2 次 \n Hi! ``` ### 四、装饰器的实际应用场景与注意事项 装饰器在开发中用途广泛，以下是一些典型场景： | 应用场景 | 描述 | 简单示例 | | :--- | :--- | :--- | | **日志记录** | 自动记录函数的调用、参数和返回值，用于调试和监控。 | 如上文的 `universal_decorator` | | **性能测试** | 计算函数的运行时间，优化代码性能。 | 使用 `time` 模块在 `wrapper` 中计时 [ref_6] | | **权限校验** | 在Web开发中，检查用户是否有权限访问某个接口。 | 在 `wrapper` 中检查 `session` 或 `token` | | **数据验证/格式化** | 在函数执行前验证输入数据，或在执行后格式化输出数据。 | 对 `*args, **kwargs` 进行预处理或后处理 | | **函数注册** | 将函数自动注册到某个列表或字典中，实现插件系统。 | 在装饰器中将 `func` 添加到全局注册表 | **重要注意事项：使用 `functools.wraps`** 装饰器会“覆盖”原始函数的元信息（如函数名 `__name__`、文档字符串 `__doc__`），这会给调试带来困难。使用 `functools.wraps` 装饰器可以解决这个问题 [ref_3][ref_5]。 ```python import functools def proper_decorator(func): @functools.wraps(func) # [ref_3] 使用 wraps 保留被装饰函数的元数据 def wrapper(*args, **kwargs): """Wrapper function doc.""" print("Do something before.") return func(*args, **kwargs) return wrapper @proper_decorator def example(): """This is the original function.""" pass print(example.__name__) # 输出: 'example'， 而不是 'wrapper' print(example.__doc__) # 输出: 'This is the original function.'， 而不是 'Wrapper function doc.' ``` ### 五、理解装饰器的执行顺序 当多个装饰器堆叠时，它们是从下往上（从内到外）应用的 [ref_3]。 ```python def decorator_one(func): def wrapper(): print("Decorator One - Before") func() print("Decorator One - After") return wrapper def decorator_two(func): def wrapper(): print("Decorator Two - Before") func() print("Decorator Two - After") return wrapper @decorator_one @decorator_two def target_function(): print("Target Function executing.") target_function() # 输出: # Decorator One - Before # Decorator Two - Before # Target Function executing. # Decorator Two - After # Decorator One - After # 执行顺序等价于：decorator_one(decorator_two(target_function)) ``` **总结**：函数装饰器是Python中实现**面向切面编程（AOP）** 思想的利器。它通过闭包机制，将核心业务逻辑与横切关注点（如日志、权限、性能）**解耦**，极大地提高了代码的**模块化、可复用性和可维护性**。从理解闭包开始，逐步掌握基本装饰器、带参装饰器和类装饰器，并善用 `functools.wraps`，你就能在项目中优雅地运用这一强大工具。