Python特定打印语句包装器实现：忽略指定输出与错误信息

你可以改造原有的过滤器类，支持多字符串匹配，同时覆盖标准输出和标准错误流，完全替代原有的shell管道方案，实现更灵活的输出过滤。

改造后的过滤器实现 #

import sys
from contextlib import redirect_stdout, redirect_stderr
from functools import wraps

class MultiPrintFilter:
    def __init__(self, stream, block_substrings):
        self.stream = stream
        # 传入需要过滤的字符串列表
        self.block_substrings = block_substrings

    def write(self, txt):
        # 只要输出包含任意一个过滤字符串就丢弃
        if not any(sub in txt for sub in self.block_substrings):
            self.stream.write(txt)
    
    # 兼容流的flush方法，避免输出缓冲异常
    def flush(self):
        self.stream.flush()

使用方式 #

方式1：上下文管理器套入主函数 #

直接在程序入口处用上下文管理器包裹原有主函数调用即可，无需改动原有业务逻辑：

# 此处配置所有需要过滤的字符串
BLOCK_LIST = [
    "first-string-to-ignore",
    "second-string-to-ignore",
    "third-party-lib-warning"
]

# 你的原有主函数，不需要做任何修改
def main():
    # 原有业务逻辑、第三方库调用都写在这里
    pass

if __name__ == "__main__":
    out_filter = MultiPrintFilter(sys.stdout, BLOCK_LIST)
    err_filter = MultiPrintFilter(sys.stderr, BLOCK_LIST)
    # 同时重定向标准输出和标准错误流，完全模拟原shell的2>&1合并过滤效果
    with redirect_stdout(out_filter), redirect_stderr(err_filter):
        main()

方式2：装饰器封装（更简洁） #

如果希望更简洁的调用，可以封装为装饰器，直接修饰主函数：

def filter_output(block_substrings):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            out_filter = MultiPrintFilter(sys.stdout, block_substrings)
            err_filter = MultiPrintFilter(sys.stderr, block_substrings)
            with redirect_stdout(out_filter), redirect_stderr(err_filter):
                return func(*args, **kwargs)
        return wrapper
    return decorator

# 配置过滤列表
BLOCK_LIST = ["ignore-1", "ignore-2", "ignore-3"]

# 直接给主函数加装饰器即可，无需修改入口代码
@filter_output(BLOCK_LIST)
def main():
    # 原有业务逻辑
    pass

if __name__ == "__main__":
    main()

扩展说明 #

如果需要更复杂的匹配规则（比如正则匹配），只需要修改MultiPrintFilter的write方法里的判断逻辑即可，无需调整其他部分代码。该方案可以覆盖绝大多数第三方库通过标准流输出的内容过滤需求。

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