python expect_python--pexpect

大家好,最近工作比较忙,所以没时间来更新博客。趁着还没在下个版本来临之前,来这边再更新更新。是之前学习到的一些老知识点,就当来巩固一下了。开心QAQ

今天给大家介绍的是--Pexpect

Expect 程序主要用于人机对话的模拟

1.运行程序

2.程序要求人的判断和输入

3.Expect 通过关键字匹配

4.根据关键字向程序发送符合的字符串

基本使用流程

基本使用流程

1.首先用 spawn 来执行一个程序

2.然后用 expect 来等待指定的关键字，这个关键字是被执行的程序打印到标准输出上面的

3.最后当发现这个关键字以后，根据关键字用 send 方法来发送字符串给这个程序

以下就是代码了,比较简单。但很实用

#-*- coding:utf-8 -*-

"""This runs a command on a remote host using SSH. At the prompts enter hostname,

user, password and the command."""

importpexpectimportgetpass, os#user: ssh 主机的用户名#host：ssh 主机的域名#password：ssh 主机的密码#command：即将在远端 ssh 主机上运行的命令

defssh_command (user, host, password, command):"""This runs a command on the remote host. This could also be done with the

pxssh class, but this demonstrates what that class does at a simpler level.

This returns a pexpect.spawn object. This handles the case when you try to

connect to a new host and ssh asks you if you want to accept the public key

fingerprint and continue connecting."""ssh_newkey= 'Are you sure you want to continue connecting'

#为 ssh 命令生成一个 spawn 类的子程序对象.

child = pexpect.spawn('ssh -l %s %s %s'%(user, host, command))

i= child.expect([pexpect.TIMEOUT, ssh_newkey, 'password:'])#如果登录超时，打印出错信息，并退出.

if i == 0: #Timeout

print 'ERROR!'

print 'SSH could not login. Here is what SSH said:'

printchild.before, child.afterreturnNone#如果 ssh 没有 public key，接受它.

if i == 1: #SSH does not have the public key. Just accept it.

child.sendline ('yes')

child.expect ('password:')

i= child.expect([pexpect.TIMEOUT, 'password:'])if i == 0: #Timeout

print 'ERROR!'

print 'SSH could not login. Here is what SSH said:'

printchild.before, child.afterreturnNone#输入密码.

child.sendline(password)returnchilddefmain ():#获得用户指定 ssh 主机域名.

host = '10.240.176.172'

#获得用户指定 ssh 主机用户名.

user = 'root'

#获得用户指定 ssh 主机密码.

password = 'tester'

#获得用户指定 ssh 主机上即将运行的命令.

command = 'ls -a /home'child=ssh_command (user, host, password, command)#匹配 pexpect.EOF

child.expect(pexpect.EOF)#输出命令结果.

printchild.beforeif __name__ == '__main__':try:

main()exceptException, e:printstr(e)#traceback.print_exc()

os._exit(1)

需要注意的知识点:

spawn() 方法用来执行一个程序,打开一个到 (user, host, command) 服务器的 ssh 连接

spawn() ,或者说 pexpect 并不会转译任何特殊字符

process = pexpect.spawn('/bin/bash –c "ls –l | grep LOG > log_list.txt"') or

cmd = "ls –l | grep LOG > log_list.txt"

process = pexpect.spawn("/bin/bash", ["-c", cmd])

process.expect(pexpect.EOF)

timeout - 超时时间

默认值： 30 (单位：秒)

maxread - 缓存设置

默认值： 2000 (单位：字符)

指定一次性试着从命令输出中读多少数据。如果设置的数字比较大，那么从 TTY 中读取数据的次数就会少一些。

设置为 1 表示关闭读缓存

logfile - 运行输出控制

默认值： None

process = pexpect.spawn("ftp sw-tftp", logfile=sys.stdout) 如果你想看到spawn过程中的输出，那么可以将这些输出写入到 sys.stdout

process = pexpect.spawn("ftp sw-tftp")

logFileId = open("logfile.txt", 'w')

process.logfile = logFileId

logfile_read - 获取标准输出的内容

默认值： None

记录执行程序中返回的所有内容，也就是去掉你发出去的命令，而仅仅只包括命令结果的部分：

process.logfile_read = sys.stdout

cwd - 指定命令执行的目录

默认值： None 或者说 ./

sendline("ls –l", cwd="/etc") 在 /etc 目录下执行 ls –l 命令

expect() - 关键字匹配

后面的匹配关键字是一个列表的话，就会返回一个数字表示匹配到了列表中第几个关键字，从 0 开始计算。

index = process.expect([

'Permission Denied',

'Terminal type',

'ftp>',

])

if index == 0:

print "Permission denied at host, can't login."

process.kill(0)

elif index == 1:

print "Login ok, set up terminal type…"

process.sendline('vty100')

process.expect("ftp>")

elif index == 2:

print "Login Ok, please send your command"

process.interact()

0.权限不足，这可能是ftp服务器出现问题，或者没有这个帐号，或者其他什么情况，反正只要发现这种情况的话，我们就给用户提示一下，

然后杀掉这个进程

1.登陆成功，但还要用户指定终端模式才能真正使用，所以我们在代码中指定了 vty100 这种模式，然后看是不是能真正使用了

2.还是登陆成功了，而且还可以直接输入命令操作 ftp 服务器了，于是我们提示用户，然后把操作权限交给用户

另外有一种特殊情况，如果同时有2个被匹配到，那么怎么办？简单来说就是这样：

原始流中，第一个被关键字匹配到的内容会被使用

匹配关键字列表中，最左边的会被使用

child.expect(['(?i)etc', '(?i)readme', pexpect.EOF, pexpect.TIMEOUT])

前 2 个匹配都是大小写无关的，关键就是这个 (?i) 匹配规则，它相当于 re.IGNORE 或者 re.I 这个关键字

send() - 发送关键字

send() 作为3个关键操作之一，用来向程序发送指定的字符串，它的使用没什么特殊的地方，比如：

process.expect("ftp>")

process.send("by\n")

sendline() - 发送带回车符的字符串

sendline() 和 send() 唯一的区别就是在发送的字符串后面加上了回车换行符，这也使它们用在了不同的地方：

只需要发送字符就可以的话用send()

如果发送字符后还要回车的话，就用 sendline()

特殊变量

pexpect.EOF - 匹配终止信号

pexpect.TIMEOUT - 匹配超时信号