python expect用法_python pexpect包用法

简介

Pexpect 是 Expect 语言的一个 Python 实现，是一个用来启动子程序，并使用正则表达式对程序输出做出特定响应，以此实现与其自动交互的 Python 模块。因为工作需要我用它来抓取交换机相关数据信息，话不多说，附上我的代码片段、说明。

代码：

def connect(user, host, password):

ssh_newkey = "Are you sure you want to continue connecting"

connStr = 'ssh ' + user + '@' + host

child = pexpect.spawn(connStr) # 实例化连接

ret = child.expect([pexpect.TIMEOUT, ssh_newkey, '[P|p]assword:', pexpect.EOF]) ##预期捕获的结果，只有遇到password才是能够继续往下走的，其他预期都是代表无法正常登录

if ret == 0: # 捕获超时

print('[-] Error Connecting timeout:' + str(host))

logger.warning("Error Connecting: timeout" + str(host))

return

if ret == 1: # 捕获了ssh_newkey的消息，需要互动

child.sendline('yes') # 发送yes

ret = child.expect([pexpect.TIMEOUT, ssh_newkey, '[P|p]assword:', pexpect.EOF])

if ret == 0:

print('[-] Error Connecting timeout:' + str(host))

logger.warning("Error Connecting: timeout" + str(host))

return

if ret == 3: # 捕获不知名的错误

print('[-] Error Connecting eof:' + str(host))

logger.warning("Error Connecting eof:" + str(host))

return

child.sendline(password)

r = child.expect(['>','Permission denied',pexpect.TIMEOUT,pexpect.EOF]) # 捕获命令提示符，因为我们交换机上的tooltip都是以">"结尾的

if r > 0:

print('[-] Permission denied:' + str(host))

logger.warning("Permission denied:" + str(host))

return

return child

#将系统ansi字符替换为空，以避免后续匹配失败问题

def escape_ansi(line):

ansi_escape = re.compile(r'(?:\x1B[@-_]|[\x80-\x9F])[0-?]*[ -/]*[@-~]')

return ansi_escape.sub('', line)

#执行命令

def send_command(child, cmd):

patterns = ['---- More ----','>', pexpect.EOF,pexpect.TIMEOUT]

data = ''

child.sendline(cmd)

#遇到需要翻页的情况，需要输入空格键，进行翻页

while True:

ret = child.expect(patterns)

data += escape_ansi(child.before.decode())

if ret == 0:

child.send(' ')

elif ret >= 1:

break

return data