Python 装饰器

在不改变函数的情况下，在代码运行期间动态增加函数功能的方式，就称之为“装饰器”，装饰器其实就是一个返回函数的高价函数。比如我们定义一个能打印日志的装饰器，可以定义如下：

def log(method):
    def wrapper(*args, **kw):
        print 'call %s()' % method.__name__
        return method(*args, **kw)
    return wrapper

怎么使用装饰器呢，Python提供了@语法，例如：

@log
def test():
    print 'test decorate'

调用test函数，得到的结果如下：

>>>test()
call test()
test decorate

test作为产生传入log，相当于执行了语句：

test = log(test)

我们来分析一下，log返回的是一个函数，所以最先执行的是log返回的wrapper，wrapper函数可以接受任意参数的调用，在wrapper函数内，首先打印日志，再接着调用传入进来的原始函数。
如果装饰器需要传入参数怎么办呢，只要多包裹一层，比如：log装饰器中传入一个text:

def log(text):
    def decorator(method):
        def wrapper(*args, **kkw):
            print '%s %s()' % (text, method.__name__)
            return wrapper
    return decorator

使用很简单，方法如下：

@log('test decorator')
def test():
    print 'test decorator'

执行结果如下：

test decorator test()

上面相当于执行了如下语句：

test = log('test decorator')(test)

分析一下，首先执行的是log(‘test decorator’)，返回的是decorator函数，再调用decorator，test作为参数，最终返回的是wrapper函数，不过，上面会存在一个问题，函数也是对象，它们有name属性，但是经过装饰器之后，最后name变成了wrapper，有些函数是依赖签名的，有可能导致出错，怎么办呢，Python提供了functools.wraps来处理这个问题，把name给纠正过来。示例如下：

from functools import wraps


def log(method):

    @wraps(method)
    def wrapper(*args, **kw):
        print 'call %s()' % method.__name__
        return method(*args, **kw)
    return wrapper

总结

在Java这种面向对象语言中，有一种设计模式也叫装饰器，是对类的一种包裹，提供更多的功能，但最终的目的都是提供更多的功能，Python中也可以解决代码重复的问题，方式也很优雅，方便，简单