用Pexpect与SSH交互（实现了参数化输入以及命令行shell交互的形式）

#!/usr/bin/python  
#coding=utf-8  
import pexpect  
from optparse import OptionParser  
  
#SSH连接成功时的命令行交互窗口中的提示符的集合  
PROMPT = ['# ','>>> ','> ','\$ ']  
  
def send_command(child,cmd):  
    #发送一条命令  
    child.sendline(cmd)  
  
    #期望有命令行提示字符出现  
    child.expect(PROMPT)  
  
    #将之前的内容都输出  
    print child.before.split('\n')[1]  
  
def connect(user,host,password):  
    #表示主机已使用一个新的公钥的消息  
    ssh_newkey = 'Are you sure you want to continue connecting'  
    connStr = 'ssh ' + user + '@' + host  
  
    #为ssh命令生成一个spawn类的对象  
    child = pexpect.spawn(connStr)  
  
    #期望有ssh_newkey字符、提示输入密码的字符出现，否则超时  
    ret = child.expect([pexpect.TIMEOUT,ssh_newkey,'[P|p]assword: '])  
  
    #匹配到超时TIMEOUT  
    if ret == 0:  
        print '[-] Error Connecting'  
        return  
  
    #匹配到ssh_newkey  
    if ret == 1:  
        #发送yes回应ssh_newkey并期望提示输入密码的字符出现  
        child.sendline('yes')  
        ret = child.expect([pexpect.TIMEOUT,ssh_newkey,'[P|p]assword: '])  
  
    #匹配到超时TIMEOUT  
    if ret == 0:  
        print '[-] Error Connecting'  
        return  
  
    #发送密码  
    child.sendline(password)  
    child.expect(PROMPT)      
    return child  
  
def main():  
    parser = OptionParser("[*] Usage : ./sshCommand2.py -H <target host> -u <username> -p <password>")  
    parser.add_option('-H',dest='host',type='string',help='specify target host')  
    parser.add_option('-u',dest='username',type='string',help='target username')  
    parser.add_option('-p',dest='password',type='string',help='target password')  
    (options,args) = parser.parse_args()  
  
    if (options.host == None) | (options.username == None) | (options.password == None):  
        print parser.usage  
        exit(0)  
  
    child=connect(options.username,options.host,options.password)  
      
    while True:  
        command = raw_input('<SSH> ')  
        send_command(child,command)  
  
if __name__ == '__main__':  
    main()