临时重定向 stdout/stderr

问题描述：

是否可以在 Python 中临时重定向 stdout/stderr（即在方法运行期间）？

编辑：

当前解决方案的问题（我一开始记得，但后来忘了）是它们不重定向；相反，它们只是完全替换流。因此，如果某个方法出于某种原因（例如，因为流作为参数传递给某个对象）拥有某个变量的本地副本，它将无法工作。

有什么解决办法吗？

解决方案 1：

您还可以将重定向逻辑放入上下文管理器中。

import os
import sys

class RedirectStdStreams(object):
    def __init__(self, stdout=None, stderr=None):
        self._stdout = stdout or sys.stdout
        self._stderr = stderr or sys.stderr

    def __enter__(self):
        self.old_stdout, self.old_stderr = sys.stdout, sys.stderr
        self.old_stdout.flush(); self.old_stderr.flush()
        sys.stdout, sys.stderr = self._stdout, self._stderr

    def __exit__(self, exc_type, exc_value, traceback):
        self._stdout.flush(); self._stderr.flush()
        sys.stdout = self.old_stdout
        sys.stderr = self.old_stderr

if __name__ == '__main__':

    devnull = open(os.devnull, 'w')
    print('Fubar')

    with RedirectStdStreams(stdout=devnull, stderr=devnull):
        print("You'll never see me")

    print("I'm back!")

解决方案 2：

从python 3.4开始，有上下文管理器contextlib.redirect_stdout：

from contextlib import redirect_stdout

with open('yourfile.txt', 'w') as f:
    with redirect_stdout(f):
        # do stuff...

要完全抑制stdout这种现象，请执行以下操作：

from contextlib import redirect_stdout

with redirect_stdout(None):
    # do stuff...

解决方案 3：

sys.stdout为了解决某些函数可能将流缓存为局部变量，因此sys.stdout在该函数内替换全局变量不起作用的问题，您可以在文件描述符级别进行重定向（ sys.stdout.fileno()），例如：

from __future__ import print_function
import os
import sys

def some_function_with_cached_sys_stdout(stdout=sys.stdout):
    print('cached stdout', file=stdout)

with stdout_redirected(to=os.devnull), merged_stderr_stdout():
    print('stdout goes to devnull')
    some_function_with_cached_sys_stdout()
    print('stderr also goes to stdout that goes to devnull', file=sys.stderr)
print('stdout is back')
some_function_with_cached_sys_stdout()
print('stderr is back', file=sys.stderr)

stdout_redirected()将所有输出重定向sys.stdout.fileno()到给定的文件名、文件对象或文件描述符（os.devnull在示例中）。

stdout_redirected()并merged_stderr_stdout()在这里定义。

解决方案 4：

我不确定临时重定向是什么意思。但是，您可以像这样重新分配流并将其重置。

temp = sys.stdout
sys.stdout = sys.stderr
sys.stderr = temp

也可以像这样在打印语句中写入 sys.stderr。

 print >> sys.stderr, "Error in atexit._run_exitfuncs:"

常规打印将会输出到标准输出。

解决方案 5：

使用如下装饰器是可能的：

import sys

def redirect_stderr_stdout(stderr=sys.stderr, stdout=sys.stdout):
    def wrap(f):
        def newf(*args, **kwargs):
            old_stderr, old_stdout = sys.stderr, sys.stdout
            sys.stderr = stderr
            sys.stdout = stdout
            try:
                return f(*args, **kwargs)
            finally:
                sys.stderr, sys.stdout = old_stderr, old_stdout

        return newf
    return wrap

用途：

@redirect_stderr_stdout(some_logging_stream, the_console):
def fun(...):
    # whatever

或者，如果你不想修改源代码fun，请直接调用它

redirect_stderr_stdout(some_logging_stream, the_console)(fun)

但请注意，这不是线程安全的。

解决方案 6：

这是我发现有用的上下文管理器。它的优点在于你可以将它与with语句一起使用，并且它还可以处理子进程的重定向。

import contextlib


@contextlib.contextmanager
def stdchannel_redirected(stdchannel, dest_filename):
    """
    A context manager to temporarily redirect stdout or stderr

    e.g.:

    with stdchannel_redirected(sys.stderr, os.devnull):
        ...
    """

    try:
        oldstdchannel = os.dup(stdchannel.fileno())
        dest_file = open(dest_filename, 'w')
        os.dup2(dest_file.fileno(), stdchannel.fileno())

        yield
    finally:
        if oldstdchannel is not None:
            os.dup2(oldstdchannel, stdchannel.fileno())
        if dest_file is not None:
            dest_file.close()

我创建这个的原因可以在这篇博文中找到。

解决方案 7：

Raymond Hettinger向我们展示了一种更好的方法[1]：

import sys
with open(filepath + filename, "w") as f: #replace filepath & filename
    with f as sys.stdout:
        print("print this to file")   #will be written to filename & -path

在 with 块之后，sys.stdout 将被重置

[1]： http: //www.youtube.com/watch? v=OSGv2VnC0go&list=PLQZM27HgcgT-6D0w6arhnGdSHDcSmQ8r3

解决方案 8：

查看contextlib.redirect_stdout(new_target)和contextlib.redirect_stderr(new_target)。redirect_stderr是 Python 3.5 中的新功能。

解决方案 9：

我们将使用python3 中的ob_start和ob_get_contentsPHP函数的语法，并将输入重定向到文件中。

输出被存储在文件中，也可以使用任何类型的流。

from functools import partial
output_buffer = None
print_orig = print
def ob_start(fname="print.txt"):
    global print
    global output_buffer
    print = partial(print_orig, file=output_buffer)
    output_buffer = open(fname, 'w')
def ob_end():
    global output_buffer
    close(output_buffer)
    print = print_orig
def ob_get_contents(fname="print.txt"):
    return open(fname, 'r').read()

用法：

print ("Hi John")
ob_start()
print ("Hi John")
ob_end()
print (ob_get_contents().replace("Hi", "Bye"))

会打印

嗨，约翰，再见，约翰