log Exception information

xiuqhou · Aug 1, 2024 · 608f210 · 608f210
1 parent d24ab49
commit 608f210
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/util/logger.py b/util/logger.py
@@ -67,6 +67,18 @@ def filter(self, record):
         return True
 
 
+def handle_exception(exc_type, exc_value, exc_traceback):
+    # Catch exception in logger
+    logger = logging.getLogger(os.path.basename(os.getcwd()) + "." + __name__)
+    if issubclass(exc_type, KeyboardInterrupt):
+        sys.__excepthook__(exc_type, exc_value, exc_traceback)
+        return
+    logger.error("Uncaught exception", exc_info=(exc_type, exc_value, exc_traceback))
+
+
+sys.excepthook = handle_exception
+
+
 @functools.lru_cache()  # so that calling setup_logger multiple times won't add many handlers
 def setup_logger(
     output=None,