Python Pexpect回传命令和命令输出

2024-05-23 17:42:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我想连接到运行CLISH的服务器,在给出密码提示时提供密码(works),然后发出“shell”命令将我转到bash shell(works),然后测试文件是否存在,如果文件丢失则打印“true”,如果文件存在则打印“空字符串”(works)。

不起作用的是,我正在用“pexpect.before”读回我的脚本。我已经让pexpect期望得到基本的提示,然后给我“before”,我的理解是:“在匹配提示之前,您在屏幕上看到了什么?”我不仅得到了预期的“true”或“”(空字符串),还得到了我发出的用于测试文件存在性的命令。

例如,如果我直接SSH到机器,我将执行以下操作: 密码:

我的脚本返回的是命令和响应(问题和答案):

[ ! -f '/etc/logrotate.d/nginx' ] && echo 'true'

true

我只想得到答案,要么是“真的”,要么是。我得到了回应,但我也得到了发出的命令。

你能看出我做错了什么吗?请帮忙!我想继续使用pexpect,因为我对SSH会话有更复杂的“期望”。

这是我的代码:

def connection(cmd):
    msg_newkey = 'Are you sure you want to continue connecting'
    p=pexpect.spawn('ssh '+user+'@'+host)
    msg = '''Super long welcome message containing your canned warning message, etc.
          myuser@'''+myhost+''''s password:'''
    i=p.expect([msg_newkey,'password:',pexpect.EOF])
    if i==0:
        print "Accepting key."
        p.sendline('yes')
        i=p.expect([msg_newkey,'password:',pexpect.EOF])
    if i==1:
        print "Supplying password"
        p.sendline(passwd)
    elif i==2:
        if "No route to host" in p.before:
            return p.before
        else:
            pass    # Add no functionality, just saying "move on"
    i=p.expect(['MYCLISHPROMPT>',pexpect.EOF])
    if i==0:
            p.sendline('shell')
    i=p.expect(['.*@.*\$',pexpect.EOF])
    if i==0:
            p.sendline(cmd)
    p.expect(['myuser@myhost:',pexpect.EOF])
    return p.before

 def look_for_file(step, filename, mytest):
    cmd = "[ ! -f '"+filename+"' ] && echo '"+mytest+"'"
    results = connection(cmd)
    print "***"+str(result)+"***"
    assert (results == '' or results == 0), "Step failed!  Error: "+results

Tags: 文件命令cmdtrue密码ifmsgpassword
2条回答

我在需要将pexpect日志设置为sys.stdout的脚本中遇到此问题,如下所示:

    shell = pexpect.spawn("bash")
    shell.logfile = sys.stdout

我尝试了这里其他答案中建议的解决方案,但没有成功;传递给sendline()的字符串被打印到日志中,不管是什么。

但是,我发现暂时禁用pexpect日志似乎可以解决问题:

    shell.logfile = None
    shell.sendline(MY_STRING)
    shell.logfile = sys.stdout

有一个函数setecho(它是一对getecho)应该控制是否回显sendline中的字符串。

setecho(self, state)
    This sets the terminal echo mode on or off. Note that anything the
    child sent before the echo will be lost, so you should be sure that
    your input buffer is empty before you call setecho().

但是,这显然不是在所有情况下都有效,您需要使用一些解决方法。一种是像this answer那样在bash中转换echo。另一种方法是使用pexpect的函数readlinereadlines来读取输出并丢弃与给定命令相呼应的第一行。

相关问题 更多 >