装饰器3_类装饰器

2021-01-28  本文已影响0人  闲云野鹤_23dd

类装饰器

前面我们是用闭包函数来实现的装饰器,那么接下来给大家扩展一下,使用类来当做一个装饰器来用如果要把类当做一个装饰器来用,有两步操作,
首先在init方法中,把被装饰的函数赋值给一个实例属性,
然后再了类里面实现一个ca11方法,在call方法中调用原来的函数。

class Test(object):
    def __init__(self, func):
        self.__func = func

    def __call__(self, *args, **kwargs):
        print('这个是类装饰器')
        print(args[0])
        print(args[1])
        self.__func(*args, **kwargs)

@Test
def log1(a,b):
    print("你好")

log1(44,55)` 

说明

  1. 当用Test来装作装饰器对func函数进行装饰的时候,首先会创建Test的实例对象,并且会把func这个函数名当做参数传递到init方法中,即在init方法中的func变量指向了func函数体
  2. func函数相当于指向了用Test创建出来的实例对象
  3. 当在使用func()进行调用时,就相当于让这个对象(),因此会调用这个对象的ca11方法
  4. 为了能够在ca11方法中调用原来func指向的函数体,所以在init方法中就需要一个实例属性来保存这个函数体的引用,所以才有了self._func=func这句代码,从而在调用ca11方法中能够调用到func之前的函数

类装饰器带参数

class Test2:
    def __init__(self, a, b, c):
        print('---装饰器初始化---')
        self.a = a
        self.b = b
        self.c = c

    def __call__(self, fun):
        print('装饰器参数:',self.a,self.b,self.c)
        def info(*args,**keywords):
            fun(*args,**keywords)
        return info

@Test2("参数a", "参数b", "参数c")
def show(name, age, sex):
    print('性别为' + sex + "的" + age + "岁的" + name)

show("小明", "20", "男")

## 输出
#---装饰器初始化---
#装饰器参数: 参数a 参数b 参数c
#函数的参数: ('小明', '20', '男')
#性别为男的20岁的小明` 

说明:
在使用类装饰器加参数之后,会先调用init方法初始化对象,然后再调用call方法传入函数, 由于函数也有参数,可再写一个闭包函数用于传入参数

上一篇下一篇

猜你喜欢

热点阅读