Python 带参数的装饰器

在本文中，我们将介绍如何使用带参数的装饰器来增强Python函数的功能。装饰器是Python中强大而灵活的功能，可以在不修改原始函数代码的情况下扩展其行为。通过使用装饰器，我们可以在函数执行之前和之后添加额外的功能，例如日志记录、性能测试和异常处理。

阅读更多：Python 教程

什么是装饰器？

在开始介绍带参数的装饰器之前，让我们先回顾一下什么是装饰器。装饰器本质上是一个函数，它接受一个函数作为输入，并返回一个新的函数作为输出。这个新函数通常会在原始函数的基础上添加一些额外的功能或修改函数的行为。

装饰器可以使用@语法来应用在函数上。例如，下面是一个简单的装饰器的例子：

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print("Before function execution")
        result = func(*args, **kwargs)
        print("After function execution")
        return result
    return wrapper

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

# 调用带有装饰器的函数
my_function()

这个装饰器 my_decorator 接受一个函数作为参数，并返回一个新的函数 wrapper。在这个例子中，wrapper 函数在调用原始函数之前和之后打印了一些额外的信息。通过将装饰器 @my_decorator 应用在函数上，我们实际上将函数 my_function 传递给了装饰器，并且在调用函数之前和之后添加了一些功能。

带参数的装饰器

装饰器不仅可以应用在函数上，还可以应用在类和方法上。在Python中，装饰器本身也可以接受参数，这使得装饰器更加灵活和强大。带参数的装饰器可以用来个性化地定制装饰器的行为，使其适用于不同的函数或不同的场景。

下面是一个带参数的装饰器的例子：

def print_decorator(prefix):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"{prefix}: Before function execution")
            result = func(*args, **kwargs)
            print(f"{prefix}: After function execution")
            return result
        return wrapper
    return decorator

@print_decorator("INFO")
def my_function():
    print("Hello, World!")

@print_decorator("DEBUG")
def another_function():
    print("Another function")

# 调用带有装饰器的函数
my_function()
another_function()

在这个例子中，我们定义了一个带有参数的装饰器 print_decorator。这个装饰器接受一个参数 prefix，用于指定前缀。装饰器内部又定义了一个匿名函数 decorator，这个函数接受一个函数作为参数，并返回一个新的函数 wrapper。

在调用装饰器时，我们将参数传递给了装饰器 @print_decorator("INFO") 和 @print_decorator("DEBUG")。这样，我们可以在调用函数之前和之后打印带有指定前缀的信息。

更复杂的带参数装饰器

有时候，我们可能需要一个更复杂的带参数装饰器，它需要接受多个参数并返回一个装饰器函数。这可以通过嵌套的装饰器来实现。

下面是一个更复杂的带参数装饰器的例子：

def debug_decorator(level):
    def decorator(func):
        indent = level * "  "
        def wrapper(*args, **kwargs):
            print(f"{indent}Before function execution")
            result = func(*args, **kwargs)
            print(f"{indent}After function execution")
            return result
        return wrapper
    return decorator

@debug_decorator(1)
def my_function():
    print("Hello, World!")

@debug_decorator(2)
def another_function():
    print("Another function")

# 调用带有装饰器的函数
my_function()
another_function()