装饰器：轻松修改函数功能，留下精简代码

后端

2023-11-04 04:00:06

Python 中级知识：装饰器

装饰器是 Python 中的一个强大功能，允许您在不修改函数原始代码的情况下修改函数的行为。这在许多情况下非常有用，例如：

添加额外的功能或行为到函数中，而无需更改函数本身。
测量函数的运行时间或内存使用情况。
缓存函数的输出，以提高性能。
控制函数的访问权限。

装饰器的工作原理

装饰器本质上是函数，它接受一个函数作为参数，并返回一个新的函数。这个新函数包含了原函数的功能，以及装饰器添加的额外功能。当您调用装饰后的函数时，实际上是在调用装饰器返回的新函数。

使用装饰器

要使用装饰器，您需要在函数定义之前添加 @ 符号，然后指定装饰器的名称。例如：

def my_decorator(func):
    def wrapper(*args, **kwargs):
        # 在函数执行前做一些事情
        result = func(*args, **kwargs)
        # 在函数执行后做一些事情
        return result
    return wrapper

@my_decorator
def my_function():
    print("Hello, world!")

在这个例子中，my_decorator 是装饰器，它接受 my_function 作为参数，并返回一个新的函数 wrapper。wrapper 函数包含了 my_function 的功能，以及装饰器添加的额外功能。当您调用 my_function 时，实际上是在调用 wrapper 函数。

装饰器的使用场景

装饰器可以用于许多不同的场景，其中一些最常见的包括：