Временно перенаправить stdout/stderr

Можно ли временно перенаправить stdout/stderr в Python (т.е. на время метода)?

Edit:

Проблема с текущими решениями (которую я сначала запомнила, но потом забыла) заключается в том, что они не перенаправляют; скорее, они просто полностью заменяют потоки. Следовательно, если метод имеет локальную копию одной переменной по любой причине (например, поскольку поток был передан как параметр чему-то), это не сработает.

Любые решения?

Ответы

Ответ 1

Чтобы решить проблему, что некоторые функции могут кэшировать поток 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() определены здесь.

Ответ 2

Вы также можете поместить логику перенаправления в contextmanager.

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

Ответ 3

Я не уверен, что означает временное перенаправление. Но вы можете переназначить такие потоки и reset обратно.

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

Также писать в sys.stderr в печатных строках, как это.

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

Обычная печать будет выводиться на стандартный вывод.

Ответ 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)

Но обратите внимание, что это не безопасно для потоков.

Ответ 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...

Ответ 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()

Контекст, почему я создал это, находится в этом сообщении в блоге.

Ответ 8

Мы будем использовать PHP синтаксис ob_start и ob_get_contents функций в Python3 и перенаправить ввод в файл.

Выходы сохраняются в файле, также может использоваться любой тип потока.

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

Будет печатать

Привет, Джон Bye John