Чтение из stdin И пересылка его в подпроцесс в Python

Question

Я пишу скрипт-оболочку для программы, которая может принимать входные данные от STDIN. Мой скрипт должен обрабатывать каждую строку файла, но ему также необходимо перенаправить STDIN в программу, которую она обертывает. В минималистской форме, это выглядит примерно так:

import subprocess 
import sys 

for line in sys.stdin: 
    # Do something with each line 
    pass 

subprocess.call(['cat']) 

Обратите внимание, что я на самом деле не пытаюсь обернуть cat, он просто служит в качестве примера для демонстрации того или нет STDIN в пересылаются правильно.

В приведенном выше примере, если я прокомментирую цикл for, он работает правильно. Но если я запустил его с помощью цикла for, ничего не будет перенаправлено, потому что я уже прочитал до конца STDIN. Я не могу seek(0) в начале файла, потому что вы не можете искать в потоках.

Одно из возможных решений, чтобы прочитать весь файл в память:

import subprocess 
import sys 

lines = sys.stdin.readlines() 
for line in lines: 
    # Do something with each line 
    pass 

p = subprocess.Popen(['cat'], stdin=subprocess.PIPE) 
p.communicate(''.join(lines)) 

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

Дополнительные ограничения:

1. Подпроцесс можно назвать только один раз. Поэтому я не могу читать строку за раз, обрабатывать ее и пересылать ее подпроцессу.
2. Решение должно работать в Python 2.6

Answer 1

ли эта работа для вас?

#!/usr/bin/env python2 
import subprocess 
import sys 

p = subprocess.Popen(['cat'], stdin = subprocess.PIPE) 

line = sys.stdin.readline() 

#################### 
# Insert work here # 
#################### 

line = line.upper() 

#################### 

p.communicate(line) 

Пример:

$ echo "hello world" | ./wrapper.py 
HELLO WORLD