盘一盘python进度库-tqdm的嵌套子任务

我是数据外星人 2024-11-07 15:34:32

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",获得本期源码和数据。

不要忘记一键三连。你的点赞、收藏、关注,是我创作的动力。

0 阅读:0

我是数据外星人

简介:感谢大家的关注