Как использовать оператор `with` для подавления` sys.stdout` или `sys.stderr`? - PullRequest
Новая
       62

Как использовать оператор `with` для подавления` sys.stdout` или `sys.stderr`?

0 голосов
/ 05 июня 2018

Я пытаюсь использовать оператор with для подавления sys.stdout или sys.stderr по отдельности. Я нашел учебник , который не работал.Я использую Python 3.6.4, и я думаю, что учебник является некоторой версией Python 2.

Я посмотрел на SO и нашел несколько, но с приложениями, которые не работали или не подходили для этой ситуации.

Это не относится: Подпроцесс Python supress stdout и stderr

Не удалось заставить работать ни один из операторов with: Suppress stdout /Печать stderr из функций Python

Это для fortran: Перенаправление вывода FORTRAN (вызывается через F2PY) в Python 

from contextlib import contextmanager
@contextmanager
def suppress_console(file=sys.stdout):
    with open(os.devnull, "w") as devnull:
        old_file = file
        file = devnull
        try:  
            yield
        finally:
            file = old_file

with suppress_console():
    print(1, file=sys.stdout)
# 1

Ответы [ 2 ]

0 голосов
/ 05 июня 2018

Я использую следующее: 

from contextlib import redirect_stdout, contextmanager
import os


@contextmanager
def suppress():
    with open(os.devnull, "w") as null:
        with redirect_stdout(null):
            yield

тест: 

print("qwer")
with suppress():
    print("asdf")
print("ghjk")
# qwer
# ghjk

обновление

Лучше: 

from contextlib import redirect_stdout, redirect_stderr, contextmanager, ExitStack
import os

@contextmanager
def suppress(out=True, err=False):
    with ExitStack() as stack:
        with open(os.devnull, "w") as null:
            if out:
                stack.enter_context(redirect_stdout(null))
            if err:
                stack.enter_context(redirect_stderr(null))
            yield
0 голосов
/ 05 июня 2018

Я использую что-то вроде этого: 

class Suppress:
    def __init__(self, *, suppress_stdout=False, suppress_stderr=False):
        self.suppress_stdout = suppress_stdout
        self.suppress_stderr = suppress_stderr
        self.original_stdout = None
        self.original_stderr = None

    def __enter__(self):
        import sys, os
        devnull = open(os.devnull, "w")

        # Suppress streams
        if self.suppress_stdout:
            self.original_stdout = sys.stdout
            sys.stdout = devnull

        if self.suppress_stderr:
            self.original_stderr = sys.stderr
            sys.stderr = devnull

    def __exit__(self, *args, **kwargs):
        import sys
        # Restore streams
        if self.suppress_stdout:
            sys.stdout = self.original_stdout

        if self.suppress_stderr:
            sys.stderr = self.original_stderr

Пример: 

import sys
print("Before")
with Suppress(suppress_stdout=True):
    print("Inside")
print("After")

print("Before", file=sys.stderr)
with Suppress(suppress_stderr=True):
    print("Inside", file=sys.stderr)
print("After", file=sys.stderr)

Вывод : 

Before
After
Before
After

Примечания:

  • Я поместил свой импорт в методы чистоты, но обычно это было бы внутри файла модуля с импортом вверху.
  • Подавление stderr рискованно, особенно потому, чтоспособа я (не) обрабатываю исключения в методе __exit__.Вы можете заняться созданием более надежного метода выхода.
...