python元编程--装饰器函数

装饰器函数可以对一个函数做包装，使其拥有额外的能力，比如记录一个函数的执行时间。实际工作中不可能在每一个函数中去计算执行时间，因为这样的功能也许仅仅是在测试的时候需要。编写一个专门用于计算函数执行时间的装饰器函数，在需要的时候使用。

#coding=utf-8
'''
Created on 2016-3-9

@author: kwsy
'''
import time
from functools import wraps

def timefunc(func):
@wraps(func)
def wrapper(*args,**kwargs):
start = time.time()
result = func(*args,**kwargs)
end = time.time()
print func.__name__,end-start
return result

return wrapper

@timefunc
def test(n):
time.sleep(n)

if __name__ == '__main__':
test(2)

timefunc 函数接受 test(n) 做为参数，在timefunc内部，调用了test函数，为了保证接受任意参数，使用了*args和**kwargs作为参数。

在实际使用中，还可以这样

def test(n):
time.sleep(n)

if __name__ == '__main__':
func = timefunc(test)
func(2)