python 管理ssh客户端

root@zenghui:~/python# cat ipnew.txt
vm_esxi133,192.168.11.133,22,1qaz2wsx,root,0
int_www45,192.168.11.45,22,wdzj201445,root,0
int_23,192.168.11.23,22,wdzj2014,root,0
int_24,192.168.11.24,2022,wdzjkey,salt,1
root@zenghui:~/python# cat ip_ssh.py 
#!/usr/bin/python
#-*- coding:utf-8 -*-
import pexpect,commands,os,re,sys
import struct, fcntl, signal,termios
ip_new=open('ipnew.txt','a+')
global ipaddress,ipaddresskey
ipaddress={}
ipaddresskey={}

#从ipnew.txt获取内容加入ipaddress字典中
def ip_name():
    file=open('ipnew.txt','r').readlines()
    ii=0
    for i in file:
    ii+=1
        aa=i.split(',')
        ipaddress.setdefault(ii,[aa[0],[aa[1],aa[2],aa[3],aa[4],aa[5]]])
        ipaddresskey.setdefault(aa[0],[aa[1],aa[2],aa[3],aa[4],aa[5]])
    #print ipaddress    
#添加用户
def ipadd():
    ipre=[]
    namere=[]
    portre=[]
    pwre=[]
    keyre=[]
    usernamere=[]
    while namere==[]:
        name=str(raw_input('请输入你的项目名称： '))
        namere=re.findall('w+',name)
        if len(namere) > 1 or ipaddresskey.has_key(name)==True:
            print '你输入的格式不对或名称已经存在，请重新输入!'
            namere=[]
            
    while ipre==[]:
        ip=str(raw_input('请输入ip地址:192.168.11.23: '))
        ipre=re.findall('d{1,3}.d{1,3}.d{1,3}.d{1,3}',ip)
        if len(ipre) > 1:
            ipre=[]
    while portre==[]:
        port=str(raw_input('请输入端口号:22: '))
        portre=re.findall('d{1,5}',port)
        if len(portre) > 1:
            portre=[]
            
    while pwre==[]:
        pw=str(raw_input('请输入密码“： '))
        pwre=re.findall('S+',pw)
        if len(pwre) > 1:
            pwre=[]    
            
    while usernamere==[]:
        username=str(raw_input('输入用户名： '))
        usernamere=re.findall('w+',username)
        if len(usernamere) > 1:
            usernamere=[]

    while keyre==[]:
        key=str(raw_input('请输入key“： '))
        keyre=re.findall('S+',key)
        if len(keyre) > 1:
            keyre=[]    

    ipadd=str(name)+','+str(ip)+','+str(port)+','+str(pw)+','+str(username)+','+str(key)+'
'
    ip_new.write(ipadd)
    print ipadd+'添加成功！'
    exit()
   # ip_new.close()
   # file.close()


#主菜单
def menu():
    os.system('clear')
    print '**************************MENU*****************************'
    for key,value in ipaddress.items():
        print str(key)+'   '+str(value[0])+'        '+str(value[1][0])
    print "回车退出!"
    print '***********************************************************'
    name=int(raw_input('请输入机器编号： '))
    if name == '':
    exit()
    ssh_command(ipaddress[name][1][0],ipaddress[name][1][1],ipaddress[name][1][2],ipaddress[name][1][3],ipaddress[name][1][4].strip())
    

#用户登陆
def ssh_command(ipaddr,port,passwd,name,key):
    try:
    global ssh_key
        if key == '0':
            keys=''
            ssh_key=pexpect.spawn('ssh %s -p  %s %s@%s' % (keys,port,name,ipaddr)) 
            status=ssh_key.expect(['password'])
        else:
            keys='-i id_dsa_key'
            ssh_key=pexpect.spawn('ssh %s -p  %s %s@%s' % (keys,port,name,ipaddr))
            status=ssh_key.expect([pexpect.TIMEOUT,'id_dsa_key'])
        if status==1:
            ssh_key.sendline(passwd)
        elif status==0:
            ssh_key.sendline('yes
')
            ssh_key.expect('password:')
            ssh_key.sendline(passwd)
        signal.signal(signal.SIGWINCH, sigwinch_passthrough)
        size = getwinsize()
        ssh_key.setwinsize(size[0], size[1])
        ssh_key.interact()
    except Exception as e:
        print e

#设置终端大小适合窗口
def sigwinch_passthrough (sig, data):
    winsize = getwinsize()
    global ssh_key
    ssh_key.setwinsize(winsize[0],winsize[1])
#获取终端屏幕大小
def getwinsize():
    if 'TIOCGWINSZ' in dir(termios):
        TIOCGWINSZ = termios.TIOCGWINSZ
    else:
        TIOCGWINSZ = 1074295912L # Assume
    s = struct.pack('HHHH', 0, 0, 0, 0)
    x = fcntl.ioctl(sys.stdout.fileno(), TIOCGWINSZ, s)
    return struct.unpack('HHHH', x)[0:2]
  

if __name__ == "__main__":
    try:
    if len(sys.argv) == 2:
            if sys.argv[1]=='ipadd':
            ip_name()
                ipadd()
        else:
            print '请输入参数"ipadd"'
    elif len(sys.argv) == 1:
            ip_name()
            menu()
    else:
        print '请重新输入!'
    except:
    pass