暂时重定向stdout/stderr

Question

是否可以在Python中暂时重定向stdout/stderr(即在方法的持续时间内)？

编辑:

当前解决方案(我最初记得但后来忘记了)的问题在于它们没有重定向 ; 相反,他们只是完整地替换了流.因此,如果一个方法有一个本地副本,一个变量的任何原因(例如,由于流被作为参数传递的东西),它不会工作.

有解决方案吗

Answer 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!")

Answer 2

我不确定临时重定向是什么意思.但是,您可以重新分配这样的流并重新设置它.

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

还要像这样在print stmts中写入sys.stderr.

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

定期打印将stdout.

Answer 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()在这里定义.

Answer 4

可以使用以下装饰器:

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)

但请注意,这不是线程安全的.

Answer 5

从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...