fix: format KeyError messages cleanly

2026-01-08 06:51:37 +01:00
parent ecc33054af
commit a07cc02fb0
1 changed files with 4 additions and 1 deletions
--- a/skywipe/logger.py
+++ b/skywipe/logger.py
@@ -94,7 +94,10 @@ def get_logger() -> logging.Logger:


 def handle_error(error: Exception, logger: logging.Logger, exit_on_error: bool = False) -> None:
-    if isinstance(error, ValueError):
+    if isinstance(error, KeyError):
+        message = error.args[0] if error.args else str(error)
+        logger.error(f"{message}")
+    elif isinstance(error, ValueError):
        logger.error(f"{error}")
    else:
        logger.error(f"Unexpected error: {error}", exc_info=True)