如何使用' sync_original_prompt'来自pxssh模块

时间:2017-01-12 10:58:27

标签: python unit-testing pexpect pxssh

>>> from pexpect import pxssh
>>> s=pxssh.pxssh()
>>> s.login('IP','USER','PASSWORD',auto_prompt_reset=True)
True
>>> s.sendline('echo Test');
10
>>> s.prompt()
True
>>> s.after
'[PEXPECT]# '
>>> s.PROMPT='BTEQ -- Enter your SQL request or BTEQ command:'
>>> s.sendline('bteq .logon dbc,dbc')
20
>>> s.prompt()
True
>>> s.after
'BTEQ -- Enter your SQL request or BTEQ command:'
>>> s.sendline('quit;')
6
>>> s.prompt()
False
>>> s.PROMPT='[PEXPECT]# '
>>> s.prompt()
False
>>> s.sync_original_prompt()
True
>>> s.prompt()
False

根据我在s.sync_original_prompt()之后的理解,PROMPT变量应重置为初始提示,此处为[PEXPECT]#,而s.prompt()应导致True。但它并没有。有人可以告诉我如何将PROMPT变量重置为原始提示吗?以及如何使用sync_original_prompt属性?

感谢。

1 个答案:

答案 0 :(得分:0)

  • 初始s.PROMPT(正则表达式字符串)实际上是'\\[PEXPECT\\][\\$\\#] ',而不是[PEXPECT]#。我建议你这样写:

    savePrompt = s.PROMPT
    
    s.PROMPT = 'a-new-prompt'
    s.sendline('bteq ...')
    s.prompt()                 # expect the new prompt
    
    s.PROMPT = savePrompt      # restore the saved prompt
    s.sendline('quit;')
    s.prompt()                 # expect the saved prompt
    
  • sync_original_prompt()应该像这样使用:

    s.sync_original_prompt() # this make sure you are at the prompt
    s.set_unique_prompt()    # change the shell's prompt to the default PROMPT
    s.sendline('run some command')
    s.prompt()