Add exception hook

diff --git a/teuthology/run.py b/teuthology/run.py
index f3e44abae9f25e1953269c2fcf597638269805c5..1baa627748bbe9dc9f0fb374d58acfaf64f4260d 100644 (file)
--- a/teuthology/run.py
+++ b/teuthology/run.py
@@ -4,6 +4,9 @@ import yaml
 import StringIO
 import contextlib
 import logging
+import sys
+from traceback import format_tb
+
 
 def config_file(string):
     config = {}
@@ -103,6 +106,17 @@ def set_up_logging(ctx):
         handler.setFormatter(formatter)
         logging.getLogger().addHandler(handler)
 
+    install_except_hook()
+
+
+def install_except_hook():
+    def log_exception(exception_class, exception, traceback):
+        logging.critical(''.join(format_tb(traceback)))
+        logging.critical('{0}: {1}'.format(exception_class, exception))
+
+    sys.excepthook = log_exception
+
+
 def write_initial_metadata(ctx):
     if ctx.archive is not None:
         with file(os.path.join(ctx.archive, 'pid'), 'w') as f: