Python装饰器

Posted 2020-11-30 八叶一刀·无仞剑

本文来介绍一下Python的函数装饰器，类装饰器以后另开一文讲。

装饰器可以看做是Python中的一个语法糖，基本的写法示例如下：

def decorator(func):
    return func

@decorator
def function():
    print "666"

decorator即是function的装饰器，在function函数前面加上@decorator的前缀相当于：

function = decorator(function)

所以之后对function的调用实际上都是在调用decorator。

以上只是最最简单的写法，一般装饰器会在函数内部定义一些自己的操作，相当于“装饰”了原本的函数。例如：

import time
def decorator(func):
    def clocked(*args):
        t0 = time.time()
        result = func(*args)
        elapsed = time.time() - t0
        name = func.__name__
        print elapsed, name, result
        return result
    return clocked

@decorator
def function():
    print "666"

function()
print function

输出如下：

666
0.0 function None
<function clocked at 0x03270A30>

可以看到：

1.该装饰器的作用是输出所装饰函数的运行时间，和函数名以及返回值

2.通过输出function，可以发现此时它已经被替换成了clocked，这与前面讲的“function = decorator(function)”是吻合的。

另外关于装饰器要注意的一点是：装饰器是在加载该模块的时候执行的。举个例子，假设有如下代码：

def decorator(func):
    print "7777"
    return func

@decorator
def function():
    print "666"

if __name__ == ‘__main__‘:
    print "start!"
    function()

运行该py文件后输出如下：

7777
start!
666

可以看到，“777”出现在“start!”之前，这说明装饰器是在实际运行该模块之前输出的。

 那么有一个问题，如何给装饰器传入参数呢？这就需要我们创建一个装饰器工厂函数，把参数传给它，返回一个装饰器，然后再把它应用到要装饰的函数上。具体示例如下：