Python 中带参数的装饰器如何实现？

推荐答案

// www.javascriptcn.com code example
def decorator_with_args(arg1, arg2):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"Decorator arguments: {arg1}, {arg2}")
            result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

@decorator_with_args("hello", 42)
def my_function():
    print("Function is called")

my_function()

本题详细解读

1. 什么是带参数的装饰器？

带参数的装饰器是指装饰器本身可以接受参数，并在装饰函数时使用这些参数。这种装饰器通常用于在装饰器中传递配置或控制装饰行为。

2. 如何实现带参数的装饰器？

带参数的装饰器通常通过三层嵌套函数来实现：

1. 外层函数：接受装饰器的参数，并返回一个装饰器函数。
2. 中层函数：接受被装饰的函数，并返回一个包装函数。
3. 内层函数：实际的包装函数，负责执行装饰逻辑并调用原始函数。

3. 代码解析

• decorator_with_args(arg1, arg2)：这是外层函数，接受装饰器的参数 arg1 和 arg2，并返回一个装饰器函数 decorator。

• decorator(func)：这是中层函数，接受被装饰的函数 func，并返回一个包装函数 wrapper。

• wrapper(*args, **kwargs)：这是内层函数，负责实际的装饰逻辑。在这个例子中，它打印了装饰器的参数，并调用了原始函数 func。

• @decorator_with_args("hello", 42)：这是装饰器的使用方式，传递参数 "hello" 和 42 给装饰器。

• my_function()：调用被装饰的函数，输出如下：

  Decorator arguments: hello, 42
  Function is called

4. 总结

带参数的装饰器通过三层嵌套函数实现，外层函数接受装饰器的参数，中层函数接受被装饰的函数，内层函数负责实际的装饰逻辑。这种结构使得装饰器可以灵活地接受参数并应用于不同的函数。