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

itomcoil 2024-12-26 13:59 24 浏览

pip 安装

pip install tqdm

上一篇文章我们已经学习了 tqdm 的基本使用，最后还给出了一个循环进度封装类。但是在工作中往往任务都是比较多，并且存在大量子任务的情况。

上面代码的后台打印效果：

可以说是一团糟，每个任务的信息没有层次感。今天我们尝试简单封装一下，让进度信息更容易查看。

先看看，最终希望的进度代码使用方式：

看看后台打印信息：

可以看到所有的信息都按任务层级做了自动缩进。

看看实现思路。

基本定义如下：

也就是说，现在可以这么用：

接下来，利用 with 语法特性，在进入 with 时，把层级计数 +1 ，离开时 -1 ，即可实现自动识别当前任务层次：

行12： yield 配合 contextmanager 装饰器，可以很容易把一个普通的函数支持 with 语法。 yield 会把函数代码分成上下两个部分。当外部执行 with 时，会进入执行上半部分。离开 with 块时，会接着执行下半部分

接下来就非常简单了。在上半部分根据参数和层级，生成相应的打印或创建 tqdm 的逻辑即可：

yield 右边可以作为 with 使用时的返回值（行16）

也就是说，上面的 pbar 就是 yield 右边返回的东西。而代码中，之所以特别创建一个 Task 实例，只是为了可以方便定制一些方法给 pbar 。比如：

转发、关注我，私信"tqdm"，获得本期源码和数据。

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

tqdm安装

相关推荐