python装饰器系列(四)

带参数的装饰器

先来看一个不带参数的装饰器

import time

def timeit(fn):
    def wrap(*args,**kwargs):
        start = time.time()
        ret = fn(*args,**kwargs)
        print(time.time() - start)
        return ret
    return wrap
 

@timeit
def sleep(x):
    time.sleep(x)

sleep(3)
3.0034420490264893

这里打印出来的是执行sleep函数所消耗的自然时间,但在执行此函数时所消耗的cpu时间真的有3.0034420490264893秒吗？当然不是。利用time包中的time.clock方法可以计算得到代码执行所消耗cpu的时间，那怎样来修改上边的timeit函数，让其即能计算代码执行的自然时间，也能计算代码执行所消耗cpu的时间？做如下改进：

def timeit_1(process_time=False):
    cacl = time.clock if process_time else time.time
    def timeit_2(fn):
        def wrap(*args,**kwargs):
            start = cacl()
            ret = fn(*args,**kwargs)
            print(cacl() - start)
            return ret
        return wrap
    return timeit_2

 
def sleep_1(x):
    time.sleep(x)

timeit_1(True)(sleep_1)(3)
0.020000000000000018

timeit_1(False)(sleep_1)(3)
3.0038363933563232

 # 参数process_time是一个默认参数，所以可以不传递值，默认为False    
timeit_1()(sleep_1)(3)  
3.003509283065796

调用过程分解：

fn1 = timeit_1(True)

调用timeit_1(True)，函数return回了timeit_2，并把fn1这个变量指向了调用结果，即指向了timeit_2，这里的timeit_2也是一个函数，此函数接收一个参数

fn2 = fn1(sleep_1)

调用fn1(sleep_1)，其实就是调用了timeit_2(sleep_1)，并把fn2这个变量指向了调用后的结果，即指向了warp，这里的warp也是一个函数，此函数能接收任意的参数

fn2(3)
0.009999999999999787

调用fn2(3)，其实是调用了wrap(3),即执行了wrap函数内的语句，此函数内的ret = fn(*args,**kwargs)语句中的fn其实是指向了sleep，所以在执行wrap函数时，sleep_1函数才真正被执行

改进的装饰器装饰一个函数：

@timeit_1(False)
def sleep_2(x):
    time.sleep(x)

sleep_2(3)
3.0039477348327637

计算代码执行的cpu时间

@timeit_1(True)
def sleep_3(x):
    time.sleep(x)

sleep_3(3)
0.0

魔法背后的原理：

其实质就是在没有用魔法的情况下直接timeit_1(True)(sleep_3)(3)。而当使用@这个魔法后，当代码执行到此行时，解析器会执行timeit_1(True),timeit_1实质就是一函数，接收一个参数，并返回一个timeit_2函数。当代码执行到@所在语句时，会把所装饰的sleep_3函数作为一个参数传递给timeit_1(True)的调用结果，即timeit_2这个函数，即sleep_3这个函数已作为一个变量传递给了timeit_2(fn)中的fn参数，并返回了一个wrap函数，在接下的调用sleep_3(3)这个操作，其实此时的sleep_3这个函数已不是原先的def sleep_3(x):中的sleep_3函数，而是一个指向了wrap的函数，wrap函数接收任何参数，所以把当执行sleep_3(3)时，把参数3传递给了wrap函数，并执行内部的代码，内部代码中ret = fn(*args,**kwargs)中的fn函数依赖还是指向原先的sleep_3(x)这个函数。

这里也有一个简单的记忆方式，如果一个函数被装饰器所装饰，在调用这个函数时其实不再是调用表面上看上去的这个函数，以来做说明

@timeit_1(True)
def sleep_3(x):
    time.sleep(x)

当执行到有@魔法所在行时，相当于执行了sleep_3 = timeit_1(True)(sleep_3)，即指向了wrap函数，既然sleep_3指向了wrap函数，那我们执行sleep_3(3)时，其实就是在进行wrap(3)这样的函数调用，记住，函数名也是一个变量