From e2e32b4b37f7fd555e9d8bc9849d0d371967888d Mon Sep 17 00:00:00 2001 From: guanghongwei Date: Mon, 22 Dec 2014 16:38:10 +0800 Subject: [PATCH] add color print fuction --- connect.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/connect.py b/connect.py index c1fb3ace4..0454b2336 100755 --- a/connect.py +++ b/connect.py @@ -16,7 +16,7 @@ try: import termios import tty except ImportError: - print 'Only postfix supported.' + print '\033[1;31mOnly postfix supported.\033[0m' sys.exit() @@ -24,6 +24,14 @@ CURRENT_DIR = os.path.abspath('.') LOG_DIR = os.path.join(CURRENT_DIR, 'logs') +def green_print(string): + print '\033[1;32m%s\033[0m' % string + + +def red_print(string): + print '\033[1;31m%s\033[0m' % string + + def get_win_size(): """This function use to get the size of the windows!""" if 'TIOCGWINSZ' in dir(termios): @@ -59,13 +67,13 @@ def posix_shell(chan, user, host): try: os.makedirs(today_connect_log_dir) except OSError: - print 'Create %s failed, Please modify %s permission.' % (today_connect_log_dir, connect_log_dir) + red_print('Create %s failed, Please modify %s permission.' % (today_connect_log_dir, connect_log_dir)) sys.exit() try: log = open(log_file_path, 'a') except IOError: - print 'Create logfile failed, Please modify %s permission.' % today_connect_log_dir + red_print('Create logfile failed, Please modify %s permission.' % today_connect_log_dir) sys.exit() old_tty = termios.tcgetattr(sys.stdin) @@ -116,11 +124,11 @@ def connect(username, password, host, port): try: ssh.connect(host, port=port, username=username, password=password, compress=True) except paramiko.ssh_exception.AuthenticationException: - print 'Password Error, Please Correct it.' + red_print('Password Error, Please Correct it.') time.sleep(2) sys.exit() except socket.error: - print 'Connect SSH Socket Port Error, Please Correct it.' + red_print('Connect SSH Socket Port Error, Please Correct it.') time.sleep(2) sys.exit()