Cleanly exit from a Ctrl-C and add 'q'/'quit' commands

fredreichbier · Aug 9, 2015 · 586a00b · 586a00b
1 parent 96e1da6
commit 586a00b
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/browse_slp.py b/browse_slp.py
@@ -262,6 +262,12 @@ def __init__(self, loader):
 
         self.loader = loader
 
+    def do_q(self, params):
+        sys.exit(0)
+
+    def do_quit(self, params):
+        self.do_q(params)
+
     def do_drs(self, filename):
         """
             Load a specified DRS file.
@@ -445,4 +451,7 @@ def do_saveall(self, params):
         for command in args.commands:
             browser.onecmd(command)
     if not args.batch_mode:
-        browser.cmdloop()
+        try:
+            browser.cmdloop()
+        except KeyboardInterrupt:
+            sys.exit(0)