如何在 Python 中执行将数据写入标准输入的进程？

Question

我的 Python 脚本中有一个标志，它指定我是否设置和使用外部进程。这个过程是一个名为my_command 的命令，它从标准输入中获取数据。如果我要在命令行上运行它，它会是这样的：

$ my_command < data > result

我想使用 Python 脚本通过修改标准输入并将其提供给 my_command 来生成 data 行。

我正在做这样的事情：

import getopt, sys, os, stat, subprocess

# for argument's sake, let's say this is set to True for now
# in real life, I use getopt.getopt() to decide whether this is True or False
useProcess = True

if useProcess:
    process = subprocess.Popen(['my_command'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)

for line in sys.stdin:
    # parse line from standard input and modify it
    # we store the result in a variable called modified_line
    modified_line = line + "foo"

    # if we want to feed modified_line to my_command, do the following:
    if useProcess:
        process.stdin.write(modified_line)

    # otherwise, we just print the modified line
    else:
        print modified_line

但是，my_command 的行为就好像它没有收到任何数据并以错误状态退出。我做错了什么？

编辑

假设我的 Python 脚本名为 my_Python_script。假设我通常会通过标准输入传递 my_command 一个名为 data 的文件：

$ my_command < data > result

但现在我将其传递给my_Python_script：

$ my_Python_script < data > some_other_result

我希望my_Python_script 有条件地设置一个子进程，该子进程在data 的内容上运行my_command（在传递给my_command 之前由my_Python_script 修改）。这更有意义吗？

如果我使用bash 作为脚本语言，我会有条件地决定运行两个函数之一。一种是将数据行传输到my_command。另一个不会。这可以用 Python 完成吗？

Answer 1

写入标准输入后，需要关闭它：

    process.stdin.write(modified_line)
    process.stdin.close()

更新

我没有注意到 process.stdin.write() 是在 for 循环中执行的。在这种情况下，您应该将process.stdin.close() 移到循环之外。

另外，Raymond 提到我们也应该打电话给process.wait()。所以更新后的代码应该是：

for ...
    process.stdin.write(modified_line)

process.stdin.close()
process.wait()

Answer 2

除了@HaiVu提到的process.stdin.close()之外，你有没有做process.wait()等待命令执行完成才能得到结果？

Answer 3

您似乎混淆了参数和标准输入。你的命令应该是

$ <data> | mycommand result

一旦调用命令就会传入数据。

使用raw_input 内置函数完成输入。 (http://docs.python.org/2/library/functions.html)