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