Handling Ctrl+C/Ctrl+D a bit more gracefully

1 files changed, 4 insertions, 1 deletions

diff --git a/ql/console.py b/ql/console.py
index 6751b83..f797da7 100755
--- a/ql/console.py
+++ b/ql/console.py
@@ -16,7 +16,10 @@ class Console(object):
 

     def run(self):

         for table in self.tables:

-            self.process_table(table)

+            try:

+                self.process_table(table)

+            except (EOFError, KeyboardInterrupt):

+                break

 

     def process_table(self, table):

         lineup = self.get_lineup(table)