Exit clean from KeyboardInterrupt also

cyberrumor · Mar 12, 2024 · 96e8dab · 96e8dab
1 parent f82510c
commit 96e8dab
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ammo/ui.py b/ammo/ui.py
@@ -259,7 +259,7 @@ def help(self):
         print(out)
         try:
             input("[Enter] ")
-        except EOFError:
+        except (KeyboardInterrupt, EOFError):
             print()
             sys.exit(0)
 
@@ -322,10 +322,10 @@ def repl(self):
             try:
                 if not (stdin := input(f"{self.controller._prompt()}")):
                     continue
-            except EOFError:
+            except (KeyboardInterrupt, EOFError):
                 print()
                 sys.exit(0)
-                
+
             cmds = stdin.split()
             args = [] if len(cmds) <= 1 else cmds[1:]
             func = cmds[0]
@@ -387,4 +387,4 @@ def repl(self):
 
             except Warning as warning:
                 print(f"\n{warning}")
-                input("[Enter] ") 
+                input("[Enter] ")