在正在运行的程序中运行 python 脚本

Question

我正在运行一个 python 脚本，它启动一个名为 ./abc 的可执行文件。这个可执行文件进入程序内部并等待如下命令：

$./abc
abc >                      \\waits for a command here.

我想做的是输入几个命令，例如：

$./abc
abc > read_blif alu.blif
abc > resyn2

我目前的情况如下：

import os
from array import *

os.system('./abc')
for file in os.listdir("ccts/"):
    print 'read_blif ' + file + '\n'
    print 'resyn2\n'
    print 'print_stats\n'
    print 'if -K 6\n'
    print 'print_stats\n'
    print 'write_blif ' + file.split('.')[0] + 'mapped.blif\n'

但是，这将执行以下操作：

abc >                 \\stays idle and waits until I ^C and then it prints
read ...blif
resyn2
...

它只打印到终端。我如何让它在程序中执行这个并等到它看到下一个 abc > 运行下一个命令。 谢谢

Answer 1

我使用subprocess 做了类似的事情。

import subprocess

cmd = './command_to_execute'
pro = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
result = pro.stdout.read()

这将执行cmd指定的命令，然后将结果读入result。在结果分配之后执行任何操作之前，它将等待将结果打印到控制台。我相信这可能是您想要的，尽管您的描述有点模糊。

Answer 2

如果您想通过管道将命令输入可执行文件，最简单的方法是使用subprocess 模块。您可以输入可执行文件的标准输入并使用Popen.communicate 获取其输出。

import subprocess

for f in os.listdir('ccts/'):
    p = subprocess.Popen('./abc', stdin=subprocess.PIPE,
                         stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = p.communicate('read_blif ' + f + '\n'
                                   'resyn2\n'
                                   'print_stats\n'
                                   'if -K 6\n'
                                   'print_stats\n'
                                   'write_blif ' + f.split('.')[0] + 'mapped.blif\n')
    # stdout will be what the program outputs and stderr will be any errors it outputs.

然而，这将每次关闭子进程的stdin，但这是可靠通信而不会死锁的唯一方法。根据https://stackoverflow.com/a/28616322/5754656，您应该将pexpect 用于“类交互式会话”程序。假设您可以让不同的子进程运行程序，则可以通过拥有多个子进程来避免这种情况。

我假设您只需要print_stats 的stdouts，所以您可以这样做（例如，您可能想要处理错误）：

import subprocess

def helper_open(cmd='./abc'):
    return subprocess.Popen(cmd, stdin=subprocess.PIPE,
                            stdout=subprocess.PIPE, stderr=subprocess.PIPE)

for f in in os.listdir('cts/'):
    helper_open().communicate('read_blif' + f + '\n'
                              'resyn2\n')
    stats, _ = helper_open().communicate('print_stats\n')
    # Do stuff with stats
    stats, _ = helper_open().communicate('if -K 6\nprint_stats\n')
    # Do more stuff with other stats
    helper_open().communicate('write_blif ' + f.split('.')[0] + 'mapped.blif\n')

Answer 3

您需要使用subprocess 库生成一个新进程，并创建两个管道，一个用于stdin，一个用于stdout。使用此管道（在 python 中表示为文件），您可以与您的进程进行通信 这是一个例子：

 import subprocess

cmd = './full/path/to/your/abc/executable'
pro = subprocess.Popen(cmd, stdout=subprocess.PIPE, 
                       stdin=subprocess.PIPE)

pro.stdin.write("read_blif alu.blif \n")
pro.stdout.read(3)

您可以使用pro.communicate，但我假设您需要为输入的每个命令获取输出。类似：

abc > command1
ouput1
abc > command2
output2 -part1
output2 -part2
output2 -part3

这样我觉得PIPE的方法更有用。

使用 python 中的dir 函数查找有关pro 对象的更多信息以及dir(proc) 可用的方法和属性。不要忘记内置的help 将显示文档字符串help(pro) 或help(pro.stdin)。

您在运行os.system 时犯了一个错误，这将在后台运行您的程序，您将无法控制它。也许你想看看什么输入/输出流。 更多阅读可以here.

Answer 4

您可能正在寻找 pexpect 模块。这是 pexpect 文档中的基本示例

# This connects to the openbsd ftp site and
# downloads the recursive directory listing.
import pexpect
child = pexpect.spawn('ftp ftp.openbsd.org')
child.expect('Name .*: ')
child.sendline('anonymous')
child.expect('Password:')
child.sendline('noah@example.com')
child.expect('ftp> ')
child.sendline('lcd /tmp')

如果您的操作系统与 pexpect 兼容，我认为它与 abc > 的工作方式相同。