github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

alert_print fuction

pull/6/head
guanghongwei 2014-12-22 16:55:49 +08:00
parent e2e32b4b37
commit e4c6823ee2
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
connect.py
View File

@ -32,6 +32,11 @@ def red_print(string):
print '\033[1;31m%s\033[0m' % string print '\033[1;31m%s\033[0m' % string
def alert_print(string):
red_print(string)
time.sleep(2)
sys.exit()
def get_win_size(): def get_win_size():
"""This function use to get the size of the windows!""" """This function use to get the size of the windows!"""
if 'TIOCGWINSZ' in dir(termios): if 'TIOCGWINSZ' in dir(termios):
@ -67,14 +72,12 @@ def posix_shell(chan, user, host):
try: try:
os.makedirs(today_connect_log_dir) os.makedirs(today_connect_log_dir)
except OSError: except OSError:
red_print('Create %s failed, Please modify %s permission.' % (today_connect_log_dir, connect_log_dir)) alert_print('Create %s failed, Please modify %s permission.' % (today_connect_log_dir, connect_log_dir))
sys.exit()
try: try:
log = open(log_file_path, 'a') log = open(log_file_path, 'a')
except IOError: except IOError:
red_print('Create logfile failed, Please modify %s permission.' % today_connect_log_dir) alert_print('Create logfile failed, Please modify %s permission.' % today_connect_log_dir)
sys.exit()
old_tty = termios.tcgetattr(sys.stdin) old_tty = termios.tcgetattr(sys.stdin)
try: try:
@ -124,13 +127,9 @@ def connect(username, password, host, port):
try: try:
ssh.connect(host, port=port, username=username, password=password, compress=True) ssh.connect(host, port=port, username=username, password=password, compress=True)
except paramiko.ssh_exception.AuthenticationException: except paramiko.ssh_exception.AuthenticationException:
red_print('Password Error, Please Correct it.') alert_print('Host Password Error, Please Correct it.')
time.sleep(2)
sys.exit()
except socket.error: except socket.error:
red_print('Connect SSH Socket Port Error, Please Correct it.') alert_print('Connect SSH Socket Port Error, Please Correct it.')
time.sleep(2)
sys.exit()
# Make a channel and set windows size # Make a channel and set windows size
global channel global channel
@ -144,7 +143,7 @@ def connect(username, password, host, port):
# Modify PS1 # Modify PS1
channel.send('PS1=%s' % ps1) channel.send('PS1=%s' % ps1)
channel.send("clear;echo -e '\\033[32mLogin %s OK.\\033[0m'\n" % host) channel.send("clear;echo -e '\\033[32mLogin %s done. Enjoy it.\\033[0m'\n" % host)
# Make ssh interactive tunnel # Make ssh interactive tunnel
posix_shell(channel, username, host) posix_shell(channel, username, host)