比tqdm更好用的进度条工具rich.progress，你值得拥有

最新推荐文章于 2025-10-20 13:31:41 发布

原创

最新推荐文章于 2025-10-20 13:31:41 发布 · 4.6k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

引子

在开发fastnlp过程遇到一个需求：嵌套使用tdqm进度条，能够在任意位置使用print而不发生冲突。为解决print打印内容乱序的问题，我们将内置的write函数置换为tqdm.write函数。

import sys
import time
import contextlib

from tqdm import tqdm


class DummyFile:
    def __init__(self, file):
        if file is None:
            file = sys.stderr
        self.file = file

    def write(self, x):
        if len(x.rstrip()) > 0:
            tqdm.write(str(x), file=self.file)


@contextlib.contextmanager
def redirect_stdout(file=None):
    if file is None:
        file = sys.stderr
    old_stdout = file
    sys.stdout.write = DummyFile(file).write
    yield
    sys.stdout.write = old_stdout.write


with redirect_stdout():
    with tqdm(total=10</