如何临时抑制 structlog 日志输出

from contextlib import contextmanager
from structlog.testing import capture_logs

@contextmanager
def suppress_logging():
    """
    一个上下文管理器，用于临时抑制 structlog 的日志输出。
    在其作用域内，所有 structlog 生成的日志事件都将被捕获，
    但不会被输出到通常的日志处理器。
    """
    with capture_logs():
        yield

import structlog
from contextlib import contextmanager
from structlog.testing import capture_logs

# 假设你的 structlog 已经配置好
# 例如：
structlog.configure(
    processors=[
        structlog.stdlib.add_logger_name,
        structlog.stdlib.add_log_level,
        structlog.processors.TimeStamper(fmt="iso"),
        structlog.dev.ConsoleRenderer()
    ],
    wrapper_class=structlog.stdlib.BoundLogger,
    logger_factory=structlog.stdlib.LoggerFactory(),
)
log = structlog.get_logger(__name__)

# 定义自定义的抑制日志上下文管理器
@contextmanager
def suppress_logging():
    with capture_logs():
        yield

def make_error_happen():
    """一个模拟会产生错误日志的函数。"""
    log.error("This is a deliberately provoked error!")
    log.warning("Something potentially problematic occurred.")
    return "Error handled."

def run_test():
    print("--- 正常日志输出区域开始 ---")
    log.info("Test setup: logging is active.")
    make_error_happen() # 此时日志会正常输出
    print("--- 正常日志输出区域结束 ---")

    print("\n--- 抑制日志输出区域开始 ---")
    log.info("Entering a block where logs should be suppressed.")
    with suppress_logging():
        log.info("Inside suppressed block: This info log will NOT be printed to console.")
        make_error_happen() # 此时日志将被抑制，不会输出到控制台
        log.debug("Inside suppressed block: This debug log will also NOT be printed.")
    log.info("Exited suppressed block: Logging is active again.")
    print("--- 抑制日志输出区域结束 ---")

    print("\n--- 再次正常日志输出区域开始 ---")
    make_error_happen() # 此时日志会再次正常输出
    log.info("Test teardown: logging remains active.")
    print("--- 再次正常日志输出区域结束 ---")

if __name__ == "__main__":
    run_test()