使用python tqdm库重定向stdout和stderr [英] Redirect both stdout and stderr with python tqdm library
问题描述
我正在Python中使用tqdm来显示控制台进度条.
I'm using tqdm in Python to display console progress bars.
我有另一个库中的函数,偶尔在tqdm循环中写入stdout和stderr.我无法破解该函数的源代码.
I have a function from another library that occasionally writes to both stdout and stderr inside the tqdm loop. I cannot hack the source code of that function.
此文档显示了如何重定向sys.stdout,但没有因为我只能将stdout或stderr之一传递给tqdm的__init__
中的file=
参数,所以很容易将其推广到stderr.请参阅此问题及其可接受的答案,以说明问题的最小代码.
While this doc shows how to redirect sys.stdout, it doesn't easily generalize to stderr because I can pass only one of stdout or stderr to the file=
parameter in tqdm's __init__
. Please refer to this question and its accepted answer for the minimal code that illustrates the problem.
如何将 的stdout和stderr重定向在一起?
How do I redirect both stdout and stderr together?
推荐答案
Python provides some helpers for you in the standard libraries, look in 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'
这篇关于使用python tqdm库重定向stdout和stderr的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!