Function log_context

logging_utils/core/logger.py:57–91

Context manager for source switching (simplified - no tree tracking). Usage: with log_context("Processing request", logger): logger.info("Step 1") With silent=True, no header is logged: with log_context("", logger, silent=True): logger.info("Som

(
    name: str,
    logger: Optional[logging.Logger] = None,
    source: Optional[str] = None,
    silent: bool = False,
)

@contextmanager
def log_context(
    name: str,
    logger: Optional[logging.Logger] = None,
    source: Optional[str] = None,
    silent: bool = False,
) -> Generator[None, None, None]:
    """
    Context manager for source switching (simplified - no tree tracking).

    Usage:
        with log_context("Processing request", logger):
            logger.info("Step 1")

    With silent=True, no header is logged:
        with log_context("", logger, silent=True):
            logger.info("Some work")
    """
    if logger is None:
        logger = logging.getLogger()

    # Handle optional source change
    source_token = None
    if source is not None:
        source_token = source_var.set(source)

    # Log the context entry (unless silent)
    if not silent and name:
        logger.info(name)

    try:
        yield
    finally:
        # Restore previous source
        if source_token is not None:
            source_var.reset(source_token)


@contextmanager