Python >=3.8 如何捕获并重抛代码中的警告（类似try/except机制）

要实现通用的警告捕获、自定义日志记录并重抛的功能，可以借助Python标准库的warnings模块，通过自定义警告处理器或上下文管理器达成类似try/except的使用体验。以下是两种可靠的实现方式：

方式一：全局警告处理器替换（推荐，简洁易用） #

这种方式通过临时替换默认警告处理器，让代码块内的所有警告先被自定义日志记录，再触发原有的警告行为（重抛），用法和try/except高度相似。

实现代码 #

import logging
import warnings

# 1. 配置自定义日志器
logger = logging.getLogger("custom_warning_logger")
logger.setLevel(logging.WARNING)
handler = logging.StreamHandler()
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
handler.setFormatter(formatter)
logger.addHandler(handler)

# 2. 实现上下文管理器
def catch_and_log_warnings(logger, filter_mode="always"):
    """
    捕获代码块内所有警告，用指定日志器记录后重抛
    :param logger: 自定义日志器实例
    :param filter_mode: 警告过滤模式，默认"always"确保所有警告被捕获
    """
    def custom_warning_handler(message, category, filename, lineno, file=None, line=None):
        # 记录警告详情到自定义日志
        logger.warning(f"{category.__name__}: {message} (来源: {filename}:{lineno})")
        # 调用默认处理器重抛警告
        warnings._showwarnmsg_impl(warnings.WarningMessage(message, category, filename, lineno, file, line))
    
    # 保存原始状态，确保上下文退出后恢复
    original_handler = warnings.showwarning
    original_filters = warnings.filters.copy()
    
    try:
        warnings.simplefilter(filter_mode)
        warnings.showwarning = custom_warning_handler
        yield
    finally:
        # 恢复默认处理器和过滤规则
        warnings.showwarning = original_handler
        warnings.filters = original_filters

# 3. 使用示例
if __name__ == "__main__":
    with catch_and_log_warnings(logger):
        # 执行可能产生警告的代码（包括第三方包调用）
        warnings.warn("第三方库抛出的测试警告", DeprecationWarning)
        # some_third_party_function()  # 替换为实际的第三方函数调用

方式二：记录并重抛指定警告（灵活可控） #

如果需要对捕获的警告做筛选（比如只处理特定类型），可以使用warnings.catch_warnings(record=True)记录所有警告，手动处理后选择性重抛。

实现代码 #

import logging
import warnings

# 配置自定义日志器（同上）
logger = logging.getLogger("custom_warning_logger")
logger.setLevel(logging.WARNING)
handler = logging.StreamHandler()
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
handler.setFormatter(formatter)
logger.addHandler(handler)

# 使用示例
if __name__ == "__main__":
    with warnings.catch_warnings(record=True) as caught_warnings:
        # 确保所有警告被记录（忽略默认过滤规则）
        warnings.simplefilter("always")
        
        # 执行产生警告的代码
        warnings.warn("测试警告1", UserWarning)
        warnings.warn("过时警告", DeprecationWarning)
        # some_third_party_function()
        
        # 遍历处理捕获的警告
        for warning in caught_warnings:
            # 记录到自定义日志
            logger.warning(f"{warning.category.__name__}: {warning.message} (来源: {warning.filename}:{warning.lineno})")
            # 重抛警告，stacklevel=2确保行号指向原警告位置
            warnings.warn(warning.message, warning.category, stacklevel=2)

关键注意事项 #

• 两种方式都能捕获第三方包通过warnings.warn()发出的警告，覆盖绝大多数场景。
• 设置filter_mode="always"或warnings.simplefilter("always")是为了绕过Python默认的警告过滤规则（比如DeprecationWarning默认不显示），确保所有警告被捕获。
• 上下文管理器会自动恢复原始的警告处理器和过滤规则，不会影响上下文外的代码行为。

内容的提问来源于stack exchange，提问作者Xxxo