Browse Source

修改登录名和记录日志不一样的问题

pull/6/head
ibuler 10 years ago
parent
commit
fead462124
  1. 14
      connect.py

14
connect.py

@ -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

Loading…
Cancel
Save