python 管理ssh客户端

  1 root@zenghui:~/python# cat ipnew.txt
  2 vm_esxi133,192.168.11.133,22,1qaz2wsx,root,0
  3 int_www45,192.168.11.45,22,wdzj201445,root,0
  4 int_23,192.168.11.23,22,wdzj2014,root,0
  5 int_24,192.168.11.24,2022,wdzjkey,salt,1
  6 root@zenghui:~/python# cat ip_ssh.py 
  7 #!/usr/bin/python
  8 #-*- coding:utf-8 -*-
  9 import pexpect,commands,os,re,sys
 10 import struct, fcntl, signal,termios
 11 ip_new=open(‘ipnew.txt‘,‘a+‘)
 12 global ipaddress,ipaddresskey
 13 ipaddress={}
 14 ipaddresskey={}
 15 
 16 #从ipnew.txt获取内容加入ipaddress字典中
 17 def ip_name():
 18     file=open(‘ipnew.txt‘,‘r‘).readlines()
 19     ii=0
 20     for i in file:
 21     ii+=1
 22         aa=i.split(‘,‘)
 23         ipaddress.setdefault(ii,[aa[0],[aa[1],aa[2],aa[3],aa[4],aa[5]]])
 24         ipaddresskey.setdefault(aa[0],[aa[1],aa[2],aa[3],aa[4],aa[5]])
 25     #print ipaddress    
 26 #添加用户
 27 def ipadd():
 28     ipre=[]
 29     namere=[]
 30     portre=[]
 31     pwre=[]
 32     keyre=[]
 33     usernamere=[]
 34     while namere==[]:
 35         name=str(raw_input(‘请输入你的项目名称： ‘))
 36         namere=re.findall(‘\w+‘,name)
 37         if len(namere) > 1 or ipaddresskey.has_key(name)==True:
 38             print ‘你输入的格式不对或名称已经存在，请重新输入!‘
 39             namere=[]
 40             
 41     while ipre==[]:
 42         ip=str(raw_input(‘请输入ip地址:192.168.11.23: ‘))
 43         ipre=re.findall(‘\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}‘,ip)
 44         if len(ipre) > 1:
 45             ipre=[]
 46     while portre==[]:
 47         port=str(raw_input(‘请输入端口号:22: ‘))
 48         portre=re.findall(‘\d{1,5}‘,port)
 49         if len(portre) > 1:
 50             portre=[]
 51             
 52     while pwre==[]:
 53         pw=str(raw_input(‘请输入密码“： ‘))
 54         pwre=re.findall(‘\S+‘,pw)
 55         if len(pwre) > 1:
 56             pwre=[]    
 57             
 58     while usernamere==[]:
 59         username=str(raw_input(‘输入用户名： ‘))
 60         usernamere=re.findall(‘\w+‘,username)
 61         if len(usernamere) > 1:
 62             usernamere=[]
 63 
 64     while keyre==[]:
 65         key=str(raw_input(‘请输入key“： ‘))
 66         keyre=re.findall(‘\S+‘,key)
 67         if len(keyre) > 1:
 68             keyre=[]    
 69 
 70     ipadd=str(name)+‘,‘+str(ip)+‘,‘+str(port)+‘,‘+str(pw)+‘,‘+str(username)+‘,‘+str(key)+‘\n‘
 71     ip_new.write(ipadd)
 72     print ipadd+‘添加成功！‘
 73     exit()
 74    # ip_new.close()
 75    # file.close()
 76 
 77 
 78 #主菜单
 79 def menu():
 80     os.system(‘clear‘)
 81     print ‘**************************MENU*****************************‘
 82     for key,value in ipaddress.items():
 83         print str(key)+‘   ‘+str(value[0])+‘        ‘+str(value[1][0])
 84     print "回车退出!"
 85     print ‘***********************************************************‘
 86     name=int(raw_input(‘请输入机器编号： ‘))
 87     if name == ‘‘:
 88     exit()
 89     ssh_command(ipaddress[name][1][0],ipaddress[name][1][1],ipaddress[name][1][2],ipaddress[name][1][3],ipaddress[name][1][4].strip())
 90     
 91 
 92 #用户登陆
 93 def ssh_command(ipaddr,port,passwd,name,key):
 94     try:
 95     global ssh_key
 96         if key == ‘0‘:
 97             keys=‘‘
 98             ssh_key=pexpect.spawn(‘ssh %s -p  %s %s@%s‘ % (keys,port,name,ipaddr)) 
 99             status=ssh_key.expect([‘password‘])
100         else:
101             keys=‘-i id_dsa_key‘
102             ssh_key=pexpect.spawn(‘ssh %s -p  %s %s@%s‘ % (keys,port,name,ipaddr))
103             status=ssh_key.expect([pexpect.TIMEOUT,‘id_dsa_key‘])
104         if status==1:
105             ssh_key.sendline(passwd)
106         elif status==0:
107             ssh_key.sendline(‘yes\n‘)
108             ssh_key.expect(‘password:‘)
109             ssh_key.sendline(passwd)
110         signal.signal(signal.SIGWINCH, sigwinch_passthrough)
111         size = getwinsize()
112         ssh_key.setwinsize(size[0], size[1])
113         ssh_key.interact()
114     except Exception as e:
115         print e
116 
117 #设置终端大小适合窗口
118 def sigwinch_passthrough (sig, data):
119     winsize = getwinsize()
120     global ssh_key
121     ssh_key.setwinsize(winsize[0],winsize[1])
122 #获取终端屏幕大小
123 def getwinsize():
124     if ‘TIOCGWINSZ‘ in dir(termios):
125         TIOCGWINSZ = termios.TIOCGWINSZ
126     else:
127         TIOCGWINSZ = 1074295912L # Assume
128     s = struct.pack(‘HHHH‘, 0, 0, 0, 0)
129     x = fcntl.ioctl(sys.stdout.fileno(), TIOCGWINSZ, s)
130     return struct.unpack(‘HHHH‘, x)[0:2]
131   
132 
133 if __name__ == "__main__":
134     try:
135     if len(sys.argv) == 2:
136             if sys.argv[1]==‘ipadd‘:
137             ip_name()
138                 ipadd()
139         else:
140             print ‘请输入参数"ipadd"‘
141     elif len(sys.argv) == 1:
142             ip_name()
143             menu()
144     else:
145         print ‘请重新输入!‘
146     except:
147     pass