如何与 Python 中的另一个程序交互？

Question

我想编写一个运行另一个程序的 Python 脚本，读取另一个程序的输出并对其进行操作。问题是该程序提示输入密码，而我不知道如何自动提供密码。 （出于此脚本的目的，密码是否以纯文本形式存储在脚本本身中并不重要。）我想做的是：

os.system('echo someinput | /var/local/bin/someprogram') 

这会导致某些程序给我不需要的密码提示，并且不给我程序的输出作为返回值。不幸的是，该程序没有办法绕过这个提示。

不幸的是，对于如何解决这个问题，我也有一些限制。首先，我坚持使用 Python 2.3（所以我无法使用 subprocess 模块）。其次，我无法安装任何新模块（所以没有预期）。幸运的是，它不必特别便携，因此仅 Linux 的解决方案就可以了。

我一直在试图找出 pty 模块，因为它看起来可以提供我需要的东西，但是在花了几个小时与它搏斗之后，我只是不知道如何让它按照我需要的方式工作。

Answer 1

我在基于终端的进程间通信方面遇到了一些类似的问题，这些问题似乎无法使用 popen （等）解决。我最终通过阅读 pexpect 的源代码学习了如何使用 pty ，其中包含如何（以及原因的注释）让 pty 跳过必要的环节的示例。

当然，根据您的需求，您也可以使用 pexpect！

这是我在自己的项目中使用的主要内容。请注意，我并没有检查子进程是否终止；而是检查子进程是否终止。该脚本旨在作为管理长时间运行的 Java 进程的守护进程运行，因此我无需处理状态代码。不过，希望这能为您提供大部分所需的东西。

import os
import pty
import select
import termios

child_pid, child_fd = pty.fork()

if not child_pid: # child process
    os.execv("/path/to/command", ["command", "arg1", "arg2"])

# disable echo
attr = termios.tcgetattr(child_fd)
attr[3] = attr[3] & ~termios.ECHO
termios.tcsetattr(child_fd, termios.TCSANOW, attr)

while True:
    # check whether child terminal has output to read
    ready, _, _ = select.select([child_fd], [], [])

    if child_fd in ready:
        output = []

        try:
            while True:
                s = os.read(child_fd, 1)

                # EOF or EOL
                if not s or s == "\n":
                    break

                # don't store carriage returns (no universal line endings)
                if not s == "\r":
                    output.append(s)
        except OSError: # this signals EOF on some platforms
            pass

        if output.find("Enter password:") > -1:
            os.write(child_fd, "password")

Answer 2

您可以使用 os.popen，它已移至 subprocess 在 2.6 中，但应该仍然存在于 2.3 的 os 模块中。将模式设置为'w'并使用close()获取返回值。

Answer 3

expect 还有另一个 Python 实现，可以轻松支持 pty。有一个封装 ssh 的库可以使用它发送密码。它是 Pycopia 中的 sshlib 模块。那里有一个处理密码的登录方法。它还使用 Pycopia 中的 Expect 和 process (proctools) 模块。我最初是为 Python 2.2 编写的，因此它们可能适合您。但是，它们可能不会，因为随着时间的推移，我自由地使用了 Python 的其他新功能

，这些功能可能已经悄然出现。这些模块的主要目标是使处理子流程（如您所描述的）变得更容易、更“Pythonic”。