python中修饰器_python 类修饰器

1. 修改类函数。

场景： 如果要给一个类的所有方法加上计时，并打印出来。demo如下：

# -*- coding:utf-8 -*-

import time

def time_it(fn):

"Example of a method decorator"

def decorator(*args, **kwargs):

t1=time.time()

ret = fn(*args, **kwargs)

print '\t\t%d seconds taken for %s'%(time.time()-t1, fn.__name__)

return ret

return decorator

def class_decorator(*method_names):

def class_rebuilder(cls):

"The class decorator example"

class NewClass(cls):

"This is the overwritten class"

def __getattribute__(self, attr_name):

attr_val = super(NewClass, self).__getattribute__(attr_name)

if callable(attr_val) and attr_name in method_names:

return time_it(attr_val)

return attr_val

return NewClass

return class_rebuilder

@class