Python 装饰器：代码复用的优雅之道

Python 装饰器是一个强大的特性，它允许你以一种优雅的方式修改或增强函数的功能，而无需改动函数本身的定义。装饰器在很多场景下都非常有用，例如日志记录、性能分析、权限控制等。

你是否曾为代码中重复的逻辑而烦恼？是否想在不修改原有函数的情况下，为其添加额外的功能？Python 装饰器正是解决这类问题的利器。

1. 装饰器的基本原理

装饰器本质上是一个函数，它接受一个函数作为输入，并返回一个新的函数。这个新的函数通常会在调用原函数的基础上，添加一些额外的操作。

让我们从一个简单的例子开始：

def my_decorator(func):
    def wrapper():
        print("Before calling the function.")
        func()
        print("After calling the function.")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

say_hello()

输出结果：

Before calling the function.
Hello!
After calling the function.

在这个例子中，my_decorator 就是一个装饰器。它接受 say_hello 函数作为输入，并返回一个新的函数 wrapper。wrapper 函数在调用 say_hello 函数前后分别打印了一条信息。@my_decorator 语法糖等价于 say_hello = my_decorator(say_hello)。

2. 带参数的装饰器

有时候，我们需要向装饰器传递一些参数。这时，我们可以定义一个返回装饰器的函数：

def repeat(num_times):
    def decorator_repeat(func):
        def wrapper(*args, **kwargs):
            for _ in range(num_times):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator_repeat

@repeat(num_times=3)
def greet(name):
    print(f"Hello, {name}!")

greet("Alice")

输出结果：

Hello, Alice!
Hello, Alice!
Hello, Alice!

在这个例子中，repeat 函数返回一个装饰器 decorator_repeat，decorator_repeat 接受一个函数作为输入，并返回一个新的函数 wrapper。wrapper 函数会重复调用原函数 num_times 次。

3. 装饰器的实际应用

• 日志记录： 记录函数的调用时间、参数、返回值等信息。
• 性能分析： 计算函数的执行时间，找出性能瓶颈。
• 权限控制： 检查用户是否有权限执行某个函数。
• 缓存： 缓存函数的返回值，避免重复计算。
• 输入验证： 验证函数的输入参数是否合法。

4. 类装饰器

除了函数装饰器，还可以使用类来实现装饰器。类装饰器通常用于需要保存状态的场景。

总结

Python 装饰器是一种强大而灵活的代码复用机制。它可以在不修改原有函数的情况下，为其添加额外的功能，使代码更加清晰、简洁、易于维护。掌握装饰器的使用，可以让你编写出更优雅、更强大的 Python 代码。