Merge pull request #6 from losswei/master

修复当用户使用Ctrl+C结束进程时,没有正确退出
2015-05-22 17:24:46 +08:00 · 2015-05-22 17:24:46 +08:00 · 4a9e002440
parent e09cb10439 6998036bdc
commit 4a9e002440
1 changed files with 2 additions and 0 deletions
--- a/connect.py
+++ b/connect.py
@ -376,6 +376,8 @@ if __name__ == '__main__':
            except EOFError:
                print
                continue
+            except KeyboardInterrupt:
+                sys.exit(0)
            if option in ['P', 'p']:
                print_user_host(LOGIN_NAME)
                continue