如何检查标准输入是否有一些数据？

Question

在Python中，如何检查sys.stdin是否有数据？

我发现os.isatty(0)不仅可以检查stdin是否连接到TTY设备，还可以检查是否有可用的数据。

但是如果有人使用诸如

之类的代码

sys.stdin = cStringIO.StringIO("ddd")

之后使用os.isatty(0)，它仍然返回True。我需要做什么来检查stdin是否有数据？

Answer 1

在上面看到了很多答案，其中几乎所有的答案都需要 sys、select、os 等。但是，我有一个非常简单的想法，即如果有数据，如何从 stdin 连续获取输入。 我们可以使用 try, except 块来这样做。 作为，

while(1):
  try:
    inp = input()
  except:
     break

上面的循环将运行，我们不断检查输入并将输入存储在变量“inp”中，如果标准输入中没有输入，那么尝试将抛出错误，但是在除了块中断语句存在，所以while 循环将被终止

Answer 2

我一直在用

if not sys.stdin.isatty()

这是一个例子：

import sys

def main():
    if not sys.stdin.isatty():
        print "not sys.stdin.isatty"
    else:
        print "is  sys.stdin.isatty"

跑步

$ echo "asdf" | stdin.py
not sys.stdin.isatty

如果stdin 未连接到交互式输入设备（例如 tty），sys.stdin.isatty() 返回 false。

isatty(...)
    isatty() -> true or false. True if the file is connected to a tty device.

Answer 3

使用内置模块，这可以通过以下代码来实现，因为 Greg 已经给出了这个想法：

import fileinput
isStdin = True
for line in fileinput.input:
    # check if it is not stdin
    if not fileinput.isstdin():
        isStdin = False
        break
    # continue to read stdin
    print(line)
fileinput.close()

Answer 4

（编辑：这回答了一个相关问题，该问题已在此处合并。）

正如其他人所提到的，没有万无一失的方法可以知道数据是否可以从标准输入获得，因为 UNIX 不允许这样做（更一般地说，因为它无法猜测标准输入连接到的任何程序的未来行为）。

始终等待标准输入，即使可能什么都没有（这就是 grep 等所做的），或者向用户询问 - 参数。

Answer 5

取决于这里的目标：

import fileinput
for line in fileinput.input():
    do_something(line)

也很有用。

Answer 6

在 Unix 系统上，您可以执行以下操作：

import sys
import select

if select.select([sys.stdin,],[],[],0.0)[0]:
    print "Have data!"
else:
    print "No data"

在 Windows 上，选择模块只能与套接字一起使用，因此您需要使用替代机制。