Python3之装饰器

  参考:https://www.cnblogs.com/songyue/p/5196809.html

  例如以前写了一个简单的函数sum1如下

def sum1():
    sum=1+2
    print(sum)


sum1()
print('函数名为',sum1.__name__)                    

  运行结果为

3
函数名为 sum1

  假如需要新加一个功能查看该函数运行执行了多长时间,可以导入时间模块记录开始时间及结束时间,用结束时间减开始时间及可得到运行时间,修改函数如下

import time 
import functools

def sum1():
    start=time.time()
    sum=1+2
    print(sum)
    end=time.time()
    print("Time used:",end-start)

sum1()
print('函数名为',sum1.__name__)

  运行结果

3
Time used: 2.6464462280273438e-05
函数名为 sum1

  可是还是许多函数需要增加运行时间的功能,难道一个个函数去改嘛,可以定义一个时间函数timeit将sum1作为参数传递给时间函数timeit在timeit里面调用sum1

import time
import functools

def sum1():
    sum=1+2
    print(sum)

def timeit(func):
    start=time.time()
    func()
    end=time.time()
    print('Time used:',end-start)
    
timeit(sum1)

  运行结果

3
Time used: 2.574920654296875e-05

  可以运行,但是还是修改了一部分代码,把sum1()改成了timeit(sum1),直接运行sum1()还是不会打印运行时间,如果sum在N处被调用了,不得不去修改N处代码。所以需要将sum1()具有和timeit(sum1)一样的效果,于是将timeit赋值给sum1,可是timeit是有参数的,所以需要找个方法去统一参数,将timeit(sum1)发返回值(计算运行时间的函数)赋值给sum1

import time
import functools

def sum1():
    sum=1+2
    print(sum)

def timeit(func):
    def test():
        start=time.time()
        func()
        end=time.time()
        print('Time used:',end-start)
    return test

sum1=timeit(sum1)
#sum1为函数名需要加括号执行执行
sum1()
print('函数名为',sum1.__name__)

  执行结果

3
Time used: 3.0279159545898438e-05
函数名为 test

  一个装饰器就做好了,我们只需要在定义sum1之后调用sum1之前,加上sum1=timeit(sum1),就可以达到计时的目的,这就是装饰器的概念,看起来像是sum1倍timeit装饰了,Python提供一个语法糖来降低字符输入量

import time
import functools


def timeit(func):
    def test():
        start=time.time()
        func()
        end=time.time()
        print('Time used:',end-start)
    return test

@timeit
def sum1():
    sum=1+2
    print(sum)


#sum1为函数名需要加括号执行执行
sum1()
print('函数名为',sum1.__name__)

  PS:这里的@timeit等价于sum1=timeit(sum1)

      执行结果

3
Time used: 2.8848648071289062e-05
函数名为 test

  我们可以看到执行的函数名为test,因为返回的是test函数,需要把原始函数sum1的__name__属性复制到test()函数中,否则,有些依赖函数前面的代码执行就会出错,不需要编写test.__name__=func.__name__这样的代码,python内置的functools.wraps就是干这个事的,所以,一个完整的装饰器写法如下

import time
import functools


def timeit(func):
    @functools.wraps(func)
    def test():
        start=time.time()
        func()
        end=time.time()
        print('Time used:',end-start)
    return test

@timeit
def sum1():
    sum=1+2
    print(sum)


#sum1为函数名需要加括号执行执行
sum1()
print('函数名为',sum1.__name__)

  执行结果

3
Time used: 3.647804260253906e-05
函数名为 sum1

  函数名为原始函数名sum1

上一篇:python – 将函数中的局部变量导入timeit


下一篇:python:面向字节码的探查器