抑制模块调用外部库的输出

Question

使用机器学习库 PyML 时，我遇到了一个恼人的问题。 PyML 使用 libsvm 来训练 SVM 分类器。问题是 libsvm 将一些文本输出到标准输出。但因为它在 Python 之外，所以我无法拦截它。我尝试使用问题 在 Python 中静默函数的 stdout，而不破坏 sys.stdout 并恢复每个函数调用，但这些都没有帮助。

有什么办法可以做到这一点。修改 PyML 不是一个选择。

Answer 1

戴夫·史密斯（Dave Smith）在他的 博客。基本上，它很好地包装了 Ignacio 的答案：

def suppress_stdout():
    with open(os.devnull, "w") as devnull:
        old_stdout = sys.stdout
        sys.stdout = devnull
        try:  
            yield
        finally:
            sys.stdout = old_stdout

现在，您可以将任何将不需要的噪音混淆到标准输出中的函数，如下所示：

print "You can see this"
with suppress_stdout():
    print "You cannot see this"
print "And you can see this again"

对于 Python 3，您可以使用：

from contextlib import contextmanager
import os
import sys

@contextmanager
def suppress_stdout():
    with open(os.devnull, "w") as devnull:
        old_stdout = sys.stdout
        sys.stdout = devnull
        try:  
            yield
        finally:
            sys.stdout = old_stdout

Answer 2

我在 portaudio/PyAudio 初始化时遇到了类似的问题。我从里德的答案开始，它有效。尽管我需要重定向 stderr。因此，这是一个更新的跨平台版本，可以重定向两者：

import sys, os

# hide diagnostic output
with open(os.devnull, 'w') as devnull:
    # suppress stdout
    orig_stdout_fno = os.dup(sys.stdout.fileno())
    os.dup2(devnull.fileno(), 1)
    # suppress stderr
    orig_stderr_fno = os.dup(sys.stderr.fileno())
    os.dup2(devnull.fileno(), 2)

    print('*** stdout should be hidden!  ****')
    print('*** stderr should be too!  ****',
          file=sys.stderr)

    os.dup2(orig_stdout_fno, 1)  # restore stdout
    os.dup2(orig_stderr_fno, 2)  # restore stderr

print('done.')

应该很容易注释掉不需要的部分。

编辑：这些可能会有所帮助，目前没有时间查看：

https://docs.python.org/3/library/contextlib.html#contextlib.redirect_stdout

Answer 3

我遇到了同样的问题并像这样修复了它：

from cStringIO import StringIO

def wrapped_svm_predict(*args):
    """Run :func:`svm_predict` with no *stdout* output."""
    so, sys.stdout = sys.stdout, StringIO()
    ret = svm_predict(*args)
    sys.stdout = so
    return ret

Answer 4

打开/dev/null进行写入，使用os.dup()复制stdout，使用os.dup2()复制你的打开/dev/null 到标准输出。之后使用 os.dup2() 将复制的标准输出复制回真实的标准输出。

devnull = open('/dev/null', 'w')
oldstdout_fno = os.dup(sys.stdout.fileno())
os.dup2(devnull.fileno(), 1)
makesomenoise()
os.dup2(oldstdout_fno, 1)