Python ssh 连接到交互式程序(例如，python 解释器)

Question

我想使用 python 通过 ssh 连接到远程机器，在远程启动交互式可执行文件，然后与所述可执行文件交互。

换句话说，我想模仿当你做这样的事情时会发生什么:

ssh me@host -t python2

(python2 就是 python，但为了避免混淆，我给它取了不同的名字)。

当我执行上面的命令行时，我正在“无缝地”与 Remote 上的 python2 进行交互。

我似乎无法在 python 中执行此操作。我试过使用 paramiko 和 subprocess，此时两者都有同样的问题。例如，在子进程中:

ssh = subprocess.Popen("ssh me@host -t python2",
                       shell=True,
                       stdout=subprocess.PIPE,
                       stderr=subprocess.PIPE)
result = ssh.stdout.readlines()
if result == []:
    error = ssh.stderr.readlines()
    print >> sys.stderr, "ERROR: %s" % error
else:
    print result

当我这样做时，我在命令行上看不到任何东西，但我可以通过 ps 看到 python2 确实在运行，我只是无法按照我喜欢的方式与它交互。我看不到 Python 的输出。

我希望能够从主机 python 脚本提交命令，例如 ssh.execute("1+1")。这基本上会告诉 python2 计算 1+1。然后我想将输出读回主机 python 脚本。

Answer 1

问题

您的问题是 ① ssh.stdout.readlines() 正在读取所有已缓冲到您访问它的点的现有输入，按行结尾拆分(\r 或 \n)。

但是因为你打开的是一个提示符，所以 ② 可能只有一行，也就是没有行尾的 >>>。

处理流

您需要做的是创建一个 while 循环来读取输出:

def handle_output(out):
    while True:
        out_line = out.readline()
        print("OUT: {}".format(out_line))

def handle_errput(err):
    while True:
        err_line = err.readline()
        print("ERR: {}".format(err_line))

这样就可以了:

handle_output(ssh.stdout)
handle_errput(ssh.stderr)

然后一旦远程 python 解释器打印出完整的一行，您就会在控制台上显示一些内容。但这只能解决①。

您仍然需要让远程进程至少输出一整行，唯一的方法就是实际向它发送数据!

所以你想添加另一个函数来执行此操作:

def handle_input(inp):
    while True:
        inp_line = sys.stdin.readline()
        inp.write(inp_line)

你会这样调用它:

handle_input(ssh.stdin)

这将解决②。

并行阅读

有线程

但是现在，您遇到了另一个问题，从逻辑上讲，如果您这样做:

handle_output(ssh.stdout)
handle_errput(ssh.stderr)
handle_input(ssh.stdin)

您的程序流永远不会到达handle_errput() 或handle_input()，因为它会陷入handle_output() 无限循环.

解决方案是并行执行三个循环:

out_thread = Thread(target=handle_output, args=(ssh.stdout,))
err_thread = Thread(target=handle_errput, args=(ssh.stderr,))
try:
    out_thread.start()
    err_thread.start()
    handle_input()
finally:
    out_thread.join()
    err_thread.join()

这样从外部进程读取将发生在另一个线程中，而写入将发生在当前线程中。您也可以启动三个线程(一个用于读取 stdout，一个用于读取 stderr，一个用于写入)并让主线程运行一个循环，该循环将执行“填充”并与这些线程通信(提示:使用 threading.队列在线程之间交换数据)。

带有事件库(仅限py3)

如果使用异步库，您可以避免使用多线程和线程间通信的痛苦 like asyncio offered with python 3 (对于 py2，您可以使用 tornado 或 greenlet)，这将为您提供一个非常好的抽象，而不是使用 select():

async def handle_output(out):
    while True:
        await out_line = out.readline()
        print("OUT: {}".format(out_line))

async def handle_errput(err):
    while True:
        await err_line = err.readline()
        print("ERR: {}".format(err_line))

async def handle_input(inp):
    while True:
        await inp_line = sys.stdin.readline()
        inp.write(inp_line)

asyncio.Task(handle_output(ssh.stdout))
asyncio.Task(handle_errput(ssh.stderr))
asyncio.Task(handle_input(ssh.stdin))

asyncio.get_event_loop().run_forever()

使用 select() 调用

在 python 2 中，最好和最简单(但远非最简单)的实现方法是使用 select() :

基本上选择调用看起来像:

while True:
  readable, writable, excepts = select.select(
           [ssh.stdout, ssh.stderr], # outputs to watch
           [sys.stdin],              # inputs to watch
           [ssh.stdout, ssh.stderr], # exceptions to watch
           1)                        # timeout
  for r in readable:
    print(r.read())
  for w in writable:
    p.stdin.write(w.read())

但你最好read a thorough documentation about select() .

从那里去哪里？

I want to be able to submit commands from the host python script, like ssh.execute("1+1"). Which would basically tell python2 to compute 1+1. I then want to read the output back into the host python script.

我给你的解决方案是帮助你处理从你的终端到远程 python 进程的透明通信。它公开了您实现上述目标所缺少的所有工具，除了一个，如何在 3 个循环之间进行通信？

最简单的方法是使用三个队列，两个处理从远程进程传入的结果，最后一个处理传出命令:

input_queue = queue.Queue()
output_queue = queue.Queue()
errput_queue = queue.Queue()

然后在你的主线程中你可以实现:

def remote_python_execute(command, output_queue, errput_queue):
    # send a command to the remote process
    input_queue.put(command)
    out, err = ([], [])
    # read the output and then the errput as long as
    # there's something to read
    while not output_queue.empty():
        out.append(output_queue.get())
    while not errput_queue.empty():
        err.append(errput_queue.get())
    return out, err

你的 handle_output 函数看起来像(你可以适应两者:

def handle_output(out, output_queue):
    while True:
        output_queue.put(out.readline())

只是给你一个想法......现在轮到你填补空白并选择最适合你需要的了😉

---

免责声明:此答案包含未经测试的代码，旨在为您指明如何实现您想要的目标。不要复制/粘贴它，而是阅读文档并理解您在做什么。也小心使用线程，作为提示，永远不要在两个线程中使用一个变量，否则你可能会被咬伤。很难。

---

HTH