У меня проблема с чтением из стандартного ввода или канала в 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 закончился?
Попробуйте следующее:
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
k = 0
try:
while True:
print sys.stdin.readline()
k += 1
except KeyboardInterrupt:
sys.stdout.flush()
pass
print k
в то время как 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, вы не можете использовать этот подход для задачи под рукой, и вы должны явно вызвать read() или readline()
Вот как я это сделал. Мне не нравилось какое-либо другое решение, они не казались очень питоническими.
Это создаст контейнер для ввода любого открытого файла для итерации по всем строкам. Это также позаботится о закрытии файла в конце диспетчера контекстов.
Мне кажется, что это, вероятно, как блок 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
Чтобы это работало, не дожидаясь окончания потока stdin, вы можете использовать его для чтения. Я думаю, что это самое простое решение.
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