在一个脚本中使用Python的子进程和Popen来运行另一个需要用户交互的Python脚本(通过raw_input)

时间:2012-10-19 18:19:32

标签: python automation subprocess popen

我遇到的问题如下,我将使用简单的例子来说明它。我编写了一个需要用户交互的python脚本,特别是它使用raw_input()函数来获取用户的输入。下面的代码只是要求用户连续输入两个数字(在每个数字之间输入),然后返回答案(惊讶,惊讶,它被称为'sum_two_numbers.py')。乏味!

#! /usr/bin/python

#  -------------------
#  sum_two_numbers.py
#  -------------------
#  This script asks the user for two numbers and returns the sum!

a = float(raw_input("Enter the first number:"))
b = float(raw_input("Enter the second number:"))

print a+b

现在,我想编写一个单独的python脚本来执行上面的脚本并将两个必要的数字“提供”给它。因此我称这个脚本为'feeder.py'。我尝试使用Python的“subprocess”模块编写此脚本,特别是使用“Popen”类及其关联的“通信”方法。下面是试图输入数字'5'和'4'的脚本。

#! /usr/bin/python

#  ----------
#  feeder.py
#  ----------
import subprocess

child = subprocess.Popen("./sum_two_numbers.py",stdin=subprocess.PIPE)

child.communicate("5")
child.communicate("4")

此代码不起作用,并在执行时返回错误:

$ ./feeder.py
Enter the first number:Enter the second number:Traceback (most recent call last):
  File "./sum_two_numbers.py", line 6, in <module>
    b = float(raw_input("Enter the second number:"))
EOFError: EOF when reading a line
Traceback (most recent call last):
  File "./feeder.py", line 8, in <module>
    child.communicate("4")
  File "/usr/lib/python2.7/subprocess.py", line 740, in communicate
    self.stdin.write(input)
ValueError: I/O operation on closed file

我不知道怎么写'feeder.py'以便它能做我想做的事,这些错误一直阻碍着我。我怀疑由于文档中的以下注释而出现此错误:

  

Popen.communicate(输入=无)

     

与流程交互:将数据发送到stdin。从stdout和stderr读取数据,直到    达到文件结尾。等待进程终止。

我不确定这句话该怎么做,以及它如何帮助我...

任何人都可以帮助我使上述脚本正常工作,即如何正确使用子进程和Popen ......或者只是如何编写“进纸器”脚本 - 用任何(不太模糊)的语言!我已经尝试过Pexpect,Expect,但遇到了一些问题,比如它没有输出子代码的输入请求,我只是一般不知道它在做什么。

1 个答案:

答案 0 :(得分:9)

您只能拨打communicate一次。因此,您需要立即传递所有输入,即child.communicate("1\n1\n")。或者你可以写stdin:

child = subprocess.Popen("./test.py", stdin=subprocess.PIPE)         

child.stdin.write("1\n")                                                       
child.stdin.write("1\n")