在Python中进行冲洗时，如何防止BrokenPipeError？

Question

问题：有没有办法使用flush=True该print()函数的方法BrokenPipeError？

我有一个脚本pipe.py：

for i in range(4000):
    print(i)

我从Unix命令行这样称呼它：

python3 pipe.py | head -n3000

它返回：

0
1
2

这个脚本也是如此：

import sys
for i in range(4000):
    print(i)
    sys.stdout.flush()

但是，当我运行此脚本并将其通过管道传递给时head -n3000：

for i in range(4000):
    print(i, flush=True)

然后我得到这个错误：

    print(i, flush=True)
BrokenPipeError: [Errno 32] Broken pipe
Exception BrokenPipeError: BrokenPipeError(32, 'Broken pipe') in <_io.TextIOWrapper name='<stdout>' mode='w' encoding='UTF-8'> ignored

我也尝试了下面的解决方案，但仍然得到了BrokenPipeError：

import sys
for i in range(4000):
    try:
        print(i, flush=True)
    except BrokenPipeError:
        sys.exit()

Answer 1

这BrokenPipeError是正常现象，因为读过程（头）会终止并关闭管道的末端，而写过程（python）仍会尝试进行写操作。

Is 是
一种异常情况，并且python脚本收到一个BrokenPipeError-更准确地说，Python解释器收到一个系统SIGPIPE信号，该信号捕获并引发该信号BrokenPipeError以允许脚本处理错误。

而且您可以有效地处理该错误，因为在上一个示例中，您仅看到一条消息，指出忽略了该异常-
好的，这不是真的，但似乎与Python中的这个开放问题有关：Python开发人员认为警告用户注意非常重要异常情况。

真正发生的是，即使您捕获到异常，Python解释器AFAIK始终会在stderr上发出信号。但是您只需要在退出前关闭stderr即可消除该消息。

我将您的脚本稍微更改为：

• 像上一个示例一样捕获错误
• 捕获IOError（在Windows64上为Python34）或BrokenPipeError（在FreeBSD 9.0上为Python 33）-并显示一条消息
• 在stderr上显示自定义的 “完成” 消息（由于管道断开，stdout已关闭）
• *退出以 *关闭 消息之前 关闭stderr

这是我使用的脚本：

import sys

try:
    for i in range(4000):
            print(i, flush=True)
except (BrokenPipeError, IOError):
    print ('BrokenPipeError caught', file = sys.stderr)

print ('Done', file=sys.stderr)
sys.stderr.close()

这是以下结果python3.3 pipe.py | head -10：

0
1
2
3
4
5
6
7
8
9
BrokenPipeError caught
Done

如果您不想要多余的消息，请使用：

import sys

try:
    for i in range(4000):
            print(i, flush=True)
except (BrokenPipeError, IOError):
    pass

sys.stderr.close()