使用Linux管道时写入sys.stdout出现BrokenPipeError的原因是什么？

根本原因 #

当你用管道连接command.py和head时，head读取指定行数（默认10行）后会立即关闭管道的读端。此时你的Python程序还在尝试往sys.stdout（也就是管道的写端）写入数据，操作系统会向Python进程发送SIGPIPE信号，Python将其转换为BrokenPipeError异常。

你的代码虽然用except Exception捕获异常，但存在两个关键问题：

1. 当fp.write(data)触发BrokenPipeError后，进入except块执行print("Could not write to file", e)——而print本身也是往sys.stdout写数据，此时管道已经断开，会再次触发BrokenPipeError，这个二次异常没有被捕获，导致程序崩溃并打印堆栈信息。
2. Python对BrokenPipeError的默认处理逻辑会在异常未被完全处理时强制输出回溯，即使你捕获了第一次异常。

修复方案 #

针对这个问题，你可以从两个方向优化代码：

方案1：精准捕获异常并避免二次写入管道

修改代码，单独捕获BrokenPipeError，且将错误信息写入sys.stderr（默认不经过管道，不会触发二次异常）：

import sys

def write(outfile: str, data: str):
    fp = sys.stdout
    try:
        if outfile:
            fp = open(outfile, "w+")
        
        fp.write(data)
    except BrokenPipeError:
        # 管道已断开，直接退出进程
        sys.exit(1)
    except Exception as e:
        # 错误信息写入stderr，避免影响stdout管道
        print(f"Could not write to file: {e}", file=sys.stderr)
    finally:
        # 仅关闭自己打开的文件，sys.stdout无需手动关闭
        if outfile:
            fp.close()

方案2：直接处理SIGPIPE信号

通过注册信号处理函数，让进程在收到SIGPIPE时直接退出，不触发Python异常：

import sys
import signal

# 捕获SIGPIPE信号，收到后立即退出进程
def handle_sigpipe(_, __):
    sys.exit(1)

signal.signal(signal.SIGPIPE, handle_sigpipe)

def write(outfile: str, data: str):
    fp = sys.stdout
    try:
        if outfile:
            fp = open(outfile, "w+")
        
        fp.write(data)
    except Exception as e:
        print(f"Could not write to file: {e}", file=sys.stderr)
    finally:
        if outfile:
            fp.close()

额外注意事项 #

• 不要手动关闭sys.stdout，这会导致后续所有标准输出操作失败。
• 使用finally块确保打开的文件被正确关闭，避免资源泄漏。

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