Question

我有一个3d party python脚本，它从命令行获取输入。此脚本（input.py）中的相关代码如下所示：

import sys

def chooseinput():
    valid_inputs = ('a', 'b')
    inp = raw_input('Enter choice (%s): ' % "/".join(valid_inputs))
    if inp not in valid_inputs:
        sys.stderr.write("Unsupported input %s\n" % inp)
        return
    print 'You chose ' + '\'' + inp + '\''
    return inp

if __name__ == "__main__":
    chooseinput()
    # do something with the input...
    chooseinput()
    # do something with the input...

我正在尝试编写另一个python脚本（harness.py）来生成上述脚本的输入。

import subprocess

def harness():
    p = subprocess.Popen(['python', 'input.py'], stdin=subprocess.PIPE)
    p.stdin.write('a')
    p.stdin.write('b')

if __name__ == '__main__':
    harness()

从命令行，我运行：

$ python harness.py
Enter choice (a/b): Enter choice (a/b): Traceback (most recent call last):
  File "input.py", line 13, in <module>
    chooseinput()
  File "input.py", line 5, in chooseinput
    inp = raw_input('Enter choice (%s): ' % "/".join(valid_inputs))
EOFError: EOF when reading a line

如果我在第一个脚本中只有一个输入，那么我可以通过删除第二个写调用来使第二个脚本工作。如果第一个脚本需要多个输入，那么我会收到上述错误。

Answer 1

尝试：

p.stdin.write('a\n')
p.stdin.write('b\n')

Answer 2

linuts的答案在您的简单示例中运行良好，但为了未来读者的利益，我强烈建议不要使用此方法在Python脚本之间进行通信。

这种方法是很少有其他选项可用时的回归。 Pexpect，祝福它的内心，可能确实是一个很好的程序，但它只是在一个悲惨的界面技术上表达了一个快乐的面孔。像这样的命令行控制通常取决于时间，很快就会变得繁琐且容易出错。只有在没有其他选择时才使用它。

Python为脚本编写带来了许多更强大的方法。除非您无法访问脚本内部（使用Python，您几乎总是这样做），否则应编写harness.py脚本以将3方脚本作为库导入，并通过直接调用其方法/函数以编程方式控制它。

您目前的困境可能不允许，但使用Python脚本，命令行通信应该是最后的选择，而不是第一个。

Answer 3

您应该查看Pexpect。

Pexpect是一个纯Python模块，用于生成子应用程序;控制他们;并回应其产出中的预期模式。子应用程序可以是任何可执行文件（例如，在您的情况下，另一个python脚本）。它的工作方式类似于unix工具“expect”。

我如何在两个python脚本之间进行通信？

3 个答案: