如何使用 Python 的子进程控制交互式 ssh 会话

Question

我正在尝试从 python 控制 interactive ssh 会话 - 所以不运行 ssh user@host command（有很多答案）但 ssh user@host 然后写入/读取该 ssh来自 Python 的会话。

我尝试过的总是以：

import subprocess as sp
import sys

ssh_cmd = ["ssh", "-tt", "whoever@myhost"]
proc = sp.Popen(ssh_cmd, stdin=sp.PIPE, stdout=sp.PIPE, stderr=sp.PIPE)
while True:
    stdin = raw_input('enter command: ')
    proc.stdin.write(b'%s\n' % stdin)

问题是如何读取proc.stdout：

• proc.stdout.readlines() 不起作用，因为它从不返回 EOF，所以它会阻塞
• proc.stdout.readline() 在有一行数据准备好的情况下有效，但 proc.stdout 在为空时不返回 '' 因此无法读取它，如果没有数据则阻塞
• select 和 poll 总是说 proc.stdout 已经“准备好”阅读，无论 proc.stdout.readline() 是否真的有效。

有什么建议吗？

编辑：这里的代码是 python2.7，但 python3 的解决方案也可以。 编辑：在这里使用proc.communicate() 不是一个选项，因为它等待进程完成，我想继​​续读/写它。

Answer 1

嗯，Marcin Orlowski 的建议肯定是不费吹灰之力，尽管我觉得我没有了解原版的问题所在！

from paramiko.client import SSHClient

client = SSHClient()
client.load_system_host_keys()
client.connect("myhost", username="whoever")
while True:
    cmd = raw_input('enter command: ')
    stdin, stdout, stderr = client.exec_command(cmd)
    print ''.join(stdout.readlines())

Answer 2

要回答代码“原件有什么问题？”，我想说可能有几件事：

1. 如果您运行的是 Python 3，那么要将raw_input 返回的字符串转换为字节，您需要使用：stdin.encode() 对其进行编码。 将使用系统默认编码，通常为UTF-8。您使用的表达式b'%s\n' % stdin（其中stdin 是str）应该抛出异常，除非您运行的是Python 2，在这种情况下str 和byte 之间没有真正的区别。我知道的 Python 3 中唯一的方法 raw_input 来自 code.InteractiveConsole 类。我没有看到导入语句，因此这表明您正在运行 Python 2，所以我不确定您为什么要尝试从字符串开始转换为字节。

2. 真正的问题是你需要调用proc.communicate()，这是将你对proc.stdin所做的所有写入的结果发送到实际进程并检索进程对stdout 和stderr 的输出。在你这样做之前，什么都不会发生。最重要的是subprocess.Popen 不适合用于交互式对话框。它应该用于仅向进程发送一些固定输入，调用communicate()，然后检索结果输出。

使用subprocess的演示

import subprocess as sp
import time

ssh_cmd = ["grep", "age"]
proc = sp.Popen(ssh_cmd, stdin=sp.PIPE)
proc.stdin.write(b'age1\n')
proc.stdin.write(b'age2\n')
proc.stdin.write(b'age3\n')
time.sleep(5) # nothing happens for these 5 seconds
proc.communicate() # now the output will start coming out


# second demo use pipes for output:
ssh_cmd = ["grep", "age"]
proc = sp.Popen(ssh_cmd, stdin=sp.PIPE, stdout=sp.PIPE, stderr=sp.PIPE)
proc.stdin.write(b'age1\n')
proc.stdin.write(b'age2\n')
proc.stdin.write(b'age3\n')
stdout, stderr = proc.communicate() # retrieve outputs form the process
#print(stdout.decode()) # convert from bytes to string for Python 3
print stdout # Python 2

打印（5 秒后）：

age1
age2
age3
age1
age2
age3