Python装饰器深入浅出

2023-11-14 20:02:39

装饰器：Python 中增强函数行为的强大工具

在软件开发中，我们经常需要对现有的函数进行修改，以满足特定的需求。这可能会很麻烦，特别是当我们不想修改原始函数的代码时。这就是 Python 装饰器的用武之地。

什么是装饰器？

装饰器是一种特殊的函数，它用于修改另一个函数的行为，而无需更改原始函数的代码。本质上，它是一个包装器，它接收一个函数作为输入，并返回一个增强后的函数作为输出。

理解装饰器语法

Python 中装饰器的语法非常简单：

@decorator_function
def original_function():
    # 原始函数代码

在这里，@decorator_function 表示 original_function 由 decorator_function 装饰。decorator_function 返回一个包装器函数，该函数包含原始函数的全部功能，以及由装饰器添加的附加功能。

装饰器示例

以下是一个简单的装饰器示例，它记录函数的执行时间：

import time

def timing_decorator(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print(f'Function {func.__name__} executed in {end - start} seconds')
        return result
    return wrapper

@timing_decorator
def my_function(n):
    # 函数代码
    for i in range(n):
        print(i)

my_function(1000000)

在这个示例中，timing_decorator 装饰了 my_function。当调用 my_function 时，包装器函数 wrapper 将执行，它会在调用 my_function 之前记录开始时间，并在调用后记录结束时间。然后，它将打印出函数执行时间，并返回函数的原始结果。

装饰器的更高级用法

除了记录执行时间，装饰器还可以用于以下方面：