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")

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

posted @ 2023-08-09 15:57  python学习者0  阅读(133)  评论(0编辑  收藏  举报