Introduce LOGGING_OVERRIDE config var

simonrw · simonrw · commit 3dd89f1a974d · 2024-05-13T16:03:48.000+01:00
Co-Authored-By: daniel.fangl@localstack.cloud
diff --git a/localstack/config.py b/localstack/config.py
@@ -418,6 +418,9 @@ def in_docker():
 LS_LOG = eval_log_type("LS_LOG")
 DEBUG = is_env_true("DEBUG") or LS_LOG in TRACE_LOG_LEVELS
 
+# allow setting custom log levels for individual loggers
+LOGGING_OVERRIDE = os.environ.get("LOGGING_OVERRIDE", "{}")
+
 # whether to enable debugpy
 DEVELOP = is_env_true("DEVELOP")
 
diff --git a/localstack/logging/setup.py b/localstack/logging/setup.py
@@ -1,3 +1,4 @@
+import json
 import logging
 import sys
 import warnings
@@ -78,6 +79,23 @@ def setup_logging_from_config():
         for name, level in trace_internal_log_levels.items():
             logging.getLogger(name).setLevel(level)
 
+    if raw_value := config.LOGGING_OVERRIDE:
+        try:
+            logging_overrides = json.loads(raw_value)
+            for logger, level_name in logging_overrides.items():
+                level = getattr(logging, level_name, None)
+                if not level:
+                    raise RuntimeError(
+                        f"Failed to configure logging overrides ({raw_value}): '{level_name}' is not a valid log level"
+                    )
+                logging.getLogger(logger).setLevel(level)
+        except RuntimeError:
+            raise
+        except Exception as e:
+            raise RuntimeError(
+                f"Failed to configure logging overrides ({raw_value}): Malformed value. ({e})"
+            ) from e
+
 
 def create_default_handler(log_level: int):
     log_handler = logging.StreamHandler(stream=sys.stderr)
