Catching KeyboardInterrupt

Scstechr · Aug 2, 2019 · 2f20169 · 2f20169
1 parent ced1345
commit 2f20169
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/gch.py b/gch.py
@@ -10,6 +10,7 @@
 import sys, subprocess as sp
 from os import path, chdir, getcwd
 import os
+import cursor
 
 from src import issues
 from src.qs import getAnswer, isExist , confirm
@@ -114,4 +115,7 @@ def main():
         issues.ok('No push')
 
 if __name__ == "__main__":
-    main()
+    try:
+        main()
+    except KeyboardInterrupt:
+        issues.abort()
diff --git a/src/qs.py b/src/qs.py
@@ -1,5 +1,6 @@
 import subprocess as sp
 from . import issues
+import sys
 
 def echo(string):
     print(string)