在 Python 中，如何从标准输入读取数据，直到管道进程终止？

Question

我想做这样的事情：

data = json.load(open(sys.argv[1]) if len(sys.argv) > 1 else sys.stdin)

在这个例子中可以简化为

data = json.load(sys.stdin)

但是 json.load 运行一个 .read() 可能不包含我通过提供的整个 JSON 字符串

generate_json_output | myprogram.py

是否有推荐的方法从stdin 读取直到generate_json_output 终止？我试图抓住BrokenPipeError，但我跑的时候它没有被提升

while True:
    data += sys.stdin.read()

Answer 1

除了while 循环，您可以使用下面的 for 循环来迭代标准输入。

import sys

data = ''
for line in sys.stdin:
    data += line

可以写成一个表达式：

data = "".join(sys.stdin)

使用管道生成的输入运行此程序会读取标准输入，直到它终止。

t.json:

{
    'a': 2,
    'b': 3
}

运行 cat t.json | python a.py 将完整的文件打印到标准输出。