python - 使用 python tqdm 库重定向 stdout 和 stderr

Question

我在 Python 中使用 tqdm 来显示控制台进度条。

我有一个来自另一个库的函数，它偶尔会在 tqdm 循环内同时写入 stdout 和 stderr 。我无法破解该函数的源代码。

虽然此文档显示了如何重定向 sys.stdout，但它并不容易推广到 stderr，因为我只能将 stdout 或 stderr 之一传递给file=tqdm's 中的参数__init__。请参阅此问题及其接受的答案以获取说明问题的最少代码。

如何将stdout和 stderr 一起重定向？

Answer 1

Python 在标准库中为您提供了一些帮助程序，请查看contextlib：

>>> import io, sys
>>> from contextlib import redirect_stdout, redirect_stderr
>>> from tqdm import tqdm
>>> def foo():
...     print('spam to stdout')
...     print('spam to stderr', file=sys.stderr)
...     
>>> out = io.StringIO()
>>> err = io.StringIO()
>>> with redirect_stdout(out), redirect_stderr(err):
...     for x in tqdm(range(3), file=sys.__stdout__):
...         print(x)  # more spam
...         foo()
...         
100%|██████████| 3/3 [00:00<00:00, 29330.80it/s]
>>> out.getvalue()
'0\nspam to stdout\n1\nspam to stdout\n2\nspam to stdout\n'
>>> err.getvalue()
'spam to stderr\nspam to stderr\nspam to stderr\n'