回顾
上一节我们进行了Python简单装饰器的讲解,但是python的装饰器还有一部分高级的使用方式,这一节就针对python装饰器高级部分进行讲解。
为一个函数添加多个装饰器
今天,老板又交给你一个任务,老板说:"之前的装饰器用用起来挺好,但是这次我需要再次添加更多的参数,这个任务就交给你了,好好干。",这种时候,我们就可以针对一个函数使用多个装饰器。
import datetime def hel(func): def inner(*args,**kwargs): r = func(*args,**kwargs) print('bye') return r return inner def hel1(func): def inner(*args,**kwargs): print('start time:%s' % datetime.datetime.now()) r = func(*args,**kwargs) print('end time:%s'%datetime.datetime.now()) return r return inner @hel1 @hel def f1(name): print('hello') print(name) name = 'alexsel' f1(name) 执行结果: start time:2018-07-01 10:36:02.085027 hello alexsel bye end time:2018-07-01 10:36:02.086027
这次我们在对之前代码添加数据的时候,我们写了第二个装饰器,我们使用这个装饰器针对之前的函数再次增加新的功能,输出了开始结束时间。
代码及输出问题解析
针对刚才多个装饰器,我们需要了解的是在进行输出的时候是怎么进行输出的,首先装饰器函数hel首先对f1函数进行装饰,然后装饰器函数hel1在对被hel装饰过函数进行装饰,最后执行两次装饰过的函数。在新添加的装饰器中(hel1)的func所指的函数是已经被hel装饰过的函数。
带参数的装饰器
装饰器在调用的时候可以传入相应的参数,传入参数以后我们可以根据不同的参数,针对函数进行相应信息输出。
def use_log(level): def hel(func): def inner(*args,**kwargs): if level == 'warning': print('The current function:%s'%func.__name__) r = func(*args,**kwargs) return r return inner return hel @use_log(level = 'warning') def f1(name): print('hello') print(name) f1(name = 'alexsel') 输出结果: The current function:f1 hello alexsel
上面的代码就是可以带有参数的装饰器,就是将以前我们使用简单的装饰器进行了一次封装,当我们调用函数的时候,python可以将我们定义的装饰器的参数传入函数中。
基于类的装饰器
讲解python的类装饰器需要用到python类的知识,python的类我们还没有进行学习,就先简单了解一下。
根据我们刚才对装饰器的学习,我们发现装饰器如果需要正常运行,装饰器所接受的对象必须是可调用的,然后在装饰器内部在返回一个可调用的对象,一般情况下,可调用的对象都是函数,特殊情况是对象中写入了__call__()方法就也是可以调用的对象。
class Test(): def __call__(self): print('test') te = Test() te() 输出结果: test
我们是有类装饰器的时候,可以让类的构造函数__init__()接受一个函数,然后重载一个__call__()并且返回一个函数,来达到装饰器的目的。
class Test(): def __init__(self,func): self.func = func def __call__(self,*args,**kwargs): print('The current function:%s' % self.func.__name__) return self.func() @Test def tes(): print('obj') tes() 输出结果: The current function:tes tes
装饰器就是对原来函数、对象功能的扩展,相当于对函数重新封装,重点需要理解的是装饰器中返回的是函数,而且装饰器所装饰器的对象也需要是可调用的。今天装饰器的高级部分就到这里。