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.
"""
 
import pexpect
import getpass, os
 
#user: ssh 主机的用户名
#host：ssh 主机的域名
#password：ssh 主机的密码
#command：即将在远端 ssh 主机上运行的命令
def ssh_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:'
        print child.before, child.after
        return None
    # 如果 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:'
            print child.before, child.after
        return None
    # 输入密码.
    child.sendline(password)
    return child
 
def main ():
    # 获得用户指定 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)
    # 输出命令结果.
    print child.before
 
if __name__ == '__main__':
    try:
        main()
    except Exception, e:
        print str(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
")

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

只需要发送字符就可以的话用send()
如果发送字符后还要回车的话，就用 sendline()

特殊变量
pexpect.EOF - 匹配终止信号
pexpect.TIMEOUT - 匹配超时信号