🧲 Как «поймать» print() в Python: перехватываем stdout без сторонних библиотек

В Python можно перехватывать то, что выводится на экран через print().
Это полезно для тестирования, отладки, логирования или подавления «шумного» вывода.
Я покажу, как сделать это элегантно — используя только стандартную библиотеку и несколько строк кода.

---

🧩 Понимание контекстных менеджеров в Python

Что такое контекстный менеджер?

Контекстный менеджер управляет ресурсами безопасно и аккуратно.
Он определяет инициализацию и очистку вокруг блока кода.

Пример: работа с файлами

with open("file.txt", "w") as f:
    f.write("Привет, мир!")

✅ Автоматическая очистка: файл будет закрыт даже при возникновении исключения.

---

Зачем нужны контекстные менеджеры

• Автоматическая очистка — ресурсы освобождаются сами
• Безопасность при ошибках — очистка выполняется даже при исключениях
• Читаемый код — логика setup/teardown в одном месте
• Повторное использование — подходит для файлов, блокировок, сетевых соединений или stdout

---

Использование @contextmanager из contextlib

Декоратор @contextmanager позволяет создавать контекстные менеджеры через генераторы:

from contextlib import contextmanager

@contextmanager
def my_context():
    print("Входим в контекст")
    resource = "некоторый ресурс"
    try:
        yield resource
    finally:
        print("Выходим из контекста")

with my_context() as res:
    print("Внутри контекста:", res)

Вывод:

Входим в контекст
Внутри контекста: некоторый ресурс
Выходим из контекста

💡 Объяснение:

• Код до yield → выполняется при входе
• Значение после yield → доступно через as res
• Код после yield → выполняется при выходе, даже если возникло исключение

Именно это делает возможным временное изменение глобального состояния, например перехват stdout.

---

⚡ Краткая таблица: контекстные менеджеры для stdout/stderr

Задача / Сценарий	Рекомендуемый менеджер
Перехват только print	capture_stdout
Перехват только ошибок/исключений	capture_stderr
Перехват stdout и stderr одновременно	capture_output
Перехват с одновременным отображением	capture_stdout_tee / capture_output_tee

---

📦 Перехват stdout

from contextlib import contextmanager
import sys, io

@contextmanager
def capture_stdout():
    old_stdout = sys.stdout
    sys.stdout = buffer = io.StringIO()
    try:
        yield buffer
    finally:
        sys.stdout = old_stdout

Пример использования:

with capture_stdout() as out:
    print("Этот вывод был перехвачен")

print("Перехвачено:", out.getvalue())

---

🔧 Перехват stderr или обоих потоков

Перехват stderr

@contextmanager
def capture_stderr():
    old_stderr = sys.stderr
    sys.stderr = buffer = io.StringIO()
    try:
        yield buffer
    finally:
        sys.stderr = old_stderr

Перехват stdout + stderr

@contextmanager
def capture_output():
    old_stdout = sys.stdout
    old_stderr = sys.stderr
    sys.stdout = sys.stderr = buffer = io.StringIO()
    try:
        yield buffer
    finally:
        sys.stdout = old_stdout
        sys.stderr = old_stderr

---

🔀 Режим Tee: перехват и отображение одновременно

class Tee(io.StringIO):
    def __init__(self, original):
        super().__init__()
        self.original = original

    def write(self, text):
        self.original.write(text)
        super().write(text)

    def flush(self):
        self.original.flush()

Контекстные менеджеры:

@contextmanager
def capture_stdout_tee():
    old_stdout = sys.stdout
    sys.stdout = tee = Tee(old_stdout)
    try:
        yield tee
    finally:
        sys.stdout = old_stdout

@contextmanager
def capture_output_tee():
    old_stdout = sys.stdout
    old_stderr = sys.stderr
    sys.stdout = sys.stderr = tee = Tee(old_stdout)
    try:
        yield tee
    finally:
        sys.stdout = old_stdout
        sys.stderr = old_stderr

Пример:

with capture_stdout_tee() as out:
    print("Видно и перехвачено")

print("Содержимое Tee:", out.getvalue())

---

📦 Полный файл: stdout_capture.py

"""
Утилита для перехвата stdout и stderr с помощью контекстных менеджеров.
Поддерживает:
- перехват stdout
- перехват stderr
- объединённый перехват
- режим Tee
"""

import sys, io
from contextlib import contextmanager

@contextmanager
def capture_stdout():
    old_stdout = sys.stdout
    sys.stdout = buffer = io.StringIO()
    try: yield buffer
    finally: sys.stdout = old_stdout

@contextmanager
def capture_stderr():
    old_stderr = sys.stderr
    sys.stderr = buffer = io.StringIO()
    try: yield buffer
    finally: sys.stderr = old_stderr

@contextmanager
def capture_output():
    old_stdout = sys.stdout
    old_stderr = sys.stderr
    sys.stdout = sys.stderr = buffer = io.StringIO()
    try: yield buffer
    finally:
        sys.stdout = old_stdout
        sys.stderr = old_stderr

class Tee(io.StringIO):
    def __init__(self, original):
        super().__init__()
        self.original = original
    def write(self, text):
        self.original.write(text)
        super().write(text)
    def flush(self):
        self.original.flush()

@contextmanager
def capture_stdout_tee():
    old_stdout = sys.stdout
    sys.stdout = tee = Tee(old_stdout)
    try: yield tee
    finally: sys.stdout = old_stdout

@contextmanager
def capture_output_tee():
    old_stdout = sys.stdout
    old_stderr = sys.stderr
    sys.stdout = sys.stderr = tee = Tee(old_stdout)
    try: yield tee
    finally:
        sys.stdout = old_stdout
        sys.stderr = old_stderr

---

✅ Примеры использования

from stdout_capture import capture_stdout, capture_output, capture_stdout_tee

def test_func():
    print("Привет из функции")

# Перехват stdout
with capture_stdout() as out:
    test_func()
print("Перехвачено stdout:", out.getvalue())

# Перехват stdout + stderr
try:
    with capture_output() as out:
        print("Что-то")
        raise Exception("Ой!")
except Exception:
    pass
print("Перехвачено output:", out.getvalue())

# Перехват и отображение одновременно
with capture_stdout_tee() as out:
    print("Видно И перехвачено")
print("Содержимое Tee:", out.getvalue())

---

⚠️ Практические советы и подводные камни

1️⃣ Большой вывод

• io.StringIO может занимать много памяти при больших объемах.
• Используйте временный файл:

import sys, tempfile

with tempfile.TemporaryFile(mode='w+') as tmp:
    old_stdout = sys.stdout
    sys.stdout = tmp
    try:
        print("Много данных...")
        tmp.seek(0)
        captured = tmp.read()
    finally:
        sys.stdout = old_stdout

2️⃣ Вывод из подпроцессов

• capture_stdout() не перехватывает вывод подпроцессов.
• Используйте subprocess.PIPE:

import subprocess
result = subprocess.run(["ls", "-la"], capture_output=True, text=True)
print(result.stdout)

3️⃣ Потокобезопасность

• sys.stdout и sys.stderr глобальные → избегайте одновременного перехвата в нескольких потоках.

4️⃣ Когда использовать Tee

• Отладка функций с логированием
• Интерактивные CLI-инструменты
• Интеграционные тесты с отображением в терминале

5️⃣ Восстановление потоков

• Всегда восстанавливайте sys.stdout/sys.stderr в finally

6️⃣ Избегайте слишком глубокой вложенности

• Несколько вложенных перехватов могут смешивать вывод. Используйте один контекст на сценарий.

7️⃣ Лучшие практики тестирования

• Unit-тесты → capture_stdout() или capture_output()
• Ручная отладка → режим Tee
• Проверяйте содержимое буфера, а не побочные эффекты