是否可以在Python中临时重定向stdout / stderr（即在方法的持续时间内）？
当前解决方案（我最初记得但后来忘记了）的问题在于它们没有重定向;相反，他们只是完整地替换了流。因此，如果一个方法由于任何原因而具有一个变量的本地副本（例如，因为该流作为参数传递给某个东西），它将不起作用。
有解决方案吗
要解决某些函数可能已将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() and merged_stderr_stdout() are defined here。
您还可以将重定向逻辑放在上下文管理器中。
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!")
我不确定临时重定向是什么意思。但是，您可以重新分配这样的流并重新设置它。
temp = sys.stdout
sys.stdout = sys.stderr
sys.stderr = temp
还要像这样在print stmts中写入sys.stderr。
print >> sys.stderr, "Error in atexit._run_exitfuncs:"
定期打印将stdout。
可以使用以下装饰器：
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)
但请注意，这不是线程安全的。
从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...
这是我发现有用的上下文管理器。关于这一点的好处是你可以将它与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()
我为什么创建这个的背景是在this blog post。
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
我们将在python3中使用PHP和ob_start函数的ob_get_contents语法，并将输入重定向到文件中。
输出存储在文件中，也可以使用任何类型的流。
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"))
会打印
嗨John Bye John