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

打印 上一主题 下一主题

主题 936|帖子 936|积分 2808

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

在装饰器函数外层再套一个函数,用来接收和处理装饰器的参数。
  1.    def decorator_with_args(arg1, arg2, arg3):
  2.        def wrapper(func):
  3.            def inner_wrapper(*args, **kwargs):
  4.                print("Decorator arguments:", arg1, arg2, arg3)
  5.                return func(*args, **kwargs)
  6.            return inner_wrapper
  7.        return wrapper
  8.    
  9.    @decorator_with_args("Hello", "World", 42)
  10.    def my_function(arg1, arg2):
  11.        print("Function arguments:", arg1, arg2)
  12.       
  13.    my_function("Hi", "Tom")
复制代码
2. 第二种装饰器带参数的写法:

使用一个装饰器函数来实现装饰器的参数传递,然后再返回一个函数来完成真正的装饰。
  1.    def decorator_with_args(arg1, arg2, arg3):
  2.        def real_decorator(func):
  3.            def wrapper(*args, **kwargs):
  4.                print("Decorator arguments:", arg1, arg2, arg3)
  5.                return func(*args, **kwargs)
  6.            return wrapper
  7.        return real_decorator
  8.    
  9.    @decorator_with_args("Hello", "World", 42)
  10.    def my_function(arg1, arg2):
  11.        print("Function arguments:", arg1, arg2)
  12.       
  13.    my_function("Hi", "Tom")
复制代码
以上两种写法都可以实现带参数的装饰器。第一种写法中,外层套一个函数来处理装饰器参数,然后再返回内层装饰器函数来实现真正的装饰器;第二种写法中,使用一个装饰器函数来接收和处理参数,然后再返回一个内部函数来完成真正的装饰。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

我可以不吃啊

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表