Custom log level mapping

Author: Artem Krivonos

awslambdaric/bootstrap.py

@@ -87,7 +87,7 @@ def make_error(
         "timestamp": time.strftime(
             DATETIME_FORMAT, logging.Formatter.converter(time.time())
         ),
-        "level": level,
+        "log_level": level,
         "logger": RUNTIME_ERROR_LOGGER_NAME,
         "errorMessage": error_message if error_message else "",
         "errorType": error_type if error_type else "",
@@ -114,8 +114,11 @@ def log_error(error_result, log_sink):
 
 
 def log_error_json(error_result, log_sink):
+    log_level = error_result.pop("log_level", logging.ERROR)
+    error_result["level"] = logging.getLevelName(log_level)
     log_sink.log_error(
-        [to_json(error_result)], log_level=error_result.get("level", logging.ERROR)
+        [to_json(error_result)],
+        log_level=log_level,
     )
 
 
@@ -147,7 +150,7 @@ def log_error_text(error_result, log_sink):
                     ]
 
     log_sink.log_error(
-        error_message_lines, log_level=error_result.get("level", logging.ERROR)
+        error_message_lines, log_level=error_result.get("log_level", logging.ERROR)
     )
 
 

awslambdaric/lambda_runtime_log_utils.py

@@ -77,7 +77,13 @@ def format_location(record: logging.LogRecord):
 
         return f"{record.pathname}:{record.funcName}:{record.lineno}"
 
+    @staticmethod
+    def format_log_level(record: logging.LogRecord):
+        level = record.levelno % 51 // 10 * 10
+        record.levelname = logging.getLevelName(level)
+
     def format(self, record: logging.LogRecord) -> str:
+        self.format_log_level(record)
         record.log_format = LambdaLogFormat.JSON
 
         result = {

tests/test_bootstrap.py

@@ -399,7 +399,7 @@ def raise_exception_handler(json_input, lambda_context):
                 )
 
         expected_response = {
-            "level": 50,
+            "log_level": 50,
             "logger": "system",
             "errorType": "FaultExceptionType",
             "errorMessage": "Fault exception msg",