0

我试图用一个with声明来压制sys.stdout或sys.stderr单独。 我找到了一个不起作用的教程。我正在使用Python 3.6.4，我认为本教程是Python 2.

我在 SO 上查找了它，发现了一些但应用程序不起作用或不适用于这种情况。

这不适用：Python subprocess supress stdout 和 stderr

无法使任何with语句起作用： 从 Python 函数中抑制 stdout / stderr 打印

这适用于 fortran：在 Python 中重定向 FORTRAN（通过 F2PY 调用）输出

from contextlib import contextmanager
@contextmanager
def suppress_console(file=sys.stdout):
    with open(os.devnull, "w") as devnull:
        old_file = file
        file = devnull
        try:  
            yield
        finally:
            file = old_file

with suppress_console():
    print(1, file=sys.stdout)
# 1
4

2 回答 2

4

我使用以下一个：

from contextlib import redirect_stdout, contextmanager
import os


@contextmanager
def suppress():
    with open(os.devnull, "w") as null:
        with redirect_stdout(null):
            yield

测试：

print("qwer")
with suppress():
    print("asdf")
print("ghjk")
# qwer
# ghjk

更新

一个更好的：

from contextlib import redirect_stdout, redirect_stderr, contextmanager, ExitStack
import os

@contextmanager
def suppress(out=True, err=False):
    with ExitStack() as stack:
        with open(os.devnull, "w") as null:
            if out:
                stack.enter_context(redirect_stdout(null))
            if err:
                stack.enter_context(redirect_stderr(null))
            yield
于 2018-06-05T03:42:46.090 回答
2

我使用这样的东西：

class Suppress:
    def __init__(self, *, suppress_stdout=False, suppress_stderr=False):
        self.suppress_stdout = suppress_stdout
        self.suppress_stderr = suppress_stderr
        self.original_stdout = None
        self.original_stderr = None

    def __enter__(self):
        import sys, os
        devnull = open(os.devnull, "w")

        # Suppress streams
        if self.suppress_stdout:
            self.original_stdout = sys.stdout
            sys.stdout = devnull

        if self.suppress_stderr:
            self.original_stderr = sys.stderr
            sys.stderr = devnull

    def __exit__(self, *args, **kwargs):
        import sys
        # Restore streams
        if self.suppress_stdout:
            sys.stdout = self.original_stdout

        if self.suppress_stderr:
            sys.stderr = self.original_stderr

例子：

import sys
print("Before")
with Suppress(suppress_stdout=True):
    print("Inside")
print("After")

print("Before", file=sys.stderr)
with Suppress(suppress_stderr=True):
    print("Inside", file=sys.stderr)
print("After", file=sys.stderr)

输出：

前
后
前
后

笔记：

  • 我将我的导入放在方法中以保持清洁，但通常这将位于顶部带有导入的模块文件中。
  • 抑制 stderr 是有风险的，特别是因为我（不）处理方法中的异常的__exit__方式。您可能会考虑制定更强大的退出方法。
于 2018-06-05T03:38:22.263 回答