Python脚本管道输出到head时无法提前退出的问题排查

当运行命令 python woo.py | head -n 1 时，期望head拿到第一行输出后Python脚本立即终止，但实际脚本会执行完time.sleep(10)才结束，即使添加了sys.stdout.flush()和SIGPIPE信号处理也无效。

问题原因 #

SIGPIPE信号仅在进程尝试向已关闭读端的管道写入数据时才会触发。你的代码写完两行并flush后，head已读取第一行并退出，关闭了管道读端，但此时脚本进入time.sleep(10)，期间没有任何写操作，因此不会触发SIGPIPE，进程会继续执行sleep。

解决方案 #

在flush之后主动执行一次写操作，触发可能的管道错误，从而让进程提前终止。以下是两种可行的修改方式：

方式一：捕获BrokenPipeError异常 #

import signal
import time
import sys 

signal.signal(signal.SIGPIPE, signal.SIG_DFL)

sys.stdout.write('hello\n')
sys.stdout.flush()  # 写完目标行后立即flush，匹配head -n1的需求

# 主动触发写操作检测管道状态
try:
    # 写入一个不影响输出结果的字符（如空格）
    sys.stdout.write(' ')
    sys.stdout.flush()
except BrokenPipeError:
    sys.exit(0)

time.sleep(10)
sys.stdout.write('the\n')
sys.stdout.write('world\n')

方式二：非阻塞IO检测管道状态 #

import signal
import time
import sys
import fcntl
import os

signal.signal(signal.SIGPIPE, signal.SIG_DFL)

sys.stdout.write('hello\n')
sys.stdout.flush()

# 将stdout设置为非阻塞模式
flags = fcntl.fcntl(sys.stdout.fileno(), fcntl.F_GETFL)
fcntl.fcntl(sys.stdout.fileno(), fcntl.F_SETFL, flags | os.O_NONBLOCK)

try:
    # 尝试写入触发管道错误
    sys.stdout.write(' ')
    sys.stdout.flush()
except BrokenPipeError:
    sys.exit(0)
finally:
    # 恢复原IO模式
    fcntl.fcntl(sys.stdout.fileno(), fcntl.F_SETFL, flags)

time.sleep(10)
sys.stdout.write('the\n')
sys.stdout.write('world\n')

补充说明 #

如果实际场景是循环输出内容，建议每次flush后都执行类似的检测，这样可以在head关闭管道后立即终止脚本，避免不必要的计算或等待。

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