输出不同步之前的Python / Pexpect

时间:2012-06-06 18:31:36

标签: python pexpect

我正在使用Python / Pexpect为多个路由器生成SSH会话。该代码适用于一个路由器,但session.before的输出将与某些路由器不同步,以便它将返回先前发送线的输出。发送空行(sendline())时尤其如此。有人有任何想法吗?任何见解都会非常感激。

以下是我所看到的示例:

ssh_session.sendline('sh version')
while (iresult==2):
    iresult = ssh_session.expect(['>','#','--More--'],timeout=SESSION_TIMEOUT)
    debug_print("execute_1 " + str(iresult))
    debug_print("execute_bef " + ssh_session.before)
    debug_print("execute_af " + ssh_session.after)

    thisoutput = ssh_session.before
    output += thisoutput

    if(iresult==2):
        debug_print("exec MORE")
        ssh_session.send(" ")
    else:
        debug_print("exec: end loop")

for cmd in config_commands:
    debug_print("------------------------------------------------\n")
    debug_print ("running command " + cmd.strip() + "\n")
    iresult=2
    ssh_session.sendline(cmd.strip())
    while (iresult==2):
        iresult = ssh_session.expect([prompt+">",prompt+"#"," --More-- "],timeout=SESSION_TIMEOUT)
        thisoutput = ssh_session.before
        debug_print("execute_1 " + str(iresult))
        debug_print("execute_af " + ssh_session.after)
        debug_print("execute_bef " + thisoutput)
        thisoutput = ssh_session.before
        output += thisoutput

        if(iresult==2):
           debug_print("exec MORE")
           ssh_session.send(" ")
        else:
           debug_print("exec: end loop")


I get this:

logged in
exec: sh version
execute_1 1
execute_bef 
R9
execute_af #
exec: end loop
------------------------------------------------

running command config t

execute_1 1
execute_af #
execute_bef sh version
Cisco IOS Software, 1841 Software (C1841-IPBASEK9-M), Version 15.1(4)M4, RELEASE SOFTWARE (fc1)
Technical Support: http://www.cisco.com/techsupport...

4 个答案:

答案 0 :(得分:1)

我之前遇到过pexpect(我正在努力记住我是如何解决这个问题的)。

您可以通过发送返回来重新与终端会话同步,然后期望循环中的提示。当期望超时时,您就知道自己已经同步。

根本原因可能是你要么:

  • 在没有匹配期望的情况下调用发送(因为您不关心输出)

  • 运行一个命令,该命令产生输出,但期望在该输出的中间有一个模式,然后不在输出结束时的下一个提示。处理此问题的一种方法是将您的期望模式更改为“(。+)PROMPT” - 这将一直持续到下一个提示并捕获发送的命令的所有输出(您可以在下一步中解析)。

答案 1 :(得分:0)

我遇到了类似的问题。我试着等待命令在屏幕上打印并发送输入。

我想执行say命令'cmd',然后执行:

    session.send(cmd)
    index = session.expect([cmd, pexpect.TIMEOUT], 1)
    session.send('\n')
    index = session.expect([whatever you expect])

为我工作。

答案 2 :(得分:0)

我不确定这是你问题的根源,但值得一试。

我遇到的一件事是,当你产生一个以shell开头或以shell为基础的会话时,你必须处理TERM类型的怪癖(vt220,color-xterm等)。您将看到用于移动光标或更改颜色的字符。问题几乎可以保证出现提示;你正在寻找识别提示的字符串出现两次,因为颜色变化的处理方式(发送提示,然后代码退格,改变颜色,然后再次发送提示......但是期望看到两个实例)提示)。

这里有处理这个的东西,保证是丑陋的,hacky的,不是非常Pythonic的,功能性的:

import pexpect

# wait_for_prompt: handle terminal prompt craziness
#   returns either the pexpect.before contents that occurred before the 
#   first sighting of the prompt, or returns False if we had a timeout
#
def wait_for_prompt(session, wait_for_this, wait_timeout=30):
    status = session.expect([wait_for_this, pexpect.TIMEOUT, pexpect.EOF], timeout=wait_timeout)
    if status != 0:
        print 'ERROR : timeout waiting for "' + wait_for_this + '"'
        return False
    before = session.before # this is what we will want to return
    # now look for and handle any additional sightings of the prompt
    while True:
        try:
            session.expect(wait_for_this, timeout=0.1)
        except:
            # we expect a timeout here. All is normal. Move along, Citizen.
            break # get out of the while loop
        return before

s = pexpect.spawn('ssh me@myserver.local')
s.expect('password') # yes, we assume that the SSH key is already there
                     # and that we will successfully connect. I'm bad.
s.sendline('mypasswordisverysecure') # Also assuming the right password
prompt = 'me$'
wait_for_prompt(s, prompt)
s.sendline('df -h') # how full are my disks?
results = wait_for_prompt(s, prompt)
if results:
    print results
    sys.exit(0)
else:
    print 'Misery. You lose.'
    sys.exit(1)

答案 3 :(得分:0)

我知道这是一个老线程,但我在网上找不到这个,我刚刚完成了自己的快速和肮脏的解决方法。我也使用pexpect来浏览网络设备列表并记录统计信息等等,而且我的pexpect.spawn.before有时也会失去同步。出于某种原因,这种情况经常发生在更快,更现代的设备上。

我的解决方案是在每个命令之间写一个空的回车符,并检查.before变量的len()。如果它太小,则意味着它只捕获了提示符,这意味着它必须至少是实际ssh会话后面的一个命令。如果是这种情况,程序会发送另一个空行,将我想要的实际数据移动到.before变量中:

def new_line(this, iteration):
    if iteration > 4:
        return data
    else:
        iteration+=1
        this.expect(":")
        this.sendline(" \r")
        data = this.before
        if len(data) < 50:
        # The numer 50 was chosen because it should be longer than just the hostname and prompt of the device, but shorter than any actual output
            data = new_line(this, iteration)
        return data

def login(hostname):
    this = pexpect.spawn("ssh %s" % hostname)
    stop = this.expect([pexpect.TIMEOUT,pexpect.EOF,":"], timeout=20)
    if stop == 2:
        try:
            this.sendline("\r")
            this.expect(":")
            this.sendline("show version\r")
            version = new_line(this,0)
            this.expect(":")
            this.sendline("quit\r")
            return version
        except:
            print 'failed to execute commands'
            this.kill(0)
    else:
        print 'failed to login'
        this.kill(0)

我通过递归命令完成此操作,该命令将调用自身直到.before变量最终捕获命令的输出,或直到它自己调用5次,在这种情况下它只是放弃。