一、pexpect的安装

pexpect作为Python的一个普通模块,可以通过pip安装,也可以通过源码安装

1、pip安装

pip install pexpect
easy_install pexpect

2、源码安装

wget https://github.com/pexpect/pexpect.git
cd pexpect
python setup.py install

二、pexpect的核心组件

pexpect包含的核心组件有:spawn类、run函数及派生类pxssh等的定义及使用方法。


1、spawn类

spawn是pexpect的主要类接口,功能是启动和控制子应用程序,以下是它的构造函数定义:


class pexpect.spawn(command, args=[], timeout=30, maxread=2000, searchwindowsize=None, logfile=None, cwd=None, env=None, ignore_sighup=True)


其中command参数可以是任意已知的系统命令,如:


child = pexpect.spawn('/usr/bin/ftp') #启动ftp客户端命令
child = pexpect.spawn('/usr/bin/ssh ip') #启动ssh远程连接命令
child = pexpect.spawn('ls -latr /tmp') #运行ls显示/tmp目录内容命令

当子程序需要参数时,还可以使用Python列表来代替参数项,如:


child = pexpect.spawn('/usr/bin/ftp', []) 
child = pexpect.spawn('/usr/bin/ssh', [ip])
child = pexpect.spawn('ls',['-latr', '/tmp'])

参数timeout为等待结果的超时时间;参数maxread为pexpect从终端控制台一次读取的最大字节数,searchwindowsize参数为匹配缓冲字符串的位置,默认是从开始位置匹配。


需要注意的是,pexpect不会解析shell命令当中的元字符,包括重定向 > 、管道 | 或通配符 * , 我们可以通过将存在这三个特殊元字符的命令作为/bin/bash的参数进行调用,例如:


child = pexpect.spawn('/bin/bash -c "ls -l | grep LOG > logs.txt"')
child.expect(pexpect.EOF)

我们可以通过将命令的参数以Python列表的形式进行替换,从而使我们的语法变成更加清晰,下面的代码等价于上面的。


shell_cmd  = 'ls -l | grep LOG > logs.txt'
child = pexpect.spawn('/bin/bash', ['-c', shell_cmd])
child.expect(pexpect.EOF)

此外,pexpect支持日志输出,可以输出到控制台,也可以输出到文件


1)输出到标准输出


child = pexpect.spwn('some_command')
child.logfile = sys.stdout

2)输出到文件


child = pexpect.spwn('some_command')
fout = file('log.txt', 'w')
child.logfile = fout

2、run函数



函数定义:


pexpect.run(command, timeout=-1, withexitstatus=False,events=NOne,extra_args=None,logfile=None,cwd=None,evn=None)

参数command可以是系统已知的任意命令,如没有写绝对路径时将会尝试搜索命令的路径,events是一个字典,定义了expect及sendline方法的对应关系,spawn方式例子如下:


from pexpect import *
child = spawn('scp foo ip:.')
child.expect('(?i)password')
child.sendline(mypassword)

使用run函数如下


from pexpect import *
run('scp foo ip:.', event={'(?i)password' : mypassword})

3、pxssh类

pxssh类定义


class pexpect.pxssh.pxssh(timeout=30,maxread=2000,searchwindowsize=None,logfile=None,cwd=None,evn=None)

常用的方法如下:


login()建立ssh连接


logout()断开连接


prompt()等待系统提示符,用户等待命令执行结束.


下面使用pxssh类实现一个ssh连接远程主机并执行命令的示例。首先使用login()方法与远程主机建立连接,再通过sendline方法发送执行的命令,prompt()方法等待执行结束并出现系统提示符,最后logout方法断开连接。


# -*- coding:UTF-8 -*-
'''
Created on 2018年1月9日

@author: liuyazhuang
'''
import pxssh
import getpass
try:
#创建pxssh对象s
s = pxssh.pxssh()
hostname = raw_input('hostname: ')
username = raw_input('username: ')
#接收密码输入
password = getpass.getpass('please input password: ')
#建立ssh连接
s.login(hostname, username, password)
#运行uptime命令
s.sendline('uptime')
#匹配系统提示符
s.prompt()
#打印出现系统提示符前的命令输出
print s.before
s.sendline('ls -l')
s.prompt()
print s.before
s.sendline('df')
s.prompt()
print s.before
#断开ssh连接
s.logout()

except pxssh.ExceptionPxssh, e:
print 'pxssh failed on login.'
print str(e)