为何print函数与异步生成器配合时无法正常工作？

这事儿我之前也碰到过，完全不是Python的bug，是标准输出缓冲机制加上Docker的终端配置共同导致的，咱们一步步拆解：

为什么print和logging的输出表现不一样？ #

核心差在它们用的输出流和缓冲策略：

• print()默认往stdout（标准输出）写内容。Python对stdout的缓冲规则是：如果是交互式终端环境（比如本地命令行），会用行缓冲——每输出一行就刷新一次；但如果是非交互式环境（比如Docker默认的无TTY模式），就会切换成块缓冲——只有缓冲区填满，或者整个程序退出时，才会把缓存的内容一次性吐出来。
• logging模块默认是往stderr（标准错误）写日志的，stderr在绝大多数场景下都是无缓冲的，不管什么环境，内容都会实时输出，不会被憋着。

Docker的-it参数为啥这么关键？ #

你提到只有在Docker环境下不加-it才会复现，这是因为：

• 加-it的时候，Docker会给容器分配一个伪TTY（模拟终端），容器里的Python进程会认为自己在交互式环境里，stdout就会用行缓冲，print()的内容自然就实时显示了；
• 不加-it的话，容器的stdout/stderr是纯管道模式，属于非交互式环境，stdout就会启用块缓冲：
  • 用有限的异步生成器时，程序最后会正常退出，退出前Python会强制刷新stdout缓冲区，所以你最后能看到所有print的内容堆在一起出来；
  • 用无限生成器时，程序永远不会退出，缓冲区一直没机会被刷新，所以print的内容就永远看不到了。

怎么解决这个问题？ #

如果想在无TTY的Docker环境下让print实时输出，有这几个办法：

• 启动容器时加-it参数，模拟交互式终端环境；
• 每次调用print时强制刷新缓冲区：print(i, flush=True)；
• 把print的输出改写到stderr：import sys; print(i, file=sys.stderr)；
• 启动Python时加-u参数，全局禁用stdout的缓冲：python -u your_script.py。

内容的提问来源于stack exchange，提问作者Daniil Stepanenko