75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
"""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()
|