如何从 pexpect.spawn() 收集所有输出

如何解决如何从 pexpect.spawn() 收集所有输出

目前，c.read() 函数仅从我正在调用的程序的输出中返回值，并且不收集交互部分期间发生的情况。（换句话说：等待用户名和密码）。由于该部分可能会发生错误，因此我需要在调用 cmd 后收集所有输出。

使用运行命令工作正常。我正在尝试转换为使用 expect.spawn()，因为它看起来非常接近于 ssh 版本的 expect。当前 run() 方法工作正常，因为它在本地工作。我找不到可以返回整个输出的 getOutput() 类型的函数。

import pexpect
class Expect( ):
    def Do( self,cmd,program: list = [],timeout: int = 20 ):
        # result = run( cmd,events=program,timeout=timeout).decode() 

        result = ''
        c = pexpect.spawn( cmd,encoding='utf-8',timeout=timeout )
        for curItem in program:
            # print( 'wait: ' + curItem[0] + ' resp: ' + curItem[1])
            c.expect( curItem[0] )
            c.sendline( curItem[1] )
            # result += c.read()
    
        result += c.read()

        print ( 'res: ' + str(result) )
        return result


cmd  = 'fmsadmin list files -s'
prog = [('username \\(.+\\):','yourUN\n'),('password:','yourPW\n')]
res  = Expect().Do( cmd,prog )    # Returns everything
# use results to verify functionality

结果

ID File                  Clients Size   Status Enabled Extended Privileges                                                                   Encrypted 
1  FMServer_Sample.fmp12 0       905216 Normal fmapp fmxml fmphp fmwebdirect                                                                 No

我期待的时候

username (yourUN):yourUN
password:
ID File                  Clients Size   Status Enabled Extended Privileges                                                                   Encrypted 
1  FMServer_Sample.fmp12 0       905216 Normal fmapp fmxml fmphp fmwebdirect                                                                 No

更新

我在 c.sendline( curItem[1] ) 下添加了 result += c.before + c.after 现在返回： c.sendline( curItem[1] )

username (yourUN): yourUN

password:
ID File                  Clients Size   Status Enabled Extended Privileges                                                                   Encrypted 
1  FMServer_Sample.fmp12 0       905216 Normal fmapp fmxml fmphp fmwebdirect                                                                 No

不幸的是，还有两个问题。 1 输出中还有其他值，例如返回值，如果我提供无效密码，我会在没有应用程序返回值的情况下返回调试详细信息。我得到：

before (last 100 chars): '\r\n/usr/local/bin/fmsadmin: Permission denied,please try again.\r\nusername (yourUN):'
after: <class 'pexpect.exceptions.TIMEOUT'>
match: None
match_index: None
etc

简而言之，我正在寻找 CLI 的完整返回值，就像我手工完成时返回的值一样。这就是 run() 的作用。

解决方法

Pexpect，就像原来的 TCL Expect，获取会话输入和输出的正确部分有点棘手。挑战在于格式化您的搜索模式以匹配它应该匹配的内容，并且仅此而已。要使其正常工作，您所要做的就是使用 before、after 和 match 属性解析 returned object。

也就是说，Pexpect 获取终端显示的所有内容，包括命令和输出。幸运的是，密码没有显示出来，很遗憾，Pexpect 看不到它们。

from pexpect import spawn,EOF  # Better import only needed methods
process = spawn('bash',encoding='utf-8')
process.expect('$')             # Expects terminal prompt

process.sendline('read -p "username: " var')
process.expect('\nusername: ')  # Gets command,newline and username prompt
process.sendline('me')
process.expect('[^\n]*\n.*$')   # Gets typed string,newlind and terminal prompt
print(process.match.group(0))   # Prints whole string matched
me                              # User typed
me@myhost ~ $                   # Terminal prompt

process.sendline('read -sp "password: " var')   # Note the '-s' for 'silent'
process.expect('\npassword: ')
process.sendline('my_passwd')
process.expect('.*$')           # Trying to match a newline just doesn't work
print(process.match.group(0))
me@myhost ~ $                   # Only terminal prompt

process.sendline('exit')
process.expect(EOF)

所以你应该知道你想解决什么样的问题。为什么要收集所有发送和返回的内容，包括密码？

顺便说一句，您可能已经知道代码中的普通密码存在问题。我的建议：getpass：

from getpass import getpass
password = getpass()
<Then use the password>

顺便说一句 2，以防万一，把 docs 放在身边。

好的，我想通了。不太预期，但它是有道理的。几个变化：

在 for 循环结束时添加结果 += c.before + c.after。这捕获帐户和密码的输出以及 CLI 的任何错误应用程序。
将 spawn、expect、send 放入 try 接受块
将结果 += c.read() 移动到 Try before except （安静例外）
最后在异常期间调用 result += c.buffer。

我可能会在下一个版本中删除本地部分（一旦我弄清楚远程情况）

from pexpect import *

class Expect( ):
    def Do( self,cmd,program: list = [],timeout: int = 20 ):
        local   = False
        result  = ''
        if local:
            result = runu( cmd,events=program,timeout=timeout ) 
        else:
            c = ''
            try:
                c = spawnu( cmd,timeout=timeout )
                for curItem in program:
                    c.expect( curItem[0] )
                    c.sendline( curItem[1] )
                    result += c.before + c.after
                result += c.read()
            except:
                result += c.buffer
        return result

cmd  = 'fmsadmin list files -s'
prog = [('username \\(.+\\):','yourUN\n'),('password:','yourPW\n')]
res  = Expect().Do( cmd,prog,5 )
print ( '-' * 50 + '\n' + str(res))

如何从 pexpect.spawn() 收集所有输出

如何解决如何从 pexpect.spawn() 收集所有输出

解决方法

相关推荐