81

我正在尝试编写一个启动子进程并写入子进程标准输入的 Python 脚本。如果子进程崩溃,我还希望能够确定要采取的行动。

我试图启动的过程是一个名为的程序nuke,它有自己的内置 Python 版本,我希望能够向它提交命令,然后在命令执行后告诉它退出。到目前为止,我已经弄清楚,如果我在命令提示符下启动 Python,然后nuke作为子进程启动,那么我可以输入命令nuke,但我希望能够将这一切都放在一个脚本中,以便主Python 程序可以启动nuke,然后写入它的标准输入(因此写入它的内置 Python 版本)并告诉它做一些时髦的事情,所以我写了一个这样开始的脚本nuke

subprocess.call(["C:/Program Files/Nuke6.3v5/Nuke6.3", "-t", "E:/NukeTest/test.nk"])

然后什么也没有发生,因为nuke正在等待用户输入。我现在将如何写入标准输入?

我这样做是因为我正在运行一个插件nuke,导致它在渲染多个帧时间歇性地崩溃。所以我希望这个脚本能够启动nuke,告诉它做某事,然后如果它崩溃,再试一次。因此,如果有一种方法可以捕捉崩溃并且仍然可以,那就太好了。

4

3 回答 3

102

使用可能会更好communicate

from subprocess import Popen, PIPE, STDOUT
p = Popen(['myapp'], stdout=PIPE, stdin=PIPE, stderr=PIPE)
stdout_data = p.communicate(input='data_to_write')[0]

“更好”,因为这个警告:

使用communicate() 而不是.stdin.write、.stdout.read 或.stderr.read 以避免由于任何其他OS 管道缓冲区填满并阻塞子进程而导致的死锁。

于 2011-12-12T13:52:56.407 回答
10

为了澄清一些观点:

正如jro提到的,正确的方法是使用subprocess.communicate.

然而,当stdin使用subprocess.communicatewith喂食时input,您需要stdin=subprocess.PIPE根据docs启动子流程。

请注意,如果您想将数据发送到进程的标准输入,您需要使用标准输入=PIPE 创建 Popen 对象。同样,要在结果元组中获得除 None 以外的任何内容,您也需要提供 stdout=PIPE 和/或 stderr=PIPE 。

qed在评论中还提到,对于 Python 3.4,您需要对字符串进行编码,这意味着您需要将 Bytes 传递给input而不是 a string。这并不完全正确。根据文档,如果流以文本模式打开,则输入应该是一个字符串(源是同一页面)。

如果流以文本模式打开,则输入必须是字符串。否则,它必须是字节。

因此,如果流没有在文本模式下显式打开,那么类似下面的东西应该可以工作:

import subprocess
command = ['myapp', '--arg1', 'value_for_arg1']
p = subprocess.Popen(command, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
output = p.communicate(input='some data'.encode())[0]

stderr故意将上面的值STDOUT作为示例。

话虽如此,有时您可能想要另一个流程的输出,而不是从头开始构建它。假设您要运行相当于echo -n 'CATCH\nme' | grep -i catch | wc -m. 这通常应该返回 'CATCH' 中的数字字符加上一个换行符,结果为 6。这里的回显点是将CATCH\nme数据提供给 grep。因此,我们可以将 Python 子进程链中的标准输入作为变量提供给 grep,然后将标准输出作为 PIPE 传递给wc进程的标准输入(同时,去掉额外的换行符):

import subprocess

what_to_catch = 'catch'
what_to_feed = 'CATCH\nme'

# We create the first subprocess, note that we need stdin=PIPE and stdout=PIPE
p1 = subprocess.Popen(['grep', '-i', what_to_catch], stdin=subprocess.PIPE, stdout=subprocess.PIPE)

# We immediately run the first subprocess and get the result
# Note that we encode the data, otherwise we'd get a TypeError
p1_out = p1.communicate(input=what_to_feed.encode())[0]

# Well the result includes an '\n' at the end, 
# if we want to get rid of it in a VERY hacky way
p1_out = p1_out.decode().strip().encode()

# We create the second subprocess, note that we need stdin=PIPE
p2 = subprocess.Popen(['wc', '-m'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)

# We run the second subprocess feeding it with the first subprocess' output.
# We decode the output to convert to a string
# We still have a '\n', so we strip that out
output = p2.communicate(input=p1_out)[0].decode().strip()

这与此处的响应有些不同,您可以直接通过管道传输两个进程,而无需直接在 Python 中添加数据。

希望能帮助别人。

于 2019-12-08T22:23:18.900 回答
-2

您可以为 的stdin参数提供一个类似文件的对象subprocess.call()

该对象的文档适用Popen于此处。

要捕获输出,您应该改用subprocess.check_output(),它采用类似的参数。从文档中:

>>> subprocess.check_output(
...     "ls non_existent_file; exit 0",
...     stderr=subprocess.STDOUT,
...     shell=True)
'ls: non_existent_file: No such file or directory\n'
于 2011-12-12T13:48:59.900 回答