新手快速理解装饰器，一篇文章就够了！

2023-03-18 00:31:29

用装饰器增强 Python 函数

Python 装饰器是一种强大的工具，它允许您在不修改原始代码的情况下，为函数或类添加额外的功能。就像魔法一样，它们提供了改变函数行为的简单方法。

装饰器的本质

想象一下，您希望在每个函数调用之前或之后执行一些特殊操作。如果没有装饰器，您将需要修改每个函数来包含这些操作。装饰器提供了更优雅的解决方案，它可以让您在不触及原始代码的情况下，轻松实现这种功能扩展。

装饰器的语法很简单：

@decorator
def function_to_decorate():
    # 函数代码

这里，@decorator是装饰器，而function_to_decorate是要装饰的函数。当function_to_decorate被调用时，装饰器@decorator首先执行，然后才是function_to_decorate本身的代码。

装饰器的应用

装饰器拥有广泛的应用场景，其中包括：

性能测量： 测量函数的运行时间，帮助您识别代码中的瓶颈。
函数调用监控： 跟踪函数的调用次数，了解其使用情况和效率。
缓存机制： 为函数调用结果添加缓存，以提高性能并减少计算开销。
安全检查： 在函数执行前进行安全检查，防止未经授权的访问或数据篡改。
日志记录： 为函数添加日志记录功能，方便调试和故障排除。

创建自己的装饰器

创建自定义装饰器也非常简单。只需定义一个函数，并在其前面加上@符号即可。例如，下面是一个用于计算函数运行时间的装饰器：

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time()
        print('Elapsed time: {}'.format(end - start))
        return result
    return wrapper

@timer
def calculate_factorial(n):
    if n == 0:
        return 1
    else:
        return n * calculate_factorial(n-1)

calculate_factorial(5)