python Decorator装饰器用法一

python装饰器有两种类型，一种是函数装饰器，另外一种是类装饰器。函数装饰器基本用法如下：

def decorator(f):
    print("Enter decorator")
    return f;

@decorator
def Function(arg):
    print "Enter Function,arg=",arg

Function(1)
Function(100)

上面的代码首先定义了一个装饰器decorator，这个装饰器的本质其实也是个函数，只是它的返回值为函数对象，然后定义一个普通函数Function，该函数使用了装饰器decorator,使用方法就是在函数定义正上方加上@装饰器的语句。

上面的代码返回结果如下：

Enter decorator
Enter Function,arg= 1
Enter Function,arg= 100

从返回结果来看，decorator只调用了一次，再调用Function(100)时，很显然没有调用到decorator，也就没有起到装饰器的作用。

其实最经典的用法就是在decorator中使用嵌套函数，在嵌套函数中进行装饰的工作，代码如下：

def decorator(f):
    print("Enter decorator")
    def wrapper(arg):
        print("Enter wrapper!")
        return f(arg)
    return wrapper;

@decorator
def Function(arg):
    print "Enter Function,arg=",arg
Function(1)
Function(100)

代码返回结果如下：

Enter decorator
Enter wrapper!
Enter Function,arg= 1
Enter wrapper!
Enter Function,arg= 100

从返回结果可以看出内嵌函数wrapper调用了两次，每次调用函数Function都会调用wrapper函数，起到了装饰器的作用。

初次接触到@语句的同学可以会觉得这个语法比较陌生，甚至有点怪异，其实@decorator 等价于Function = decorator(Function),所以decorator函数本身只会调用一次。decorator函数的作用就是返回一个新函数，这个新函数会先做一些额外的工作，然后再调用原函数。例如上面代码中，decorator返回了新函数wrapper，wrapper先做了自己的处理，然后调用原函数Function。

def decorator(f):
    print("Enter decorator")
    def wrapper(arg):
        print("Enter wrapper!")
        return f(arg)
    return wrapper;

def Function(arg):
    print "Enter Function,arg=",arg
Function = decorator(Function)

Function(1)
Function(100)

上面代码返回结果和前面的返回结果一致。

函数装饰器也可以装饰类的成员函数，装饰器返回新的函数的参数个数应该和原有函数的参数个数一致。

def decorator(f):
    print("Enter decorator")
    def wrapper(*arg):
        print("Enter wrapper!")
        return f(*arg)
    return wrapper;

class A:
    @decorator
    def method(self,arg1,arg2):
        print "method,arg1=",arg1,",arg2=",arg2

a = A()
a.method(1,2)
a.method("Hello","World")

类A的成员函数method有三个参数，所以decorator中的wrapper函数也应该有三个参数，为了通用简单起见，可以用*args表示多个参数，*args可以匹配任意个参数。

代码返回结果如下：

Enter decorator
Enter wrapper!
method,arg1= 1 ,arg2= 2
Enter wrapper!
method,arg1= Hello ,arg2= World