如何检查stdin是否有一些数据？

Question

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

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

但如果有人使用像

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

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

Answer 1

在Unix系统上,您可以执行以下操作:

import sys
import select

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

在Windows上,select模块可能只与套接字一起使用,因此您需要使用替代机制.

Answer 2

我一直在用

if not sys.stdin.isatty()

这是一个例子:

4 import sys
5
6 def main():
7     if not sys.stdin.isatty():
8         print "not sys.stdin.isatty"
9     else:
10         print "is  sys.stdin.isatty"

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

如果stdin中存在某些内容,则sys.stdin.isatty()返回false.

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