Làm cách nào để kiểm tra xem liệu stdin có một số dữ liệu?

Question

Trong Python, làm thế nào để bạn kiểm tra xem sys.stdin có dữ liệu hay không?

Tôi thấy rằng os.isatty(0) không chỉ có thể kiểm tra xem stdin có được kết nối với thiết bị TTY hay không, mà còn nếu có sẵn dữ liệu.

Nhưng nếu ai đó sử dụng mã như

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

và sau đó sử dụng os.isatty(0), nó vẫn trả về True. Tôi cần phải làm gì để kiểm tra xem liệu stdin có dữ liệu?

Answer 1

Trên hệ thống UNIX bạn có thể làm như sau:

import sys 
import select 

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

Trên Windows các chọn mô-đun chỉ có thể được sử dụng với ổ cắm mặc dù vậy bạn sẽ cần phải sử dụng một cơ chế thay thế.

Answer 2

Tùy thuộc vào mục tiêu ở đây:

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

cũng có thể hữu ích.

Answer 3

Tôi đã sử dụng

if not sys.stdin.isatty() 

Dưới đây là một ví dụ:

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 

sys.stdin.isatty() trả về false nếu có điều gì đó trong stdin.

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

Answer 4

Như mentionned bởi những người khác, không có cách nào hoàn hảo để làm điều đó, bởi vì UNIX không cho phép nó. Luôn đợi stdin, ngay cả khi có thể không có gì (đó là những gì grep, v.v.) hoặc yêu cầu người dùng tham số -.