Exit gracefully on KeyboardInterrupt

Signed-off-by: Zack Cerza <zack.cerza@inktank.com>

diff --git a/teuthology/beanstalk.py b/teuthology/beanstalk.py
index bcc4ceb28bbd7c654b995db3bf82912a342cb243..d76d914f5b7445f1ec6ab1df362524b5e961ac06 100644 (file)
--- a/teuthology/beanstalk.py
+++ b/teuthology/beanstalk.py
@@ -65,6 +65,10 @@ def main(args):
     machine_type = args['--machine_type']
     delete = args['--delete']
     show_desc = args['--description']
-    beanstalk = beanstalk_connect(machine_type)
-    walk_jobs(beanstalk, machine_type, show_desc=show_desc, delete=delete)
-    beanstalk.close()
+    try:
+        beanstalk = beanstalk_connect(machine_type)
+        walk_jobs(beanstalk, machine_type, show_desc=show_desc, delete=delete)
+    except KeyboardInterrupt:
+        log.info("Interrupted.")
+    finally:
+        beanstalk.close()