python - 使用进程替换作为 Python 的输入文件两次

Question

考虑以下 python 脚本

#test.py
import sys
inputfile=sys.argv[1]
with open(inputfile,'r') as f:
    for line in f.readlines():
        print line

with open(inputfile,'r') as f:
    for line in f.readlines():
        print line

现在我想在替代进程上运行 test.py，例如，

python test.py <( cat file | head -10)

似乎第二个 f.readlines 返回空。为什么会这样？有没有一种方法可以做到这一点而无需指定两个输入文件？

Answer 1

• 这是为什么。
  • 进程替换通过创建命名管道来工作。因此，所有数据都在第一个打开/读取循环中消耗。
• 有没有办法不用指定两个输入文件就可以做到这一点。
  • 如何在使用数据之前缓冲数据。

这是一个示例代码

import sys
import StringIO
inputfile=sys.argv[1]

buffer = StringIO.StringIO()

# buffering
with open(inputfile, 'r') as f:
    buffer.write(f.read())

# use it 
buffer.seek(0)
for line in buffer:
    print line

# use it again
buffer.seek(0)
for line in buffer:
    print line