### 装饰器的基本概念 装饰器是一种特殊类型的函数，它接受另一个函数作为参数，并返回一个新的函数。装饰器的主要作用是在不修改原函数代码的情况下，为其添加新的功能。在 Python 中，装饰器通过 `@` 符号进行语法糖式的调用，使得代码更加简洁和易读。 ### 无参装饰器示例 下面是一个简单的无参装饰器示例，用于记录函数的执行时间： ```python import time def timer(func): def wrapper(*args, **kwargs): start_time = time.time() result = func(*args, **kwargs) end_time = time.time() print(f"函数 {func.__name__} 执行耗时: {end_time - start_time:.4f} 秒") return result return wrapper @timer def my_function(): time.sleep(1) print("函数执行完毕") my_function() ``` 在这个例子中，`timer` 是一个装饰器函数，它包裹了 `my_function`，并在其执行前后打印了时间信息。`*args` 和 `**kwargs` 允许装饰器处理任意数量的位置参数和关键字参数 [^3]。 ### 有参装饰器示例 当装饰器本身需要参数时，我们需要编写一个返回装饰器函数的函数。下面是一个有参装饰器的示例，该装饰器会在函数执行前后打印指定的消息： ```python def message_decorator(msg): def decorator(func): def wrapper(*args, **kwargs): print(f"装饰器消息: {msg}") print("函数开始执行") result = func(*args, **kwargs) print("函数执行结束") return result return wrapper return decorator @message_decorator("这是一个装饰器消息") def greet(name): print(f"你好, {name}") greet("张三") ``` 在这个例子中，`message_decorator` 是一个带有参数 `msg` 的函数，它返回了一个装饰器 `decorator`，而 `decorator` 又返回了 `wrapper` 函数来实际包裹目标函数 [^2]。 ### 多个装饰器的应用 可以在一个函数上应用多个装饰器，它们会按照从内到外的顺序依次应用。例如： ```python def decorator1(func): def wrapper(*args, **kwargs): print("装饰器1 - 前") result = func(*args, **kwargs) print("装饰器1 - 后") return result return wrapper def decorator2(func): def wrapper(*args, **kwargs): print("装饰器2 - 前") result = func(*args, **kwargs) print("装饰器2 - 后") return result return wrapper @decorator1 @decorator2 def say_hello(): print("你好") say_hello() ``` 在这个例子中，`say_hello` 函数同时被 `decorator1` 和 `decorator2` 装饰。执行时，`decorator2` 会先于 `decorator1` 被调用，但 `decorator1` 的前置操作会在 `decorator2` 的前置操作之后执行 [^1]。 ### 类作为装饰器 除了函数，类也可以作为装饰器使用。类装饰器通常需要实现 `__call__` 方法，以便实例可以像函数一样被调用： ```python class ClassDecorator: def __init__(self, func): self.func = func def __call__(self, *args, **kwargs): print("类装饰器 - 前") result = self.func(*args, **kwargs) print("类装饰器 - 后") return result @ClassDecorator def hello(): print("Hello") hello() ``` 在这个例子中，`ClassDecorator` 是一个类，它的 `__call__` 方法定义了装饰器的行为。当 `hello` 函数被 `@ClassDecorator` 装饰时，`hello` 会被替换为 `ClassDecorator(hello)` 的实例 [^4]。 ### 总结 装饰器是 Python 中非常强大的工具，它允许你在不修改原有函数代码的情况下扩展其行为。无论是无参装饰器、有参装饰器，还是多个装饰器的组合使用，都可以通过嵌套函数或类来实现。掌握装饰器的使用可以让你的代码更加简洁、灵活和 Pythonic [^1]。