Python中decorator使用实例

在我以前介绍 Python 2.4 特性的Blog中已经介绍过了decorator了，不过，那时是照猫画虎，现在再仔细描述一下它的使用。

关于decorator的详细介绍在 Python 2.4中的What's new中已经有介绍，大家可以看一下。

如何调用decorator

基本上调用decorator有两种形式

第一种：
代码如下:
@A
def f ():

这种形式是decorator不带参数的写法。最终 Python 会处理为：
代码如下:
f = A(f)

还可以扩展成：
代码如下:
@A
@B
@C
def f ():
   

最终 Python 会处理为：
代码如下:
f = A(B(C(f)))

注：文档上写的是@A @B @C的形式，但实际上是不行的，要写成多行。而且执行顺序是按函数调用顺序来的，先最下面的C，然后是B，然后是A。因此，如果decorator有顺序话，一定要注意：先要执行的放在最下面，最后执行的放在最上面。(应该不存在这种倒序的关系)

第二种：
代码如下:
@A(args)
def f ():
   

这种形式是decorator带参数的写法。那么 Python 会处理为:
代码如下:
def f():
_deco = A(args)
f = _deco(f)

可以看出， Python 会先执行A(args)得到一个decorator函数，然后再按与第一种一样的方式进行处理。

decorator函数的定义

每一个decorator都对应有相应的函数，它要对后面的函数进行处理，要么返回原来的函数对象，要么返回一个新的函数对象。请注意，decorator只用来处理函数和类方法。

第一种：
针对于第一种调用形式
代码如下:
def A(func):
    #处理func
    #如func.attr='decorated'
    return func
@A
def f(args):pass

上面是对func处理后，仍返回原函数对象。这个decorator函数的参数为要处理的函数。如果要返回一个新的函数，可以为：
代码如下:
def A(func):
    def new_func(args):
        #做一些额外的工作
        return func(args) #调用原函数继续进行处理
    return new_func
@A
def f(args):pass

要注意 new_func的定义形式要与待处理的函数相同，因此还可以写得通用一些，如：

代码如下:
def A(func):
    def new_func(*args, **argkw):
        #做一些额外的工作
        return func(*args, **argkw) #调用原函数继续进行处理
    return new_func
@A
def f(args):pass

可以看出，在A中定义了新的函数，然后A返回这个新的函数。在新函数中，先处理一些事情，比如对参数进行检查，或做一些其它的工作，然后再调原始的函数进行处理。这种模式可以看成，在调用函数前，通过使用decorator技术，可以在调用函数之前进行了一些处理。如果你想在调用函数之后进行一些处理，或者再进一步，在调用函数之后，根据函数的返回值进行一些处理可以写成这样：