与子进程进行多次通信（多次调用标准输出）

Question

[此线程][1] 上有一个与我类似的问题。

我想向我的子进程发送一个命令，解释响应，然后发送另一个命令。必须启动一个新的子进程来完成此任务似乎很遗憾，特别是如果 subprocess2 必须执行许多与 subprocess1 相同的任务（例如 ssh、打开 mysql）。

我尝试了以下操作：

subprocess1.stdin.write([my commands])
subprocess1.stdin.flush()
subprocess1.stout.read()

但是如果没有给 read() 指定字节的明确参数，程序就会卡住执行该指令，并且我无法为 read() 提供参数> 因为我无法猜测流中有多少字节可用。

我正在运行 WinXP，Py2.7.1

编辑

感谢 @regularfry 为我提供了实现我真正意图的最佳解决方案（阅读他的回复中的评论，因为它们与通过 SSH 隧道实现我的目标有关）。 （他/她的答案已被投票通过。）不过，为了以后前来寻求标题问题答案的任何观众的利益，我已经接受了@Mike Penningtion 的答案。

Answer 1

您的选择是：

1. 使用面向行的协议（并使用 readline() 而不是 read()），并确保发送的每个可能的行都是有效的消息；
2. 使用 read(1) 和解析器来告诉您何时阅读了完整的消息；或者
3. 将消息对象从子流程放入流中，然后在父流程中取消它们。这可以为您解决消息长度问题。

Answer 2

@JellicleCat，我正在跟进评论。我相信 wexpect 是 sage 的一部分...AFAIK，它没有单独打包，但你可以下载 我们期望在这里。

老实说，如果您要驱动程序化 ssh 会话，请使用 paramiko。它支持独立安装，具有良好的封装，并且应该在 Windows 上本机安装。

编辑

示例 paramiko 脚本以 cd 到目录，执行 ls 并退出...捕获所有结果...

import sys
sys.stderr = open('/dev/null')       # Silence silly warnings from paramiko
import paramiko as pm
sys.stderr = sys.__stderr__
import os

class AllowAllKeys(pm.MissingHostKeyPolicy):
    def missing_host_key(self, client, hostname, key):
        return

HOST = '127.0.0.1'
USER = ''
PASSWORD = ''

client = pm.SSHClient()
client.load_system_host_keys()
client.load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
client.set_missing_host_key_policy(AllowAllKeys())
client.connect(HOST, username=USER, password=PASSWORD)

channel = client.invoke_shell()
stdin = channel.makefile('wb')
stdout = channel.makefile('rb')

stdin.write('''
cd tmp
ls
exit
''')
print stdout.read()

stdout.close()
stdin.close()
client.close()

Answer 3

这种方法可以工作（我已经这样做了），但需要一些时间，并且它使用特定于 Unix 的调用。您必须放弃 subprocess 模块并基于 fork/exec 和 os.pipe() 推出您自己的等效模块。

使用 os.pipe() 创建子进程的 stdin/stdout 文件描述符（读和写）后，使用 fcntl.fcntl 函数将其置于非阻塞模式（O_NONBLOCK 选项常量）。

使用 select.select 函数轮询或等待文件描述符的可用性。为了避免死锁，您需要使用 select() 来确保写入不会阻塞，就像读取一样。即使如此，您在读写时也必须考虑 OSError 异常，并在收到 EAGAIN 错误时重试。 （即使在读/写之前使用 select，EAGAIN 也可能在非阻塞模式下发生；这是一个常见的内核错误，已被证明很难修复。）

如果您愿意在 Twisted 框架上实现，他们应该已经解决了这个问题为你;您所要做的就是编写一个 Process 子类。但我自己还没有尝试过。