Tìm hiểu xem có đầu vào từ một đường ống hay không trong Python?

Question

(Tôi nghĩ) Tôi biết cách đọc từ một đường ống. Ý tôi là các cuộc gọi trên bash như thế này

echo Bähm | ./script.py 

này Python3 kịch bản làm việc với điều đó.

#!/usr/bin/env python3 
import sys 

x = sys.stdin.read() 

if x: 
    print(x) 
else: 
    print('no pipe') 

sys.exit() 

Nhưng khi tôi chỉ đơn giản là làm điều này trên bash

./script.py 

có gì xảy ra bởi vì nó chờ đợi cho đầu vào.

Vì vậy, tôi muốn kiểm tra (khi bắt đầu tập lệnh) nếu có đầu vào từ đường ống hay không. Nhưng tôi không biết tại sao size hoặc len không hoạt động trên sys.stdin hoặc sys.stdin.buffer.

Có thể có sự khác biệt trong việc xử lý giữa phiên bản 2 và 3 của Python?

Answer 1

có gì xảy ra bởi vì nó chờ đợi cho đầu vào

Và đó là hoàn toàn tốt. Bạn muốn sửa một cái gì đó mà không cần bất kỳ sửa chữa, và như là một sản phẩm phụ bạn đang thực sự tạo ra vấn đề.

Hãy tưởng tượng quy trình đường ống thực hiện một số phép tính nặng và do đó, cần một thời gian để tạo đầu ra. Bạn không thể kiểm tra tính sẵn sàng của dữ liệu trong tập lệnh python của bạn. Bạn không biết khi nào dữ liệu sẽ có sẵn. Bạn phải kiên nhẫn.

Vì vậy, tôi muốn kiểm tra (khi bắt đầu tập lệnh) nếu có đầu vào từ đường ống hay không.

Không. Nếu, vì một lý do nào đó, ứng dụng của bạn không có nghĩa là chạy với một tty (chờ đầu vào của người dùng), hãy kiểm tra điều đó.

sys.stdin.isatty() 

Chỉ kiểm tra tính khả dụng của dữ liệu nếu có hạn chế về thời gian khó: ví dụ: người dùng phải trả lời trong vòng 2 giây.

Answer 2

Bạn có thể đặt tệp nhập đầu vào thành không bị chặn và bắt giữ IOError: [Errno 11] Resource temporarily unavailable khi không có gì được truyền.

import fcntl 
import os 
import sys 

# make stdin a non-blocking file 
fd = sys.stdin.fileno() 
fl = fcntl.fcntl(fd, fcntl.F_GETFL) 
fcntl.fcntl(fd, fcntl.F_SETFL, fl | os.O_NONBLOCK) 

try: 
    print(sys.stdin.read()) 
except: 
    print('No input') 

Answer 3

Bạn có thể sử dụng lựa chọn, nhưng nó không hoạt động trên Windows:

import sys 
import select 

# Input: zero or more file-like objects 
# Returns: list of objects that can be read 
def can_read_files(*args): 
    return select.select(args, [], [], 0.0)[0] 

if can_read_files(sys.stdin): 
    print(sys.stdin.read()) 
else: 
    print('No input') 

Đối với chúng tôi nó inline:

if select.select([sys.stdin], [], [], 0.0)[0]: 
    ... 

Answer 4

Ngoài ra để @ câu trả lời Cyrbil nếu bạn thực sự cần phải biết nếu đầu vào được nhập vào, bạn có thể sử dụng isatty:

from sys import stdin 
from os import isatty 

is_pipe = not isatty(stdin.fileno())