Clean up main loop, ctrl+c or ctrl+d to exit

1 files changed, 7 insertions, 1 deletions

diff --git a/kasus.py b/kasus.py
index 68b0be0..57ebc4a 100644
--- a/kasus.py
+++ b/kasus.py
@@ -15,10 +15,16 @@ while True:
     total += 1
     print(f"{Style.BRIGHT}--- Übung {total} ---{Style.NORMAL}")
     print(parameters)
+    print()
     print(queryString)
     print()
 
-    response = input("> ")
+    try:
+        response = input("> ")
+    except (KeyboardInterrupt, EOFError):
+        # TODO print statistics
+        print("\nBis später!")
+        break
 
     print(f"{Style.BRIGHT}", end="")