|
|
|
@ -37,6 +37,8 @@ CONF = ConfigParser()
|
|
|
|
|
CONF.read(os.path.join(CURRENT_DIR, 'jumpserver.conf'))
|
|
|
|
|
LOG_DIR = os.path.join(CURRENT_DIR, 'logs')
|
|
|
|
|
KEY = CONF.get('web', 'key')
|
|
|
|
|
LOGIN_NAME = getpass.getuser()
|
|
|
|
|
#LOGIN_NAME = os.getlogin()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def green_print(string):
|
|
|
|
@ -225,7 +227,7 @@ def verify_connect(username, part_ip):
|
|
|
|
|
except (ObjectDoesNotExist, IndexError):
|
|
|
|
|
red_print('Get get_connect_item Error.')
|
|
|
|
|
else:
|
|
|
|
|
connect(username, password, host, port)
|
|
|
|
|
connect(username, password, host, port, LOGIN_NAME)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def print_prompt():
|
|
|
|
@ -244,7 +246,7 @@ def print_user_host(username):
|
|
|
|
|
print '[%s] %s -- %s' % (hosts_attr[ip][0], ip, hosts_attr[ip][1])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def connect(username, password, host, port):
|
|
|
|
|
def connect(username, password, host, port, login_name):
|
|
|
|
|
"""
|
|
|
|
|
Connect server.
|
|
|
|
|
"""
|
|
|
|
@ -277,7 +279,7 @@ def connect(username, password, host, port):
|
|
|
|
|
channel.send(login_msg)
|
|
|
|
|
|
|
|
|
|
# Make ssh interactive tunnel
|
|
|
|
|
posix_shell(channel, username, host)
|
|
|
|
|
posix_shell(channel, login_name, host)
|
|
|
|
|
|
|
|
|
|
# Shutdown channel socket
|
|
|
|
|
channel.close()
|
|
|
|
@ -285,8 +287,6 @@ def connect(username, password, host, port):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
#login_name = os.getlogin()
|
|
|
|
|
login_name = getpass.getuser()
|
|
|
|
|
print_prompt()
|
|
|
|
|
try:
|
|
|
|
|
while True:
|
|
|
|
@ -296,14 +296,14 @@ if __name__ == '__main__':
|
|
|
|
|
print
|
|
|
|
|
continue
|
|
|
|
|
if option in ['P', 'p']:
|
|
|
|
|
print_user_host(login_name)
|
|
|
|
|
print_user_host(LOGIN_NAME)
|
|
|
|
|
continue
|
|
|
|
|
elif option in ['E', 'e']:
|
|
|
|
|
pass
|
|
|
|
|
elif option in ['Q', 'q']:
|
|
|
|
|
sys.exit()
|
|
|
|
|
else:
|
|
|
|
|
verify_connect(login_name, option)
|
|
|
|
|
verify_connect(LOGIN_NAME, option)
|
|
|
|
|
except IndexError:
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|