从标准输入读取并将其转发到 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