python装饰器

当我们要扩展一个函数的功能，却不想修改原函数的时候，可以用装饰器，比如：

有函数function1:

def function1(val):
    val += 1
    return val

当我们想输出这个函数的执行时间时，通常可以考虑修改原函数:

from time import time


def function1(val):
    start = time()
    val += 1
    end = time()
    print("time cost {0}".format(end - start))

但是有时候我们不能修改原函数的代码，比如该函数为内置函数，或者第三方的，而且及时可以修改如果每个函数都需要该功能，就得修改所有的函数，很麻烦，此时，可以用python的装饰器功能，先建立一个需要扩展功能的装饰器：

from time import time


def deco(f):
    def inner(*args,**kwargs):
        start = time()        #需要扩展的功能
        f(*args,**kwargs)    #执行被装饰的函数
        end = time()          #需要扩展的功能
        print("time cost {0}".format(end - start)
    return inner

然后在需要被扩展的函数上加上@deco即可扩展原函数的功能。

@deco
def function1(val):
    val += 1
    return val