Python函数装饰器的用法

装饰器是Python中一种强大的语法特性，它允许在不修改已有代码的情况下，对函数或类进行增强或修改。装饰器的本质是一个函数，它接受一个函数作为参数，并返回一个新的函数。

---

1.装饰器的优点

通过使用装饰器，我们可以在不改变原函数定义的情况下，在函数执行前后添加额外的功能。这样做具有以下几个优点：

1. 代码复用：可以将一些通用的功能抽象出来，作为装饰器应用到多个函数上，避免了重复编写相同的代码。
2. 解耦合：装饰器将原函数与额外的功能解耦合，使得每个函数只关注自己的核心逻辑，提高了代码的可读性和可维护性。
3. 可扩展性：通过使用装饰器，我们可以方便地在已有函数上添加新的功能，而不需要修改原函数的定义，从而遵循了开放封闭原则。

下面我们以一个示例来说明装饰器的使用。

---

2. 使用装饰器前

假设我们有一个计算函数执行时间的需求，在使用装饰器前，可以这样实现：

import timedef my_function():# 记录开始时间start_time = time.time()# 模拟函数执行过程time.sleep(2)print("函数执行完毕")# 记录结束时间并计算执行时间end_time = time.time()print(f"函数的执行时间为：{end_time - start_time}秒")# 调用原函数
my_function()

输出结果：
            

---

3. 使用装饰器后

此外，我们还可以使用装饰器来实现。

import time# 定义一个装饰器函数
def calculate_time(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 resultreturn wrapper# 原函数
@calculate_time
def my_function():# 模拟函数执行过程time.sleep(2)print("函数执行完毕")# 调用原函数
my_function()

输出结果：
            

---

在上述代码中，将计算函数执行时间的功能定义为了装饰器函数calculate_time，
它接受一个函数作为参数，并返回一个新的函数wrapper。
wrapper函数实现了在原函数执行前后记录时间的功能。

通过在原函数my_function之前加上@calculate_time，我们将装饰器应用到了该函数上。当调用my_function时，实际上调用的是装饰后的函数wrapper，它会在执行原函数前后记录时间并打印出来。这样的写法也称装饰器的语法糖写法。

这样以来，原函数的功能和计算执行时间的功能被分离开，计算执行时间的函数被视为拓展功能。想要拓展其他功能只需继续新建其他装饰器函数进行拓展即可，而不必对原函数大动干戈。

---

通过对比使用装饰器前后的代码，我们可以看到使用装饰器后的代码更加简洁，而且将计算时间的逻辑与原函数分离开来，提高了代码的可读性和可维护性。此外，如果在其他函数中也需要统计执行时间，只需简单地加上装饰器即可，避免了重复编写相同的计时代码。这就是装饰器的优越性所在。