装饰器其实是一次函数调用,以被装饰对象(只能是函数、类方法)为参数,覆盖被装饰的对象。
先来看几个例子:
@decorator
def sum(a, b):
return a + b
=>
sum = decorator(sum)
def decorator(funt):
print "calling …\n"
return funt
@decorator
class A:
pass
=>
A = decorator(A
装饰器的问题在于它返回的对象和原对象本质上是不一样的。但是在很多时候我们需要这两个对象看起来一样,主要有:
from decorator import decorator
@decorator
def mytrace(func, *args, **kw):
print "mytrace starts....\n"
ret = func(*args, **kw)
print "mytrace ends....\n"
return ret
@mytrace
def nsum(a, b):
return a + b
nsum(3, 4)
模块decorator会处理上面列出来的亮点,从而避免很多不必要的麻烦。
来个带参数的:
def defaultDec(msg="abc"):
@decorator
def _defaultDec(f, *args, **kw):
print "start %s\n" % msg
ret = f(*args, **kw)
print "end %s\n" % msg
return ret
return _defaultDec
@defaultDec(msg="this is a msg")
def nsum2(a, b):
return a + b
下面介绍常用的装饰器: