setbranch : make sure it stops with keyboard interrupt

1 files changed, 8 insertions, 0 deletions

diff --git a/src/backend/setbranch.py b/src/backend/setbranch.py
index 23b1b86..ca349f9 100644
--- a/src/backend/setbranch.py
+++ b/src/backend/setbranch.py
@@ -3,6 +3,7 @@
 import animation
 import git
 import os
+import signal
 import sys
 import sisyphus.checkenv
 import sisyphus.getcolor
@@ -26,6 +27,13 @@ brch_rmt_map = {
 }
 
 
+def sigint_handler(signal, frame):
+    sys.exit(0)
+
+
+signal.signal(signal.SIGINT, sigint_handler)
+
+
 def get_brch_rmt(branch, remote):
     g_rmt = []
     r_rmt = []