python 登陆线上服务器

#!/usr/bin/env python
#coding=utf-8
#QQ: 917611
import os
import time
import sys
import re

class Login:
    def __init__(self,user,port,ip):
        self.user = user
        self.port = port
        self.ip_dict = ip

    def server(self):
        for k,v in ip.items():
            print k,v

    def ssh1(self):
        #传入了变量直接登陆
        for key in ip.values():
            if re.findall(sys.argv[1],key):
                print "正在登陆服务器%s...." %key
                cmd = 'ssh -p%s ops@114.114.114.114  -t ssh -p22 %s@%s' %(port,user,key)
                os.system(cmd)
                sys.exit()

    def ssh2(self):
        stats = True
        while stats:
            search_id = raw_input('please input ip or id Login server: ')
            if len(search_id) == 0:
                continue
            if search_id == 'q':
                sys.exit()
            else:
                stats = False

            #判断输入的是id还是ip
            if len(search_id) == 1:
                cmd = 'ssh -p%s ops@114.114.114.114 -t ssh -p22 %s@%s' %(port,user,ip[int(search_id)])
                os.system(cmd)
            else:
                for key in ip.values():
                    if re.findall(search_id,key):
                        print "正在登陆服务器%s...." %key
                        cmd = 'ssh -p%s ops@114.114.114.114 -t ssh -p22  %s@%s' %(port,user,key)
                        os.system(cmd)
user = 'ops'
port = 51826

ip = {
              1:"172.20.12.1" , #php
              2:"172.20.12.2" , #pre_php
              3:"172.20.12.3" , #API
              4:"172.20.12.4" , #FastDFS
              5:"172.20.12.5", #JAVA
              6:"172.20.12.6", #JAVA
              7:"172.20.12.7", #JAVA
              8:"172.20.12.8" , #LB-Haproxy
              0:"172.20.12.9" , #LB-Haproxy
    }

if __name__ == '__main__':
    Job = Login(user,port,ip)
    Job.server()
    print '========================='
    if len(sys.argv) == 2:
        Job.ssh1()
    else:
        Job.ssh2()