import logging
import sys

import structlog


def setup_logging(level: str) -> None:
  logging.basicConfig(
    format="%(message)s",
    stream=sys.stdout,
    level=getattr(logging, level.upper(), logging.INFO),
  )
  structlog.configure(
    processors=[
      structlog.contextvars.merge_contextvars,
      structlog.processors.add_log_level,
      structlog.processors.TimeStamper(fmt="iso"),
      structlog.processors.StackInfoRenderer(),
      structlog.processors.format_exc_info,
      structlog.processors.JSONRenderer(),
    ],
    wrapper_class=structlog.make_filtering_bound_logger(
      getattr(logging, level.upper(), logging.INFO)
    ),
    context_class=dict,
    logger_factory=structlog.PrintLoggerFactory(),
    cache_logger_on_first_use=True,
  )