通过tqdm.write（）在python脚本中重定向打印命令

Question

我tqdm在Python中使用脚本来显示控制台进度条。但是，我必须调用将print消息也发送到控制台且不能更改的函数。通常，在控制台中显示进度条时写入控制台会使显示混乱，如下所示：

from time import sleep
from tqdm import tqdm

def blabla():
  print "Foo blabla"

for k in tqdm(range(3)):
  blabla()
  sleep(.5)

这将创建输出：

0%|                                           | 0/3 [00:00<?, ?it/s]Foo
blabla
33%|###########6                       | 1/3 [00:00<00:01,  2.00it/s]Foo
blabla
67%|#######################3           | 2/3 [00:01<00:00,  2.00it/s]Foo
blabla
100%|###################################| 3/3 [00:01<00:00,  2.00it/s]

根据tqdm该方法的文档，tqdm.write()提供了一种在不中断显示的进度条的情况下将消息写入控制台的方法。因此，此代码段提供了正确的输出：

from time import sleep
from tqdm import tqdm

def blabla():
  tqdm.write("Foo blabla")

for k in tqdm(range(3)):
  blabla()
  sleep(.5)

看起来像这样：

Foo blabla
Foo blabla
Foo blabla
100%|###################################| 3/3 [00:01<00:00,  1.99it/s]

另一方面，存在这种[解决方案，该解决方案允许通过相当优雅地重定向sys.stdout到void中来使那些功能静音。这对于使功能静音非常有效。

由于我仍然希望显示这些功能的消息而又不破坏进度条，因此我尝试通过重定向sys.stdout到 老
解决方案tqdm.write()，然后依次tqdm.write()写入 老
解决方案，将两种解决方案合并为一个sys.stdout。结果为代码段：

from time import sleep

import contextlib
import sys

from tqdm import tqdm

class DummyFile(object):
  file = None
  def __init__(self, file):
    self.file = file

  def write(self, x):
    tqdm.write(x, file=self.file)

@contextlib.contextmanager
def nostdout():
    save_stdout = sys.stdout
    sys.stdout = DummyFile(save_stdout)
    yield
    sys.stdout = save_stdout

def blabla():
  print "Foo blabla"

for k in tqdm(range(3)):
  with nostdout():
    blabla()
    sleep(.5)

但是，这实际上会像以前那样产生更加混乱的输出：

0%|                                           | 0/3 [00:00<?, ?it/s]Foo
blabla


33%|###########6                       | 1/3 [00:00<00:01,  2.00it/s]Foo
blabla


67%|#######################3           | 2/3 [00:01<00:00,  2.00it/s]Foo
blabla


100%|###################################| 3/3 [00:01<00:00,  2.00it/s]

仅供参考：tqdm.write(..., end="")内部调用DummyFile.write()产生的结果与仍然混乱的第一个输出相同。

我不明白为什么这行不通，因为tqdm.write()应该在写消息之前先清除进度条，然后再重写进度条。

我想念什么？

Answer 1

重定向sys.stdout总是很棘手，并且当两个应用程序同时纠缠在一起时，它将变成一场噩梦。

这里的窍门是tqdm默认情况下打印到sys.stderr，而不是sys.stdout。通常，tqdm对于这两个特殊通道具有反混合策略，但是由于您正在重定向sys.stdout，tqdm因此由于文件句柄发生更改而感到困惑。

因此，您只需要显式指定file=sys.stdoutto即可tqdm使用：

from time import sleep

import contextlib
import sys

from tqdm import tqdm

class DummyFile(object):
  file = None
  def __init__(self, file):
    self.file = file

  def write(self, x):
    # Avoid print() second call (useless \n)
    if len(x.rstrip()) > 0:
        tqdm.write(x, file=self.file)

@contextlib.contextmanager
def nostdout():
    save_stdout = sys.stdout
    sys.stdout = DummyFile(sys.stdout)
    yield
    sys.stdout = save_stdout

def blabla():
  print("Foo blabla")

# tqdm call to sys.stdout must be done BEFORE stdout redirection
# and you need to specify sys.stdout, not sys.stderr (default)
for _ in tqdm(range(3), file=sys.stdout):
    with nostdout():
        blabla()
        sleep(.5)

print('Done!')

我还增加了一些技巧，让输出更好的（例如，没有无用的\n使用时print()无end=''）。

/编辑：其实好像你可以做stdout启动后重定向tqdm，您只需要指定dynamic_ncols=True的tqdm。