Verify log_level

dnstapir · Jan 15, 2025 · fa3bdc3 · fa3bdc3
1 parent dcc18a4
commit fa3bdc3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dnstapir/logging.py b/dnstapir/logging.py
@@ -7,6 +7,8 @@
 import structlog
 from structlog.types import EventDict, Processor
 
+VALID_LOG_LEVELS = set(["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"])
+
 
 def drop_color_message_key(_, __, event_dict: EventDict) -> EventDict:
     """
@@ -18,6 +20,11 @@ def drop_color_message_key(_, __, event_dict: EventDict) -> EventDict:
 
 
 def setup_logging(json_logs: bool = False, log_level: str = "INFO") -> None:
+    """Set up logging"""
+
+    if log_level.upper() not in VALID_LOG_LEVELS:
+        raise ValueError(f"Invalid log level: {log_level}")
+
     timestamper = structlog.processors.TimeStamper(fmt="iso")
 
     shared_processors: list[Processor] = [