Python 装饰器

装饰器 decorator：可动态为函数添加新的功能

使用方法：

import time
def performance(f): #定义装饰器函数，并传入需要被包装的函数
def fn(*args, **kw): #对传进来的函数进行包装的函数，此参数自适应任何参数定义的函数
t_start = time.time() #记录函数开始时间
r = f(*args, **kw) #调用函数
t_end = time.time() #记录函数结束时间
print 'call %s() in %fs' % (f.__name__, (t_end - t_start)) #打印调用函数的属性信息，并打印调用函数所用的时间
return r #返回包装后的函数
return fn #调用包装后的函数

@performance #调用装饰器
def factorial(n): #定义乘数函数（需要包装的函数）
return reduce(lambda x, y: x * y, range(1, n + 1))
print factorial(10)

上述装饰器属于无参装饰器，含参装饰器参考http://www.imooc.com/code/6066
带参数的log函数首先返回一个decorator函数，再让这个decorator函数接收原函数并返回新函数

import time

def performance(unit):
def p_decorator(f):
def fn(*args, **kw):
t_start = time.time()
r = f(*args, **kw)
t_end = time.time()
print 'call %s() in %f%s' % (f.__name__, (t_end - t_start)*1000,unit)
return r
return fn
return p_decorator

@performance('ms')
def factorial(n):
return reduce(lambda x,y: x*y, range(1, n+1))

print factorial(10)

有时候我们需要将原函数的一些属性复制到新函数中

Python内置的functools可以用来自动化完成这个“复制”的任务
import time, functools

def performance(unit):
def p_decorator(f):
@functools.wraps(f)
def fn(*args, **kw):
t_start = time.time()
r = f(*args, **kw)
t_end = time.time()
print 'call %s() in %f%s' % (f.__name__, (t_end - t_start)*1000,unit)
return r
return fn
return p_decorator

@performance('ms')
def factorial(n):
return reduce(lambda x,y: x*y, range(1, n+1))

print factorial.__name__