标准输入不应等待“CTRL+D”

Question

我得到了一个应该从标准输入读取的简单 python 脚本。 因此，如果我将程序的标准输出重定向到我的 python 脚本的标准输入。

但我的程序记录到 python 脚本的内容只有在记录这些内容的程序被杀死时才会“到达”python 脚本。

但实际上，我希望在程序可用时立即处理程序记录的每一行，而不是在我应该 24/7 实际运行的程序退出时处理。

那么我怎样才能做到这一点呢？如何让标准输入在处理数据之前不等待 CTRL+D 或 EOF？

例子

# accept_stdin.py
import sys
import datetime

for line in sys.stdin:
    print datetime.datetime.now().second, line

# print_data.py
import time

print "1 foo"
time.sleep(3)
print "2 bar"

# bash
python print_data.py | python accept_stdin.py

Answer 1

像所有文件对象一样，sys.stdin 迭代器以块的形式读取输入；即使一行输入已准备好，迭代器也会在输出任何内容之前尝试读取块大小或 EOF。您可以使用没有此行为的 readline 方法解决此问题：

while True:
    line = sys.stdin.readline()
    if not line:
        # End of input
        break
    do_whatever_with(line)

您可以将其与 iter 的 2 参数形式结合使用 for 循环：

for line in iter(sys.stdin.readline, ''):
    do_whatever_with(line)

我建议在您的代码中留下注释，解释您为什么不使用常规迭代器。

Answer 2

这也是您的生产者程序的问题，即您将标准输出通过管道传输到 python 脚本的那个。

确实，由于该程序只打印而不刷新，因此它打印的数据保存在内部程序缓冲区中用于标准输出，而不是刷新到系统。

在print_data.py 中的print 语句之后添加sys.stdout.flush() 调用。

退出程序时您会看到数据，因为它会在退出时自动刷新。

请参阅this问题的解释，

Answer 3

正如@user2357112 所说，您需要使用：

for line in iter(sys.stdin.readline, ''):

之后，您需要使用 -u 标志启动 python 以立即刷新标准输入和标准输出。

python -u print_data.py | python -u accept_stdin.py

你也可以在shebang中指定flag。