修复当用户使用Ctrl+C结束进程时,没有正确退出

10 years ago · 6998036bdc
parent e09cb10439
commit 6998036bdc
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