Python 装饰器函数：让代码更简短、更强大！

2023-12-25 16:00:45

Python装饰器函数是一种用于修改函数或类的方法的强大工具。装饰器函数可以添加到任何函数或类的方法前面，以改变函数或方法的行为。这是一种非常强大的技术，可以用来简化代码，使其更容易维护和扩展。

装饰器函数的运作原理

装饰器函数的工作原理很简单。当您在函数或类的方法前面添加一个装饰器函数时，装饰器函数会被立即调用，并将函数或类的方法作为参数传递给它。装饰器函数可以对函数或类的方法进行任意修改，然后返回一个新的函数或类的方法。这个新的函数或类的方法将取代原来的函数或类的方法。

装饰器函数的常见用法

装饰器函数可以用于各种各样的目的。一些常见的用途包括：

添加功能 ：装饰器函数可以向函数或类的方法添加新的功能。例如，您可以使用装饰器函数来添加日志记录、性能分析或缓存。
修改行为 ：装饰器函数可以修改函数或类的方法的行为。例如，您可以使用装饰器函数来改变函数或类的方法的参数、返回值或异常处理。
简化代码 ：装饰器函数可以用来简化代码。例如，您可以使用装饰器函数来消除重复的代码或将复杂的代码封装成一个单独的函数。

装饰器函数的示例

以下是一个使用装饰器函数来添加日志记录功能的示例：

def log(func):
    def wrapper(*args, **kwargs):
        print("Calling function:", func.__name__)
        result = func(*args, **kwargs)
        print("Function returned:", result)
        return result
    return wrapper

@log
def add(a, b):
    return a + b

add(1, 2)

输出：

Calling function: add
Function returned: 3

在这个示例中，log 装饰器函数被用来向 add 函数添加日志记录功能。当 add 函数被调用时，log 装饰器函数会首先被调用，它会打印出正在调用的函数的名称。然后，log 装饰器函数会调用 add 函数，并将 add 函数的返回值打印出来。最后，log 装饰器函数会返回 add 函数的返回值。