Как вы читаете из stdin в Python из канала, который не имеет окончания - PullRequest
Новая
       12

Как вы читаете из stdin в Python из канала, который не имеет окончания

21 голосов
/ 17 августа 2011

У меня проблема с чтением из стандартного ввода или канала в Python, когда канал из "открытого" (не знаю правильное имя) файл.

у меня в качестве примера pipetest.py: 

import sys
import time
k = 0
try:
   for line in sys.stdin:
      k = k + 1
      print line
except KeyboardInterrupt:
   sys.stdout.flush()
   pass
print k

Я запускаю программу с продолжением вывода и Ctrl + c через некоторое время 

$ ping 127.0.0.1 | python pipetest.py
^C0

Я не получаю вывод. Но если я иду через обычный файл, он работает. 

$ ping 127.0.0.1 > testfile.txt

это заканчивается Ctrl + c через короткое время 

$ cat testfile.txt |  python pipetest.py

PING 127.0.0.1 (127.0.0.1) 56(84) bytes of data.
64 bytes from 127.0.0.1: icmp_seq=1 ttl=64 time=0.017 ms
64 bytes from 127.0.0.1: icmp_seq=2 ttl=64 time=0.015 ms
64 bytes from 127.0.0.1: icmp_seq=3 ttl=64 time=0.014 ms
64 bytes from 127.0.0.1: icmp_seq=4 ttl=64 time=0.013 ms
64 bytes from 127.0.0.1: icmp_seq=5 ttl=64 time=0.012 ms

--- 127.0.0.1 ping statistics ---
5 packets transmitted, 5 received, 0% packet loss, time 3998ms
rtt min/avg/max/mdev = 0.012/0.014/0.017/0.003 ms
10

Как мне получить какой-либо вывод до завершения программы, в этом случае ping закончился?

Ответы [ 5 ]

26 голосов
/ 17 августа 2011

Попробуйте следующее: 

import sys
import time
k = 0
try:
    buff = ''
    while True:
        buff += sys.stdin.read(1)
        if buff.endswith('\n'):
            print buff[:-1]
            buff = ''
            k = k + 1
except KeyboardInterrupt:
   sys.stdout.flush()
   pass
print k
8 голосов
/ 20 июля 2017

Чтобы это работало, не дожидаясь окончания потока stdin, вы можете его прочитать на readline. Я думаю, что это самое простое решение. 

import sys
k = 0
try:
   for line in iter(sys.stdin.readline, b''):
      k = k + 1
      print line
except KeyboardInterrupt:
   sys.stdout.flush()
   pass
print k
7 голосов
/ 17 августа 2011 
k = 0
try:
    while True:
        print sys.stdin.readline()
        k += 1
except KeyboardInterrupt:
    sys.stdout.flush()
    pass
print k
4 голосов
/ 17 августа 2011

, хотя sys.stdin является файловым объектом, то есть вы можете перебирать его строки, он будет блокироваться до тех пор, пока не будет вставлен EOF.

Поведение может быть описано с помощью следующего псевдокода: 

while True:
    input = ""
    c = stdin.read(1)
    while c is not EOF:
        input += c
        c = stdin.read(1)
    for line in input.split('\n'):
        yield line

это означает, что, хотя вы можете перебирать строки sys.stdin, вы не можете использовать этот подход для задачи вhand, и вы должны явно вызвать read () или readline ()

3 голосов
/ 12 августа 2015

Вот так я и сделал.Мне не очень понравились какие-либо другие решения, они не казались очень питонными.

Это создаст контейнер для любого ввода открытого файла, который будет повторяться по всем строкам.Это также позаботится о закрытии файла в конце диспетчера контекста.

Мне кажется, что, вероятно, именно так должен работать блок for line in sys.stdin: по умолчанию. 

class FileInput(object):                                                        
    def __init__(self, file):                                                   
        self.file = file                                                       

    def __enter__(self):                                                        
        return self                                                             

    def __exit__(self, *args, **kwargs):                                        
        self.file.close()                                                       

    def __iter__(self):                                                         
        return self                                                             

    def next(self):                                                             
        line = self.file.readline()                                             

        if line == None or line == "":                                          
            raise StopIteration                                                 

        return line  

with FileInput(sys.stdin) as f:
    for line in f:
        print f

with FileInput(open('tmpfile') as f:
    for line in f:
        print f

Из командной строки должны работать оба следующих параметра: 

tail -f /var/log/debug.log | python fileinput.py
cat /var/log/debug.log | python fileinput.py
...