Требуется ли использовать yield в контекстном менеджере в Python

Yield в контекстном менеджере

Да, использование yield в контекстных менеджерах очень удобно и часто является предпочтительным подходом. Рассмотрим, как это работает и когда это применяется.

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

В Python есть два способа создания контекстных менеджеров:

1. Класс с методами __enter__ и __exit__ — классический подход
2. Функция-генератор с декоратором @contextmanager — современный и удобный подход

Декоратор @contextmanager из модуля contextlib позволяет превратить функцию-генератор в контекстный менеджер. yield разделяет код на две части:

• До yield: выполняется при входе в блок with (аналог __enter__)
• После yield: выполняется при выходе из блока with (аналог __exit__)

Пример с yield

from contextlib import contextmanager

@contextmanager
def database_connection(url):
    """Контекстный менеджер для работы с БД"""
    print(f"Подключение к {url}")
    connection = {"url": url, "connected": True}
    
    try:
        yield connection  # Передаём ресурс пользователю
    finally:
        # Это выполнится в любом случае
        print("Закрытие соединения")
        connection["connected"] = False

# Использование
with database_connection("postgres://localhost") as conn:
    print(f"Работаю с {conn}")
    # conn доступен здесь
print("Соединение закрыто")

Сравнение с классическим подходом

# Классический способ — через класс
class DatabaseConnection:
    def __init__(self, url):
        self.url = url
        self.connection = None
    
    def __enter__(self):
        print(f"Подключение к {self.url}")
        self.connection = {"url": self.url, "connected": True}
        return self.connection
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("Закрытие соединения")
        self.connection["connected"] = False
        return False  # Пробросить исключение дальше

# Использование
with DatabaseConnection("postgres://localhost") as conn:
    print(f"Работаю с {conn}")

Обработка исключений

yield позволяет обрабатывать исключения изнутри контекстного менеджера:

from contextlib import contextmanager

@contextmanager
def safe_division():
    try:
        yield
    except ZeroDivisionError:
        print("Ошибка: деление на ноль перехвачено")
        # Не пробросываем исключение дальше

with safe_division():
    result = 10 / 0  # Исключение будет обработано

print("Программа продолжает работу")

Когда использовать yield

✅ Используй @contextmanager с yield:

• Для простых контекстных менеджеров
• Когда логика setup/cleanup понятна в одном месте
• Для одноразовых менеджеров

✅ Используй класс с __enter__/__exit__:

• Для сложных контекстных менеджеров
• Когда нужна инициализация или состояние
• Когда переиспользуешь менеджер в разных местах

Выводы

Не требуется, но рекомендуется использовать yield в контекстных менеджерах, потому что:

1. Код чище — нет необходимости в отдельном классе
2. Логика明ясна — setup и cleanup в одном месте
3. Меньше кода — декоратор выполняет всю работу
4. Pythonic — это идиоматичный способ в Python

У вас есть выбор, но функции с @contextmanager и yield — это современный стандарт в Python сообществе.