Python中decorator的使用（@函数的使用）

使用在Python 2.4 or Later.

两种调用形式：
1.不带参数

@A

def f():

Python最后会处理为：

f = A(f)

，可扩展为：

@A
@B
@C
def f():

最终Python会处理为:

f = A(B(C(f)))

 

2.带参数：

@A(args)
def f():

Python会处理为：

def f();
  _deco = A(args)
  f = _deco(f)

Python先执行A(args)得到_deco，然后跟第一种方式一样的处理函数f。


对应的两种函数定义方法：
第一种：

def A(func):
  #处理func
  #如func.attr='decorated'
  return func

@A
def f(args): pass

A对f处理完后，仍返回函数对象。 如果要返回一个新的函数，可以为：

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):
    result = func(*args, **argkw)
    if result:
      return new_result
    else:
      return result
  return new_func

@A
def f(args): pass

 

第二种： 

def A(arg):
  def _A(func):
    def new_func(args):
      #do any operation
      return func(args)
    return new_func
  return _A

@A(arg)
def f(args): pass

文档上说：如果你的decorator在调用时使用了参数，那么你的decorator函数只会使用这些参数进行调用。因此你需要返回一个新的decorator函数，就和第一种方式定义形式一致了。