2

我需要使用 Python 编写/自动化交互式终端客户端的脚本。客户端接受三个参数并运行如下：

>./myclient <arg1> <arg2> <arg3>
Welcome...
blah...
blah..
[user input]
some more blah... blah... for the input entered
blah.. 
blah..
[basically it accepts input and puts the output in the console until the user types 'quit']

现在我需要在 Python 中自动执行此操作，并将控制台输出保存在文件中。

4

2 回答 2

7

你可以看看http://docs.python.org/library/cmd.html。

示例代码：

import cmd
import sys

class Prompt(cmd.Cmd):
    def __init__(self, stufflist=[]):
        cmd.Cmd.__init__(self)
        self.prompt = '>>> '
        self.stufflist = stufflist
        print "Hello, I am your new commandline prompt! 'help' yourself!"

    def do_quit(self, arg):
        sys.exit(0)

    def do_print_stuff(self, arg):
        for s in self.stufflist:
            print s

p = Prompt(sys.argv[1:])
p.cmdloop()

示例测试：

$ python cmdtest.py foo bar
Hello, I am your new commandline prompt! 'help' yourself!
>>> help

Undocumented commands:
======================
help  print_stuff  quit

>>> print_stuff
foo
bar
>>> quit

为了将输出保存到文件中，您可以将通常发送到 stdout 的内容也写入文件，例如使用此类：

class Tee(object):
    def __init__(self, out1, out2):
        self.out1 = out1
        self.out2 = out2

    def write(self, s):
        self.out1.write(s)
        self.out2.write(s)

    def flush(self):
        self.out1.flush()
        self.out2.flush()

你可以像这样使用它：

with open('cmdtest.out', 'w') as f:
    # write stdout to file and stdout
    t = Tee(f, sys.stdout)
    sys.stdout = t

一个问题是通过stdin读入的命令没有出现在这个输出中，但我相信这可以很容易地解决。

于 2012-05-03T14:32:38.413 回答
7

您可能想要使用pexpect（它是古老的 expect 的纯 python 版本）。

import pexpect
proc = pexpect.spawn('./myclient <arg1> <arg2> <arg3>')
proc.logfile = the_logfile_you_want_to_use
proc.expect(['the string that tells you that myclient is waiting for input'])
proc.sendline('line you want to send to myclient')
proc.expect(['another line you want to wait for'])
proc.sendline('quit') # for myclient to quit
proc.expect([pexpect.EOF])

这样的事情应该足以解决您的问题。pexpect 有更多功能，因此请阅读文档以获取更高级的用例。

于 2012-05-03T15:28:30.387 回答