使用类来写装饰器

前两天发现了装饰器可以直接使用类来写，分享一下，需要用类里面的__call__方法，__call__方法就是可以把这个实例当成一个函数来调用，如果正常使用类里面的方法的话，实例方法要先实例化类，然后才能调用，静态方法、类方法则需要用类名或者实例化之后的对象来调用，而实例化之后的这个对象，是不能被调用的，__call__方法就是把这个实例变成一个可以调用的对象，也就是说实例化之后，这个对象就可以和一个普通函数一样被调用。

示例代码如下：

 

1 2 3 4 5 6 7 8 9

class Foo:     def __call__(self, *args, **kwargs):         print('call....')     def test(self):#         print('test....') if __name__ == '__main__':     t = Foo()#实例化类     t.test()#正常调用实例方法     t()#直接调用实例化之后的对象

运行结果：

 

1 2	>>>test.... #这个是调用test方法的时候输出的 >>>call....#这个是执行调用这个实例化之后的方法输出的

理解了上面的之后，就可以使用class来写一个装饰器了，计算程序的运行时间，当然思想和以前用函数写装饰器是一样的

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

class Fuck(object):     def __init__(self, func):         self.func = func     def __call__(self, *args, **kwargs):         import time         start_time = time.time()         res = self.func(*args, **kwargs)         end_time = time.time()         print('the function "%s" run time is %s' % (self.func.__name__,                                                     (end_time - start_time)))         return res @Fuck def run(name):     import time     time.sleep(1)     return 'sb_%s' % name print(run('hyf'))

运行结果：

 

1 2	>>>the function "run" run time is 1.0001001358032227#这个是装饰器帮我们计算的函数运行时间 >>>sb_hyf#这个是正常运行run函数的时候，返回的值

转载于:https://www.cnblogs.com/sriba/p/8043623.html