Python装饰器设计模式：为函数增添风味

news2024/12/14 16:34:51

Python装饰器设计模式：为函数增添风味

- - 什么是装饰器？
  - 为什么需要装饰器？
  - 如何使用装饰器？
  - - 示例1：简单的装饰器
    - 示例2：带参数的装饰器
  - 装饰器的使用场景
  - 总结

大家好，今天我们要学习一个非常有趣的Python特性——装饰器（Decorator）。装饰器在Python中是一个非常强大的工具，它允许我们在不改变原函数代码的情况下，给函数添加额外的功能。接下来，我将用一个生活中的比喻和示例代码来帮助你理解装饰器。

什么是装饰器？

想象一下你在饭店里点菜，可以选择加一些调料，比如辣椒酱、酱油等。这些调料不会改变菜的本质，但能提升味道。装饰器在Python中的作用就类似于这些调料，它们可以让你在不改变原函数代码的情况下，给函数添加一些额外的功能。

为什么需要装饰器？

在编程中，我们经常需要在函数执行前后添加一些额外的操作，比如：

日志记录：在函数执行前后记录日志，方便调试。
性能测试：测量函数的执行时间。
权限验证：在函数执行前检查用户是否有权限执行该函数。

装饰器可以帮助我们轻松地实现这些功能，而无需修改原函数的代码。

如何使用装饰器？

装饰器的语法非常简单，通常由@decorator_name表示，放在函数定义的上方。例如：

@my_decorator
def my_function():
    pass

这相当于：

my_function = my_decorator(my_function)

示例1：简单的装饰器

我们来写一个简单的装饰器，它在函数执行前后打印一些消息。

def my_decorator(func):
    def wrapper():
        print("开胃菜上桌...")
        func()
        print(" dessert 上桌...")
    return wrapper

@my_decorator
def main_course():
    print("主菜上桌！")

main_course()

运行结果：

开胃菜上桌...
主菜上桌！
dessert 上桌...

在这个例子中，my_decorator 是一个装饰器，它接受一个函数 func 作为参数，并返回一个新的函数 wrapper。wrapper 函数在调用 func 之前和之后分别打印了一些消息。

示例2：带参数的装饰器

有时候，我们可能希望装饰器接受参数。例如，我们想记录函数的执行时间，但只在满足某些条件时才记录。

def repeat(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                func(*args, **kwargs)
        return wrapper
    return decorator

@repeat(times=3)
def say_hello():
    print("Hello, World!")

say_hello()

运行结果：

Hello, World!
Hello, World!
Hello, World!

在这个例子中，repeat 是一个带参数的装饰器，它接受一个 times 参数，并返回一个装饰器 decorator。decorator 接受被装饰的函数 func，并返回一个新的函数 wrapper。wrapper 函数根据 times 参数重复调用 func。