简介:Pexpect 是 Don Libes 的 Expect 语言的一个 Python 实现,是一个用来启动子程序,并使用正则表达式对程序输出做出特定响应,以此实现与其自动交互的 Python 模块
一:基本函数和类
- run 函数
#run 函数的定义
'''
withexitstatus:是否返回命令执行状态
events:自动化交互字典
logfile:定义命令输出打印位置(tty或者file)
'''
run(command,timeout=-1,withexitstatus=False,events=None, extra_args=None,logfile=None, cwd=None, env=None)
events 是一个字典,用来自动化交互,具体示例如下:
#########################################################################
# -*- coding:utf-8 -*-
# File Name: expect4.py
# Author: lcs
# Mail: liuchengsheng95@qq.com
# Created Time: 2017-09-28 15:04:54
# Description: 自动输入密码执行ssh命令
#########################################################################
# /usr/bin/python
import pexpect
import sys
pexpect.run('ssh lcs@192.168.1.12 ls -l /home/lcs', events={
# yes 是第一次 ssh 连接需要回复 "yes"
'yes':'yes',
'password:': '123456\n'
}, logfile=sys.stdout)
- spawn 核心类
class spawn:
'''
command:将要执行的命令
args:命令的参数
maxread:从tty一次读取的最大字节数
searchwindowsize:从开始开始匹配 expect 方法所匹配字符
logfile:日志文件的记录位置(tty或者file)
'''
def __init__(self,command,args=[],timeout=30,maxread=2000, searchwindowsize=None, logfile=None, cwd=None, env=None)
expect方法:等待子程序产生特定输出,做出特定的响应
expect(self, pattern, timeout=30, searchwindowsize=None)
pattern 可以是正则表达式, pexpect.EOF , pexpect.TIMEOUT ,或者由这些元素组成的列表。
需要注意的是,当 pattern 的类型是一个列表时,且子程序输出结果中不止一个被匹配成功,
则匹配返回的结果是缓冲区中最先出现的那个元素,或者是列表中最左边的元素。
使用 timeout 可以指定等待结果的超时时间 ,该时间以秒为单位。
当超过预订时间时, expect 匹配到pexpect.TIMEOUT。
send,sendall 和 sendcontrol: 与 send() 不同的是 sendline() 会额外输入一个回车符 ,更加适合用来模拟对子程序进行输入命令的操作。 当需要模拟发送 “Ctrl+c” 的行为时,还可以使用 sendcontrol('c') 发送控制字符
具体实例:
#########################################################################
# -*- coding:utf-8 -*-
# File Name: xxx.py
# Author: lcs
# Mail: liuchengsheng95@qq.com
# Created Time: 2017-09-28 10:57:06
#########################################################################
# /usr/bin/python
import pexpect
import sys
def commands():
child.sendline('123456')
child.expect('\$')
child.sendline('ip addr | grep inet')
child.logfile = sys.stdout
child.expect('\$')
def ssh(hostinfo):
global child
child = pexpect.spawn('ssh {}'.format(hostinfo))
index = child.expect(['yes', 'password' ,pexpect.TIMEOUT])
if index == 0:
child.sendline('yes')
commands()
elif index == 1:
commands()
else:
print('TIMEOUT')
if __name__ == '__main__':
hostinfo = raw_input('Pls input ssh host, expmale(user@host -p port):')
ssh(hostinfo)
- Pxssh 类
Pxssh 做为 pexpect 的派生类可以用来建立一个 ssh 连接,它相比其基类增加了如下方法:
login() 建立到目标机器的ssh连接 ;
losuckgout() 释放该连接 ;
prompt() 等待提示符,通常用于等待命令执行结束
具体实例:
#########################################################################
# -*- coding:utf-8 -*-
# File Name: xssh.py
# Author: lcs
# Mail: liuchengsheng95@qq.com
# Created Time: 2017-09-30 15:22:58
# Description:
#########################################################################
#/usr/bin/python
from pexpect import pxssh
import getpass
try:
s = pxssh.pxssh()
host = raw_input('Hostname:')
user = raw_input('Username:')
passwd = getpass.getpass('Password:')
s.login(host,user,passwd)
s.sendline('uptime')
s.prompt()
print(s.before)
s.sendline('ls -l')
s.prompt()
print(s.before)
s.logout()
except pxssh.ExceptionPxssh, e:
print('pxssh failed on login.')
print(str(e))