使用Python中的装饰器实现函数扩展

装饰器是Python中一种常用的编程技术，可以在不修改被装饰函数源代码的情况下，对函数进行扩展或修改。使用装饰器可以优雅地实现一些功能，比如日志记录、性能分析、权限校验等。

在Python中，装饰器是一个函数，它接收一个函数作为参数，并返回一个新的函数。被装饰的函数会在执行之前或执行之后，调用装饰器返回的函数来进行扩展。

接下来，我们将使用Python中的装饰器来实现一个函数扩展的例子。假设我们有一个计算函数，用于计算两个数的和：

def add(a, b):
    return a + b

现在我们希望在这个函数执行之前输出一条日志，表示函数开始执行；在函数执行完成之后输出一条日志，表示函数执行结束。我们可以使用装饰器来实现这个功能：

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print("函数开始执行")
        result = func(*args, **kwargs)
        print("函数执行结束")
        return result
    return wrapper

@log_decorator
def add(a, b):
    return a + b

在上面的例子中，我们定义了一个名为log_decorator的装饰器函数，它接收一个函数作为参数（这里是add函数），并返回了一个新的函数wrapper。在新的函数wrapper中，我们先输出一条日志，表示函数开始执行，然后调用原始的add函数，将结果保存在result变量中，最后再输出一条日志，表示函数执行结束，并返回结果。

为了使用log_decorator装饰器函数，我们使用@符号将其应用到了add函数上。这样，当我们调用add函数时，实际上是调用了由log_decorator返回的新函数wrapper。

现在，让我们来使用这个装饰器来扩展add函数的功能：

result = add(3, 5)
print(result)  # 输出：8

在执行上面的代码之后，我们会得到以下输出：

函数开始执行
函数执行结束
8

可以看到，在调用add函数时，log_decorator装饰器会先输出一条日志，然后执行原始的add函数，并将结果保存在result变量中，最后再输出一条日志。

这个例子演示了如何使用装饰器来实现函数扩展。通过定义一个装饰器函数，并将其应用到目标函数上，可以在不修改目标函数源代码的情况下，对目标函数进行功能扩展。装饰器是Python语言中的一项强大工具，可以提高代码的可维护性和可扩展性，值得在实际开发中广泛应用。