Какую проблему решают менеджеры контекста?

Проблемы, которые решают менеджеры контекста

Менеджеры контекста (context managers) в Python решают одну критичную проблему: гарантировать очистку ресурсов, даже если произойдёт ошибка.

Основная проблема

Представь, что нужно работать с ресурсами (файлы, соединения с БД, блокировки, сессии HTTP). Без менеджеров контекста нужно вручную закрывать их:

# Плохо: нужно помнить о закрытии
file = open("data.txt", "r")
data = file.read()
file.close()  # Легко забыть!

Проблемы:

1. Забывчивость. Разработчик может забыть вызвать close()
2. Утечки ресурсов. Файл останется открытым, соединение в БД будет висеть
3. Исключения. Если возникнет ошибка ДО close(), ресурс не закроется

# Сломается: исключение до close()
file = open("data.txt", "r")
data = file.read()
process(data)  # Может выкинуть исключение
file.close()   # Никогда не выполнится!

Решение: менеджеры контекста

with statement гарантирует очистку:

# Хорошо: файл закроется даже при ошибке
with open("data.txt", "r") as file:
    data = file.read()
    process(data)  # Ошибка здесь
# file.close() вызовется автоматически

Менеджер контекста вызывает метод __exit__() независимо от того, была ошибка или нет.

Как работает менеджер контекста

class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None
    
    def __enter__(self):
        """Вызывается при входе в блок with"""
        print(f"Открываем файл {self.filename}")
        self.file = open(self.filename, self.mode)
        return self.file  # Это получит переменная после 'as'
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        """Вызывается при выходе из блока with (всегда!)"""
        print(f"Закрываем файл {self.filename}")
        if self.file:
            self.file.close()
        
        # Можем обработать исключение
        if exc_type is not None:
            print(f"Произошла ошибка: {exc_val}")
        
        # return True = подавить исключение
        # return False или None = пробросить исключение дальше
        return False

# Использование
with FileManager("data.txt", "r") as f:
    data = f.read()
    print(data)
    # Даже если здесь ошибка, __exit__ вызовется!

Реальные примеры из стандартной библиотеки

1. Работа с файлами:

with open("file.txt") as f:
    data = f.read()
# f закроется автоматически

2. Работа с БД:

from contextlib import contextmanager
from sqlalchemy.orm import Session

with Session(engine) as session:
    user = session.query(User).first()
    session.commit()
# Сессия закроется и откатит транзакцию при ошибке

3. Блокировки (threading):

import threading

lock = threading.Lock()

with lock:
    # Критичная секция
    shared_resource.update()
# lock.release() вызовется даже при исключении

4. HTTP сессии:

import requests

with requests.Session() as session:
    response = session.get("https://api.example.com/users")
    data = response.json()
# Соединение закроется правильно

Создание собственных менеджеров контекста

Вариант 1: Класс с enter и exit

class DatabaseConnection:
    def __init__(self, host, user, password):
        self.host = host
        self.user = user
        self.password = password
        self.connection = None
    
    def __enter__(self):
        print(f"Подключаемся к {self.host}...")
        # Имитируем подключение
        self.connection = {"connected": True}
        return self.connection
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(f"Отключаемся от {self.host}...")
        self.connection = None
        
        # Обработка исключений
        if exc_type:
            print(f"Ошибка при работе: {exc_type.__name__}: {exc_val}")
            # return True чтобы подавить исключение
        return False

with DatabaseConnection("localhost", "user", "pass") as conn:
    print(f"Работаем с подключением: {conn}")

Вариант 2: Декоратор contextmanager (проще)

from contextlib import contextmanager

@contextmanager
def db_connection(host, user, password):
    print(f"Подключаемся к {host}...")
    connection = {"connected": True}
    try:
        yield connection  # Этот объект получит переменная после 'as'
    finally:
        print(f"Отключаемся...")
        connection = None

# Использование
with db_connection("localhost", "user", "pass") as conn:
    print(f"Работаем: {conn}")

Множественные менеджеры контекста

# Открываем два файла одновременно
with open("input.txt") as fin, open("output.txt", "w") as fout:
    for line in fin:
        fout.write(line.upper())
# Оба файла закроются

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

class SafeResource:
    def __enter__(self):
        print("Выделяем ресурс")
        return self
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(f"Освобождаем ресурс")
        
        # exc_type = тип исключения (None если ошибки не было)
        # exc_val = значение исключения
        # exc_tb = traceback
        
        if exc_type is ValueError:
            print("Была ошибка ValueError, подавляем её")
            return True  # Подавить исключение
        
        if exc_type is not None:
            print(f"Была ошибка {exc_type.__name__}")
        
        return False  # Не подавлять исключение

with SafeResource() as res:
    raise ValueError("Ошибка!")  # Будет подавлена

Практический пример: работа с временем

import time
from contextlib import contextmanager

@contextmanager
def timer(label):
    """Менеджер контекста для измерения времени выполнения"""
    start = time.time()
    try:
        yield
    finally:
        elapsed = time.time() - start
        print(f"{label}: {elapsed:.2f}с")

with timer("Обработка данных"):
    data = [i**2 for i in range(1000000)]
# Выведет: Обработка данных: 0.05с

Главная проблема, которую решают менеджеры контекста

Менеджеры контекста гарантируют:

• Выполнение кода очистки (cleanup) — __exit__() всегда вызовется
• Безопасность при исключениях — ресурсы освободятся даже при ошибке
• Читаемость кода — явно видно, где начинается и заканчивается использование ресурса
• Предотвращение утечек ресурсов — нет необходимости помнить о close()

Вывод: Менеджеры контекста решают проблему гарантированной очистки ресурсов. Всегда используй with для работы с файлами, БД, сетью, блокировками и другими ресурсами, которые нужно явно освобождать.