python实现decorator模式

ty9919

python有个很常用的语法糖是@decorator，使用它可以很方便的创建decorator装饰器模式。（当然，@的用处可不只用在创建装饰器模式）方法有两种，一种是通过创建一个包裹类Wrapper，另一种就是直接通过一个函数创建closure
简要说下装饰器模式：不改变目标的内部行为，改变目标的外在表现方式的一种模式。重点在于，它一定不干涉“内政”，只可以在目标的外围进行修饰，对目标是透明的。
函数的方法最简单
def wrap(num):
print("wrap initialed")
def w(func):
print("w start", num)
def w2(*args):
print("w2 start")
func(*args)
print("w2 end")
return "some thing"      
return w2
return w
@wrap(10)
def foo2(num, string):
print(string * num)
print("BEING:")
foo2(3, 'z')
foo2(4, 'y')
print(foo2)
输出结果是：
wrap initialed
('w start', 10)
BEING:
w2 start
zzz
w2 end
w2 start
yyyy
w2 end
<function w2 at 0x7f25d74735f0>
这里就相当与执行了语句
wrap(10)(foo2)(3, 'z')
每一层都先计算出本层的对象，然后再调用被嵌套的函数
这里可以看到
wrap initialed
('w start', 10)
最先输出了一次，说明虽然是包裹函数，但是其已经形成了闭包，全局唯一
如果使用类来实现，情况也比较类似
import datetime
class Wrapper(object):
def __init__(self, num = -1):
print("Wrapper Initialed", num)
self.num = num

def __call__(self, func):
print("Wrapper Called")
# 以下表明参数的写法虽运行无误，但是是没有意义的
def w(fff, ggg, *args):
print("Wrapper Preparing", datetime.datetime.now())
# 这样传参数是没有意义的
func(num = ggg, x = self.num,  *args)
print("Wrapper Finished", datetime.datetime.now())
#        w.__name__ = func.__name__
return w
#将值3传入Wrapper的self.num
@Wrapper(3)
def foo(num = 1, x = 'z', *args):
print(x * num, args, 'x=', x, 'num=', num)

if __name__ == '__main__':
print('BEGIN:')
foo(2, 'a')
foo(3, 'b')
print(foo)
输出结果是：
('Wrapper Initialed', 3)
Wrapper Called
BEGIN:
('Wrapper Preparing', datetime.datetime(2012, 5, 5, 0, 25, 58, 136713))
('aaa', (), 'x=', 3, 'num=', 'a')
('Wrapper Finished', datetime.datetime(2012, 5, 5, 0, 25, 58, 136829))
('Wrapper Preparing', datetime.datetime(2012, 5, 5, 0, 25, 58, 136864))
('bbb', (), 'x=', 3, 'num=', 'b')
('Wrapper Finished', datetime.datetime(2012, 5, 5, 0, 25, 58, 136920))
<function w at 0x7f11d4f21758>
可以看出Wrapper类也只执行了一次初始化。我怀疑可不可能是因为
foo = Wrapper(3)(foo)(*args)返回了一个闭包。通过最后一行的打印结果证实了我的想法。这个类也形成了一个闭包。
值得注意的有两点：一是Wrapper类使用了__call__函数，这个函数仅在发生调用时instance(*args)，才会被激活，相当与instance.__call__(*args)
二是不要瞎折腾。 我在这里尝试把Wrapper的参数，传递给foo，这里需要w函数和foo的参数列表保持一致就可以不报错。但是这样做起不到改变foo函数行为的作用，顶多改变这个foo函数的默认参数，所以是毫无意义的！如果没有特殊必要，请不要把Wrapper的参数往foo中塞，Wrapper的函数仅在foo调用前后调用才有意义。
最后一点写参数表的心得体会是，在写装饰器的时候，如果拿不准func参数怎么写，怎么把它包装起来，那么就先写func(*args)，然后再完善上面一层封包，直至最后把w(func)函数参数写好。