Python中的装饰器

Python作为一门动态且富有表现力的编程语言，其设计哲学之一是“简洁胜于复杂”。在追求简洁的道路上，装饰器（Decorators）无疑是一个闪耀的明星，它为开发者提供了一种优雅的方式去包装、扩展函数或方法的功能。
首先，让我们来理解什么是装饰器。简而言之，装饰器就是一个函数，它接受一个函数作为参数并返回一个新的函数，通常这个新函数会包含额外的功能。在Python中，装饰器使用@符号来表示。
下面是一个最基础的装饰器示例：

def my_decorator(func):
    def wrapper():
        print("Something is happening before the function is called.")
        func()
        print("Something is happening after the function is called.")
    return wrapper
@my_decorator
def say_hello():
    print("Hello!")
say_hello()

在这个例子中，my_decorator就是一个装饰器。当我们用@my_decorator修饰say_hello函数时，实际上是将say_hello函数作为参数传递给了my_decorator。然后，my_decorator返回了一个新的函数wrapper，而这个新的函数在调用原始函数之前后添加了一些额外的行为。
现在，我们来深入探讨装饰器的工作原理。当Python解释器遇到@符号时，它会将后面的函数（比如say_hello）作为参数传递给前面的函数（比如my_decorator）。然后，被装饰的函数（say_hello）不再是原始的那个函数，而是装饰器返回的新函数（wrapper）。因此，当我们调用say_hello()时，实际上执行的是wrapper()。
装饰器的强大之处在于它的灵活性。我们可以使用装饰器来实现日志记录、性能测试、访问控制等功能，而不需要修改原始函数的代码。这对于保持代码的整洁性和可维护性至关重要。
例如，假设我们需要在多个函数中添加日志记录功能，而不是在每个函数中重复编写日志代码，我们可以创建一个日志装饰器：

import functools
def log_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print(f"Calling {func.__name__}")
        value = func(*args, **kwargs)
        print(f"{func.__name__} returned {value}")
        return value
    return wrapper
@log_decorator
def add(x, y):
    return x + y
add(1, 2)

在这个例子中，我们定义了一个log_decorator装饰器，它会在调用被装饰的函数前后打印日志信息。通过使用functools.wraps，我们还保留了原始函数的名称和文档字符串。