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

叁歲伎倆 2024-03-23 23:05 45阅读 0赞

装饰器是 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")

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