def main():
inputClient = InputClient()
api = Api()
usage = "usage: %prog [options] [arg]"
parser = OptionParser(usage)
parser.add_option('-l',
'--list',
action='store_true',
dest="iflist",
help=u"列出所有的ssh连接")
parser.add_option('-a',
'--add',
action='store_true',
dest="ifadd",
help=u"添加ssh连接")
parser.add_option('-i',
'--host',
action="store",
dest="host",
help=u"连接的host")
parser.add_option('-d', action='store', dest='del_host', help=u"删除ssh连接")
parser.add_option('-D',
'--delete-by-index',
action='store_true',
dest='del_by_index',
help=u"通过index删除ssh连接")
parser.add_option('-u', '--update', action="store_true", dest="ifupdate")
# parser.add_option('-')
(option, args) = parser.parse_args()
opt_dict = eval(str(option))
opt_values = opt_dict.values()
param_len = len(opt_values) - opt_values.count(None)
output = OutputClient()
output.set_header(['index', 'Host', 'UserName'], [7, 17, 15])
if param_len > 1:
raise ParamsException('参数错误')
elif param_len == 0:
if len(args) > 0:
host = args[0]
ssh_conn = api.get_ssh_connect(host)
if ssh_conn is not None:
api.login(ssh_conn.get('host'), ssh_conn.get('name'),
ssh_conn.get('passwd'))
else:
ssh_conns = api.login_fuzzy(host)
con_len = len(ssh_conns)
if con_len == 0:
sys.stdout.write("没有匹配的Host!\n")
return
elif con_len == 1:
api.login(ssh_conns[0].get('host'),
ssh_conns[0].get('name'),
ssh_conns[0].get('passwd'))
return
else:
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
ssh_conn = output.select()
api.login(ssh_conn.get('host'), ssh_conn.get('name'),
ssh_conn.get('passwd'))
return
return
else:
ssh_conns = api.list_ssh_connects()
output.set_values(ssh_conns)
# print output.show()
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select()
api.login(ssh_conn.get('host'), ssh_conn.get('name'),
ssh_conn.get('passwd'))
return
else:
iflist = option.iflist
ifadd = option.ifadd
host = option.host
ifupdate = option.ifupdate
del_host = option.del_host
del_by_index = option.del_by_index
try:
if iflist:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
return
if ifadd:
host = inputClient.input_host()
username = inputClient.input_username()
password = inputClient.input_password()
api.add_ssh_connect(host, username, password)
# print host, username, password
return
if host:
ssh_conn = api.get_ssh_connect(host)
if ssh_conn is None:
raise Exception("错误: 主机 %s 不存在!" % host)
else:
api.login(ssh_conn.get('host'), ssh_conn.get('name'),
ssh_conn.get('passwd'))
if ifupdate:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select_to_update()
username = inputClient.input_username()
password = inputClient.input_password()
api.update_ssh_connect(ssh_conn.get('host'), username,
password)
sys.stdout.write('更新成功!\n')
if del_host:
ssh_conn = api.get_ssh_connect(del_host)
if ssh_conn is None:
raise Exception("错误: 主机 %s 不存在!" % host)
api.del_ssh_connect(del_host)
sys.stdout.write('删除成功!\n')
if del_by_index:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select_to_del()
api.del_ssh_connect(ssh_conn.get('host'))
sys.stdout.write('删除成功!\n')
return
except ParamsException as e:
sys.stdout.write(e.msg)
sys.stdout.write('\n')
except TIMEOUT as e:
sys.stdout.write("连接超时!\n")
except Exception as e:
sys.stdout.write(e.message)
sys.stdout.write('\n')
def main():
inputClient = InputClient()
api = Api()
usage = "usage: %prog [options] [arg]"
parser = OptionParser(usage)
parser.add_option('-l', '--list', action='store_true', dest="iflist", help=u"list all ssh information")
parser.add_option('-a', '--add', action='store', dest="ifadd",
help=u"Add ssh information.<name>@<ip>[:<ssh port>][@<alias>]. Example: root:1.1.1.1:1010-home1 or root:1.1.1.2")
parser.add_option('-i', '--host', action="store", dest="host", help=u"Connect remote with the host ip")
parser.add_option('-d', action='store', dest='del_host', help=u"Remove ssh information")
parser.add_option('-D', '--delete-by-index', action='store_true', dest='del_by_index',
help=u"Remove ssh information by index id")
parser.add_option('-u', '--update', action="store_true", dest="ifupdate")
# parser.add_option('-')
(option, args) = parser.parse_args()
opt_dict = eval(str(option))
opt_values = opt_dict.values()
param_len = len(opt_values) - opt_values.count(None)
output = OutputClient()
output.set_header(['Index','UserName', 'Host', 'Port', 'Alias'], [7, 17, 15, 10, 30])
if param_len > 1:
raise ParamsException('Param Error')
elif param_len == 0:
try:
if len(args) > 0:
host = args[0]
ssh_conn = api.get_ssh_connect(host)
if ssh_conn is not None:
api.login(ssh_conn.get('host'), ssh_conn.get('name'), ssh_conn.get('passwd'), ssh_conn.get('port'))
else:
ssh_conns = api.login_fuzzy(host)
con_len = len(ssh_conns)
if con_len == 0:
sys.stdout.write(redStr('No Matched Host\n'))
return
elif con_len == 1:
api.login(ssh_conns[0].get('host'), ssh_conns[0].get('name'), ssh_conns[0].get('passwd'),ssh_conns[0].get('port') )
return
else:
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
ssh_conn = output.select()
api.login(ssh_conn.get('host'), ssh_conn.get('name'), ssh_conn.get('passwd'), ssh_conn.get('port'))
return
return
else:
ssh_conns = api.list_ssh_connects()
output.set_values(ssh_conns)
# print output.show()
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select()
api.login(ssh_conn.get('host'), ssh_conn.get('name'), ssh_conn.get('passwd'), ssh_conn.get('port'))
return
except ParamsException as e:
sys.stdout.write(e.msg)
sys.stdout.write('\n')
except TIMEOUT as e:
sys.stdout.write("Connection Timeout!\n")
except Exception as e:
sys.stdout.write(e.message)
sys.stdout.write('\n')
else:
iflist = option.iflist
add = option.ifadd
host = option.host
ifupdate = option.ifupdate
del_host = option.del_host
del_by_index = option.del_by_index
try:
if iflist:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
return
if add:
s_l = add.split('@')
if len(s_l) < 2:
sys.stdout.write(
redStr('The data format is not correct. Example: <name>@<ip>[:<ssh port>][@<alias>]'))
return
username = s_l[0]
port = 22
alias = ''
if len(s_l) == 3:
alias = s_l[2]
ip_port_arr = s_l[1].split(':')
host = ip_port_arr[0]
try:
if len(ip_port_arr) == 2:
port = int(ip_port_arr[1])
except Exception, e:
sys.stdout.write(
redStr('The data format is not correct. Example: <name>@<ip>[:<ssh port>][@<alias>]'))
return
password = getpass.getpass('Input Your Password:'******'host'), ssh_conn.get('name'), ssh_conn.get('passwd'), ssh_conn.get('port'))
if ifupdate:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select_to_update()
username = inputClient.input_username()
password = inputClient.input_password()
api.update_ssh_connect(ssh_conn.get('host'), username, password)
sys.stdout.write(greenStr('Update Successfully!\n'))
if del_host:
ssh_conn = api.get_ssh_connect(del_host)
if ssh_conn is None:
# raise Exception("错误: 主机 %s 不存在!" % host)
raise Exception("Error: Host %s is not exist!" % host)
api.del_ssh_connect(del_host)
# sys.stdout.write('删除成功!\n')
sys.stdout.write(greenStr('Delete Successfully!'))
if del_by_index:
ssh_conns = api.list_ssh_connects()
# print ssh_conns
output.set_values(ssh_conns)
sys.stdout.write(output.show())
sys.stdout.write('\n')
if len(ssh_conns) > 0:
ssh_conn = output.select_to_del()
api.del_ssh_connect(ssh_conn.get('host'))
sys.stdout.write(greenStr('Delete Successfully!\n'))
return
except ParamsException as e:
sys.stdout.write(e.msg)
sys.stdout.write('\n')