在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} 秒")
        return result
    return wrapper

@timer
def my_function():
    # 添加一些需要执行的操作
    time.sleep(2)
    return

my_function()

在上面的代码中，定义了一个timer装饰器函数，它接受一个函数作为参数，并返回一个新的函数。新的函数wrapper在调用原始函数之前和之后，分别记录下当前时间，并计算时间差。最后，打印出函数执行时间，并返回原始函数的结果。

然后，我们使用@timer语法来向函数my_function应用装饰器。这样，当我们调用my_function时，实际上会执行装饰器函数timer中的wrapper函数。

通过以上的装饰器的使用，我们可以很方便地为函数添加一些额外的功能，比如计时、日志等等。只需要在需要增强功能的函数上应用相应的装饰器即可，不需要修改原始函数的代码。

除了以上的计时装饰器示例外，还可以根据需求自定义各种不同的装饰器来增强函数的功能。装饰器是Python中非常强大和常用的特性之一，可以让我们在不修改原始函数的情况下，为函数增加各种额外的功能，提高代码复用性和可维护性。