pip 安装
pip install tqdm上一篇文章我们已经学习了 tqdm 的基本使用,最后还给出了一个循环进度封装类。但是在工作中往往任务都是比较多,并且存在大量子任务的情况。
上面代码的后台打印效果:
可以说是一团糟,每个任务的信息没有层次感。今天我们尝试简单封装一下,让进度信息更容易查看。
先看看,最终希望的进度代码使用方式:
行2:需要全局生成一个进度信息管理的对象行7:通过 with 语法表示任务范围。此时后台应该打印信息行37:在任何的函数中可以随时打印信息行39: with 语法创建的对象,可以再次调用 print 方法打印信息看看后台打印信息:
可以看到所有的信息都按任务层级做了自动缩进。
看看实现思路。
基本定义如下:
行8:全局使用,所以得有一个变量记录当前是哪一层任务行10-17:一个 task 方法,让方法的返回值可以使用 with 语法,得用 "contextmanager" 装饰器也就是说,现在可以这么用:
接下来,利用 with 语法特性,在进入 with 时,把层级计数 +1 ,离开时 -1 ,即可实现自动识别当前任务层次:
行12: yield 配合 contextmanager 装饰器,可以很容易把一个普通的函数支持 with 语法。 yield 会把函数代码分成上下两个部分。当外部执行 with 时,会进入执行上半部分。离开 with 块时,会接着执行下半部分接下来就非常简单了。在上半部分根据参数和层级,生成相应的打印或创建 tqdm 的逻辑即可:
yield 右边可以作为 with 使用时的返回值(行16)
也就是说,上面的 pbar 就是 yield 右边返回的东西。而代码中,之所以特别创建一个 Task 实例,只是为了可以方便定制一些方法给 pbar 。比如:
转发、关注我,私信"tqdm",获得本期源码和数据。
不要忘记一键三连。你的点赞、收藏、关注,是我创作的动力。