Python闭包函数和装饰器练习

引言

    装饰器的大致思想是,闭包函数引用了外部函数的参数(在装饰器里是个函数),因此利用闭包函数除了调用外部函数这个参数函数以外,包装一些其他的功能,试图扩展这个原始函数

参考

    具体的装饰器的理论不做过多的解释,网上参考资料很多,例如下面这篇。点击这篇博客查看

代码解读

    我简单实现了一个计算函数运行时间的装饰器,利用这段代码解读一下装饰器的定义以及使用方法。
    timerFunc是外部函数,这个外部函数的参数是一个函数名,即我们要扩展(包装)的函数,例如下面的addMy函数或者multiMy函数。而我们想要对这些基本函数扩展计算运行时间的功能。因此在timerFunc内定义一个闭包函数wrapper,这个函数里调用addMy并将执行结果返回,然后添加上我们想要的计算运行时间的部分。而timerFunc则返回wrapper函数,即扩展后的函数。因此,timerFunc的返回值实则是一个函数句柄。

#计算函数运行时间的装饰器
def timerFunc(f):
    def wrapper(*args, **kwargs):
        start = time.time()
        res = f(*args, **kwargs)
        end = time.time()
        print('Lapse: ',end-start)
        return res
    return wrapper

#加法函数
def addMy(a,b):
    return a+b

#乘法函数
@timerFunc      #语法糖
def multiMy(a,b):
    return a*b

    具体的使用如下。在使用装饰器的时候,一般都是像上面乘法函数一样,在要扩展的函数定义之前加上语法糖。那么原函数名则变成了wrapper函数的句柄。通过调用multiMy则会增加wrapper函数中附加的功能。

#使用闭包函数的方式调用
>>> a = timerFunc(addMy)
>>> a = timerFunc(addMy)
>>> a(1,2)
Lapse:  4.291534423828125e-06
3
#使用语法糖的方式
>>> multiMy(2,3)
Lapse:  4.0531158447265625e-06
6

你可能感兴趣的:(Python)