装璜器

装璜器(decorator)能够加强函数的性能,定义起来尽管有点简单,但应用起来非常灵活和不便。

因为函数也是一个对象,而且函数对象能够被赋值给变量,所以,通过变量也能调用该函数。

>>> def now():...     print('2015-3-25')...>>> f = now>>> f()2015-3-25

函数对象有一个__name__属性,能够拿到函数的名字:

>>> now.__name__'now'>>> f.__name__'now'

假如咱们要加强 now() 函数的性能,比方,在函数调用前后主动打印日志,但又不心愿批改now()函数的定义,这种在代码运行期间动静减少性能的形式,称为“装璜器”(Decorator)。

import functoolsdef log(func):    # 把原始函数的__name__等属性复制到wrapper()函数中    @functools.wraps(func)    def wrapper(*args, **kw):        print('call %s():' % func.__name__)        return func(*args, **kw)    return wrapper@logdef now():    print('2015-3-25')

执行后果如下:

>>> now()call now():2015-3-25

@log 放到 now() 函数的定义处,相当于执行了语句:

now = log(now)

因为 log() 是一个decorator,返回一个函数,所以,原来的now()函数依然存在,只是当初同名的 now 变量指向了新的函数,于是调用 now() 将执行新函数,即在 log() 函数中返回的 wrapper() 函数。

wrapper() 函数的参数定义是 (*args, **kw),因而,wrapper() 函数能够承受任意参数的调用。

带参数的装璜器

如果decorator自身须要传入参数,那就须要编写一个返回decorator的高阶函数,写进去会更简单。比方,要自定义log的文本:

import functoolsdef log(text):    def decorator(func):        # 把原始函数的__name__等属性复制到wrapper()函数中        @functools.wraps(func)        def wrapper(*args, **kw):            print('%s %s():' % (text, func.__name__))            return func(*args, **kw)        return wrapper    return decorator

这个3层嵌套的decorator用法如下:

@log('execute')def now():    print('2015-3-25')

执行后果如下:

>>> now()execute now():2015-3-25

和两层嵌套的decorator相比,3层嵌套的成果是这样的:

>>> now = log('execute')(now)

咱们来分析下面的语句,首先执行log('execute'),返回的是decorator函数,再调用返回的函数,参数是now函数,返回值最终是wrapper函数。