装饰器是 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")
以上两种写法都可以实现带参数的装饰器。第一种写法中,外层套一个函数来处理装饰器参数,然后再返回内层装饰器函数来实现真正的装饰器;第二种写法中,使用一个装饰器函数来接收和处理参数,然后再返回一个内部函数来完成真正的装饰。
内容来源于网络如有侵权请私信删除
文章来源: 博客园
- 还没有人评论,欢迎说说您的想法!