类装饰器
前面我们是用闭包函数来实现的装饰器,那么接下来给大家扩展一下,使用类来当做一个装饰器来用如果要把类当做一个装饰器来用,有两步操作,
首先在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)`
说明
- 当用Test来装作装饰器对func函数进行装饰的时候,首先会创建Test的实例对象,并且会把func这个函数名当做参数传递到init方法中,即在init方法中的func变量指向了func函数体
- func函数相当于指向了用Test创建出来的实例对象
- 当在使用func()进行调用时,就相当于让这个对象(),因此会调用这个对象的ca11方法
- 为了能够在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方法传入函数, 由于函数也有参数,可再写一个闭包函数用于传入参数
网友评论