python带参数装饰器的两种写法

(相关资料图)

装饰器是 Python 中非常有用的语法特性，可以用于包装或者修改函数的行为。有时候我们希望给装饰器添加参数，以便于在装饰器内部使用，那么这时候就需要使用带参数的装饰器。常用的两种带参数装饰器的写法如下：

1. 第一种装饰器带参数的写法：

在装饰器函数外层再套一个函数，用来接收和处理装饰器的参数。

def decorator_with_args(arg1, arg2, arg3):       def wrapper(func):           def inner_wrapper(*args, **kwargs):               print("Decorator arguments:", arg1, arg2, arg3)               return func(*args, **kwargs)           return inner_wrapper       return wrapper      @decorator_with_args("Hello", "World", 42)   def my_function(arg1, arg2):       print("Function arguments:", arg1, arg2)          my_function("Hi", "Tom")

2. 第二种装饰器带参数的写法：

使用一个装饰器函数来实现装饰器的参数传递，然后再返回一个函数来完成真正的装饰。

def decorator_with_args(arg1, arg2, arg3):       def real_decorator(func):           def wrapper(*args, **kwargs):               print("Decorator arguments:", arg1, arg2, arg3)               return func(*args, **kwargs)           return wrapper       return real_decorator      @decorator_with_args("Hello", "World", 42)   def my_function(arg1, arg2):       print("Function arguments:", arg1, arg2)          my_function("Hi", "Tom")

以上两种写法都可以实现带参数的装饰器。第一种写法中，外层套一个函数来处理装饰器参数，然后再返回内层装饰器函数来实现真正的装饰器；第二种写法中，使用一个装饰器函数来接收和处理参数，然后再返回一个内部函数来完成真正的装饰。