Python 学习日知录（十）- 装饰器初步

Python 学习日知录（十）- 装饰器初步

1. 概念

装饰器（Decorator）是Python的一种语法糖，允许开发者在已实现的函数或类的外面扩充新的功能，新的函数名和原函数一样，因此可以不改变调用该函数的代码。

装饰器是一个函数，以被修饰函数为参数，返回值也是一个函数。

2. 语法

@表示对下面的函数使用装饰器，语法如下：

@Decorator
def func(*args):
    pass

3. 一个简单的例子

# -*- coding:utf8 -*-


def log_cost_time(func): # 定义一个装饰器
    def wrapped(*args, **kwargs): # 装饰器返回的函数
        import time
        begin = time.time()
        try:
            return func(*args, **kwargs) # 执行被装饰的函数
        finally:
            print 'func %s cost %s' % (func.__name__, time.time() - begin) # 装饰器执行的内容
    return wrapped # 返回函数


@log_cost_time # 使用装饰器装饰complex_func函数
def complex_func(num):
    ret = 0
    for i in xrange(num):
        ret += i * i
    return ret


print complex_func(10000000)

上面的例子创建了一个装饰器log_cost_time，用于打印被装饰函数的运行时间。运行结果如下：

func complex_func cost 1.56242513657
333333283333335000000

可见，在正常调用complex_func函数返回结果时，运行时间也被打印出来了。

装饰器还有许多高级的用法，待以后发掘。