Python：如何判断输入文件是否有输入待处理而无需等待

Question

Normally, you process a file line by line in Python using a loop like:

import sys
for s in sys.stdin:
    # do something with the line in s

or

import sys
while True:
    line = sys,stdin.readline()
    if len(line) == 0: break
    # process input line

Of course, you can also use raw_input() in soemthing like this:

try:
    while True:
        s = raw_input()
        # process input line
except EOFError:
    # there's EOF.

Of course in all these cases, if there's no input ready to be read, the underlying read() operation suspends waiting for I/O.

What I want to do is see if there is input pending without suspending, so I can read until input is exhausted and then go do something else. That is, I'd like to be able to do something like

while "there is input pending":
    #get the input

but when no more input is pending, break the loop.

Answer 1

If you are using some variant of Unix, and your standard input is a pipe and not a file, you can use the select module to check to see whether there is waiting input. At a minimum, the code might look like:

import select

rlist, wlist, elist = select.select([sys.stdin], [], [])
if rlist:
    s = raw_input()
else:
    pass # no input ready right now

Answer 2

Okay, here's something that works well on UNIX:

import sys
import select
import tty
import termios


def isData():
    return select.select([sys.stdin], [], [], 0) == ([sys.stdin], [], [])

old_settings = termios.tcgetattr(sys.stdin)
try:
    tty.setcbreak(sys.stdin.fileno())

    i = 0
    while 1:
        print i
        i += 1

        if isData():
            c = sys.stdin.read(1)
            if c == '\x1b':         # x1b is ESC
                break

finally:
    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)

I'll modify/extend this answer when I have a chance to make a somewhat better test program. I'm (so far) unclear on how well tty and termios work on Windows.

Update : Grmph. This depends on select . There are reasons I don't like Windows.