claude final draft

shared/logging.py

@@ -0,0 +1,74 @@
+"""Structured logging configuration."""
+
+import logging
+import sys
+from typing import Any
+
+import structlog
+
+
+def setup_logging(
+    service_name: str,
+    log_level: str = "INFO",
+    log_format: str = "json",
+) -> structlog.BoundLogger:
+    """
+    Configure structured logging for a service.
+
+    Args:
+        service_name: Name of the service for log context
+        log_level: Logging level (DEBUG, INFO, WARNING, ERROR)
+        log_format: Output format ("json" or "console")
+
+    Returns:
+        Configured structlog logger
+    """
+
+    # Shared processors
+    shared_processors: list[Any] = [
+        structlog.contextvars.merge_contextvars,
+        structlog.processors.add_log_level,
+        structlog.processors.TimeStamper(fmt="iso"),
+        structlog.processors.StackInfoRenderer(),
+    ]
+
+    if log_format == "json":
+        # JSON format for production
+        processors = shared_processors + [
+            structlog.processors.format_exc_info,
+            structlog.processors.JSONRenderer(),
+        ]
+    else:
+        # Console format for development
+        processors = shared_processors + [
+            structlog.dev.ConsoleRenderer(colors=True),
+        ]
+
+    structlog.configure(
+        processors=processors,
+        wrapper_class=structlog.make_filtering_bound_logger(
+            getattr(logging, log_level.upper(), logging.INFO)
+        ),
+        context_class=dict,
+        logger_factory=structlog.PrintLoggerFactory(),
+        cache_logger_on_first_use=True,
+    )
+
+    # Also configure standard library logging
+    logging.basicConfig(
+        format="%(message)s",
+        stream=sys.stdout,
+        level=getattr(logging, log_level.upper(), logging.INFO),
+    )
+
+    # Get logger with service context
+    logger = structlog.get_logger(service=service_name)
+
+    return logger
+
+
+def get_logger(name: str | None = None) -> structlog.BoundLogger:
+    """Get a logger instance, optionally with a specific name."""
+    if name:
+        return structlog.get_logger(component=name)
+    return structlog.get_logger()