python之pexpect模块

时间:2023-01-27 05:26:24

最近在看《Python自动化运维技术与最佳实战》这本书,学到了一个运维中用到的模块:pexpect

下面是其定义:

Pexpect 是一个用来启动子程序并对其进行自动控制的 Python 模块,它可以用来和像 ssh、ftp、passwd、telnet 等命令行程序进行自动交互。

从书中我摘取了相关实例并配有注释,用来记录pexpect常用方法:

下面这个实例功能是:以ssh远程登录,登录成功后执行命令‘ls -lh’

 import pexpect
import sys #通过spawn类启动和控制子应用程序
child = pexpect.spawn('ssh root@192.168.1.22') #将pexpect的输入输出信息写到mylog.txt文件中
fout = file('mylog.txt','w')
child.logfile = fout #将pexpect的输入输出信息输出到标准输出
#child.logfile = sys.stdout #expect方法用来判断子程序产生的输出,判断是否匹配相应字符串
child.expect('password:')
#字符串匹配则使用sendline进行回应-----send:发送命令,不回车、sendline:发送命令,回车、sendcontrol:发送控制符,如:sendctrol('c')等价于‘ctrl+c'、sendeof:发送eof
child.sendline('')
child.expect('#')
child.sendline('ls -lh')
child.expect('#')

pexpect之spawn类

当然,ssh登录还可以使用pexpect的run函数实现:

#events是一个字典,定义了expect和sendline的对应关系
pexpect.run('ssh root@192.168.1.22',events={'password:',''})

针对ssh远程登录,pexpect又派生出了pxssh类,在ssh会话操作上再做一层封装

其常用方法是:

login()  #建立ssh连接
logout() #断开ssh连接
prompt() #等待系统提示符,用于等待命令执行结束

下面是使用pxssh类来进行远程连接并进行操作的实例:

 from pexpect import pxssh
import getpass try:
s = pxssh.pxssh() #创建pxssh对象 hostname = raw_input('hostname:')
username = raw_input('username:')
password = getpass.getpass('password:') #接收密码输入 s.login(server=hostname,username=username,password=password) #建立ssh连接 s.sendline('uptime') #运行uptime命令
s.prompt() #匹配系统提示符
print s.before #打印出现系统提示符前的命令输出 s.sendline('ls -lh') #运行命令
s.prompt() #匹配系统提示符
print s.before #打印出现系统提示符前的命令输出 s.sendline('df -h') #运行命令
s.prompt() #匹配系统提示符
print s.before #打印出现系统提示符前的命令输出 s.logout() #断开ssh连接 except pxssh.ExceptionPxssh as e:
print 'pxssh failed on login'
print str(e)

pxssh类