Python功能点实现:函数级/代码块级计时器


Posted in Python onJanuary 02, 2019

工程中我们常常需要对某一个函数或者一块代码计时,从而监测系统关键位置的性能。计时方法是在代码块前后分别记录当前系统时间,然后两者相减得到代码块的耗时。最简单原始的实现类似:

from datetime import datetime
start = datetime.now()
# some code you want to measure
end = datetime.now()
print("Processing time for {} is: {} seconds".format('You Name It', elapse))

这种方式缺点明显:假如系统内有很多地方都需要计时,那么每个地方都需要插入这样的计时代码,首先是重复性工作很麻烦,其次这样会降低代码的可读性,干扰对业务逻辑的理解。本文将给出一些更好的实现,主要涉及的技术是装饰器(Decorator)和运行时上下文(runtime context)。

基于装饰器的函数级计时器

第一种计时器是比较常见的函数级计时器,通过装饰器完成,将原函数改装成拥有计时功能的新函数,使其可以完成运行原来函数和计时两件事。在使用时,只用在需要计时功能的函数代码前加上类似@timer的语法糖,这样每次调用原函数时,运行的将会是新函数。这样就大大减少了重复性劳动。

具体实现如下:

from datetime import datetime
def timer(func):
  '''Function Level Timer via Decorator'''
  def timed(*args, **kwargs):
    start = datetime.now()
    result = func(*args, **kwargs)
    end = datetime.now()
    elapse = (end - start).total_seconds()
    print("Processing time for {} is: {} seconds".format(func.__name__, elapse))
    return result
  return timed
@timer
def test_1(a):
  '''Function Level'''
  a *= 2
  return a
if __name__ == '__main__':
  print(test_1(1))

基于上下文的代码块级计时器

装饰器实现的计时器可以为函数添加计时功能,可以满足大部分情况的需要,但是假如我们想要更灵活一些,对任意一段连续的代码块做计时,怎样做?使用原始的插计时代码的方法显然不是我们想要的;也可以将代码块重构成一个函数,再在上面加装饰器,然而这就显得不够优雅。因此我做出了下面的实现。

首先了解上下文管理的概念。大致是说Python中允许创建一种叫上下文管理器(Context Manager)的对象,它可以管理一个代码块执行时的上下文信息。具体的方法是创建一个类,并为其实现object.__enter__和object.__exit__方法,前者在进入代码块时自动执行,后者在完成代码块执行时自动执行。

在使用时,通过with和as关键字,将__enter__的返回值绑定到某一个变量名,这个返回值里可以储存代码块运行过程中得到的一些信息,在这里就是运行时间啦。具体的实现是创建一个计时器类Timer,在enter时记录代码块运行的开始时间,exit时记录完成时间、计算并储存耗时到Timer实例中。在使用时,将with Timer() as t加到要计时的代码块前面,t.elapse中将会储存代码块耗时,可以任意使用。

在这个基础上,我们还可以做出一个装饰器timer_来实现基于上下文的函数级计时器。

具体实现如下:

class Timer(object):
  '''Code Block Level Timer via Context'''
  def __enter__(self):
    self.start = datetime.now()
    return self
  def __exit__(self, *args):
    self.end = datetime.now()
    self.elapse = (self.end - self.start).total_seconds()
def timer_(func):
  '''Function Level Timer via Context & with Statement'''
  def timed(*args, **kw):
    with Timer() as t:
      result = func(*args, **kw)
    print("Processing time for {} is: {} seconds".format(func.__name__, t.elapse))
    return result
  return timed
def test_2(a):
  '''Code Block Level'''
  with Timer() as t:
    a *= 2
  print("Processing time for {} is: {} seconds".format('You Name It', t.elapse))
  return a
@timer_
def test_3(a):
  '''Function Level'''
  a *= 2
  return a
if __name__ == '__main__':
  print(test_2(2))
  print(test_3(3))

更灵活的实现

更优雅地,我们还可以使用contextlib自带的ContextDecorator,参考官方示例,做出既可以with又可以作为装饰器的计时器timer_elegant:

from datetime import datetime
from contextlib import ContextDecorator
class timer_elegant(ContextDecorator):
  '''Elegant Timer via ContextDecorator'''
  def __init__(self, name):
    self.name = name
  def __enter__(self):
    self.start = datetime.now()
  def __exit__(self, *args):
    self.end = datetime.now()
    self.elapse = (self.end - self.start).total_seconds()
    print("Processing time for {} is: {} seconds".format(self.name, self.elapse))
@timer_elegant('test_4')
def test_4(a):
  a *= 2
  return a
def test_5(a):
  a *= 2
  return a
if __name__ == '__main__':
  print(test_4(4))
 
  with timer_elegant('test 5'):
    result_5 = test_5(5)
  print(result_5)

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对三水点靠木的支持。如果你想了解更多相关内容请查看下面相关链接

Python 相关文章推荐
Python pass详细介绍及实例代码
Nov 24 Python
Python sqlite3事务处理方法实例分析
Jun 19 Python
python 创建弹出式菜单的实现代码
Jul 11 Python
Python 判断 有向图 是否有环的实例讲解
Feb 01 Python
python递归函数绘制分形树的方法
Jun 22 Python
对python字典元素的添加与修改方法详解
Jul 06 Python
对PyQt5中树结构的实现方法详解
Jun 17 Python
python pygame实现五子棋小游戏
Oct 26 Python
如何基于Python实现电子邮件的发送
Dec 16 Python
Python阶乘求和的代码详解
Feb 14 Python
Java ExcutorService优雅关闭方式解析
May 30 Python
Python实现七个基本算法的实例代码
Oct 08 Python
详解如何在Apache中运行Python WSGI应用
Jan 02 #Python
漂亮的Django Markdown富文本app插件的实现
Jan 02 #Python
对Python发送带header的http请求方法详解
Jan 02 #Python
Django渲染Markdown文章目录的方法示例
Jan 02 #Python
使用python 打开文件并做匹配处理的实例
Jan 02 #Python
对Xpath 获取子标签下所有文本的方法详解
Jan 02 #Python
python之验证码生成(gvcode与captcha)
Jan 02 #Python
You might like
php获取本周星期一具体日期的方法
2015/04/20 PHP
在CentOS上搭建LAMP+vsftpd环境的简单指南
2015/08/01 PHP
浅谈php中fopen不能创建中文文件名文件的问题
2017/02/06 PHP
PHP给前端返回一个JSON对象的实例讲解
2018/05/31 PHP
php自定义排序uasort函数示例【二维数组按指定键值排序】
2019/06/19 PHP
jquery实现html页面 div 假分页有原理有代码
2014/09/06 Javascript
C#中使用迭代器处理等待任务
2015/07/13 Javascript
基于jQuery仿淘宝产品图片放大镜代码分享
2020/06/23 Javascript
js捕捉键盘事件和按键键值的方法
2016/10/10 Javascript
jQuery File Upload文件上传插件使用详解
2016/12/06 Javascript
JavaScript实现格式化字符串函数String.format
2016/12/16 Javascript
webpack开发跨域问题解决办法
2017/08/03 Javascript
JS实现中文汉字按拼音排序的方法
2017/10/09 Javascript
小程序视频或音频自定义可拖拽进度条的示例代码
2018/09/30 Javascript
vue如何截取字符串
2019/05/06 Javascript
vue  elementUI 表单嵌套验证的实例代码
2019/11/06 Javascript
JavaScript实现字符串与HTML格式相互转换
2020/03/17 Javascript
python实现删除文件与目录的方法
2014/11/10 Python
浅析Python中return和finally共同挖的坑
2017/08/18 Python
python 一篇文章搞懂装饰器所有用法(建议收藏)
2019/08/23 Python
简单了解python shutil模块原理及使用方法
2020/04/28 Python
基于 HTML5 的 WebGL 3D 版俄罗斯方块的示例代码
2018/05/28 HTML / CSS
详解如何用canvas画一个微笑的表情
2019/03/14 HTML / CSS
伦敦时尚生活的缩影:LN-CC
2017/01/24 全球购物
一套Java笔试题
2016/08/20 面试题
编程实现当输入某产品代码则打印出该产品记录的功能
2014/05/03 面试题
广州地球村科技数据库题目
2016/04/25 面试题
大学生年度自我鉴定
2013/10/31 职场文书
公积金转移接收函
2014/01/11 职场文书
咖啡店自主创业商业计划书
2014/01/22 职场文书
毕业生求职信
2014/06/10 职场文书
党员个人剖析材料(四风问题)
2014/10/07 职场文书
学校推普周活动总结
2015/05/07 职场文书
解约证明模板
2015/06/19 职场文书
Redis安装使用RedisJSON模块的方法
2022/03/23 Redis
python blinker 信号库
2022/05/04 Python