В чем разница между контекстным менеджером и обычной функцией в Python?

Разница между контекстным менеджером и обычной функцией в Python

Контекстные менеджеры (context managers) — это мощный инструмент Python, который обеспечивает безопасное управление ресурсами. Они отличаются от обычных функций тем, что гарантируют выполнение кода до и после основного блока, даже при возникновении исключений.

Обычная функция

Обычная функция выполняет свою логику и возвращает результат. Управление ресурсами лежит на разработчике.

# Без контекстного менеджера - небезопасно
def read_file(filename):
    f = open(filename, "r")
    content = f.read()
    f.close()  # может не выполниться при исключении!
    return content

# Проблема: если возникнет исключение, файл останется открыт
try:
    data = read_file("data.txt")
    result = int(data)  # может быть ValueError
except ValueError:
    print("Invalid data")  # файл никогда не закроется!

Проблемы:

• Риск утечки ресурсов (файлы, подключения БД, блокировки)
• Требует явного управления (открыть → использовать → закрыть)
• Легко забыть закрыть ресурс
• Усложняется при наличии исключений

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

Контекстный менеджер гарантирует выполнение кода инициализации и очистки, используя протокол enter и exit. Используется с оператором with.

# С контекстным менеджером - безопасно
with open("data.txt", "r") as f:  # __enter__ вызвано
    content = f.read()
    result = int(content)  # может быть ValueError
# __exit__ вызвано в любом случае!

# Файл ВСЕГДА закроется, даже при исключении

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

class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None
    
    def __enter__(self):
        print(f"Opening {self.filename}")
        self.file = open(self.filename, self.mode)
        return self.file  # присваивается переменной после as
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(f"Closing {self.filename}")
        if self.file:
            self.file.close()
        return False  # не подавляем исключение

# Использование
with FileManager("data.txt", "r") as f:
    print("Inside context")
    content = f.read()

print("After context")  # файл уже закрыт

Управление исключениями

class DatabaseConnection:
    def __init__(self, connection_string):
        self.connection_string = connection_string
        self.conn = None
    
    def __enter__(self):
        print(f"Connecting to {self.connection_string}")
        self.conn = self._connect()
        return self.conn
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type is not None:
            print(f"Exception occurred: {exc_type.__name__}")
            print("Rolling back transaction...")
            self.conn.rollback()
        else:
            print("Committing transaction...")
            self.conn.commit()
        
        self.conn.close()
        return False
    
    def _connect(self):
        return type("Connection", (), {
            "rollback": lambda: None,
            "commit": lambda: None,
            "close": lambda: None,
        })()

@contextmanager декоратор

from contextlib import contextmanager
import time

@contextmanager
def timer(name):
    print(f"Starting {name}...")
    start = time.time()
    try:
        yield  # код до yield в __enter__
    finally:
        elapsed = time.time() - start
        print(f"Finished {name} in {elapsed:.2f}s")

# Использование
with timer("data processing"):
    time.sleep(0.5)
    print("Processing data...")

Контекстный менеджер для транзакций

@contextmanager
def database_transaction(connection):
    try:
        connection.begin()
        yield connection
    except Exception as e:
        connection.rollback()
        raise
    else:
        connection.commit()
    finally:
        connection.close()

Вложенные менеджеры

# Компактная форма
with open("input.txt", "r") as infile, open("output.txt", "w") as outfile:
    for line in infile:
        outfile.write(line.upper())

Сравнение

Практические применения

# 1. Блокировки потоков
import threading
lock = threading.Lock()
with lock:  # автоматически захватывает и отпускает
    print("Critical section")

# 2. Подавление исключений
from contextlib import suppress
with suppress(ValueError):
    result = int("invalid")  # исключение подавляется

# 3. Временные директории
import tempfile
with tempfile.TemporaryDirectory() as tmpdir:
    # tmpdir существует и используется
    pass  # автоматически удаляется

# 4. Мокирование в тестах
from unittest.mock import patch
with patch("os.remove") as mock_remove:
    pass  # os.remove заменён, затем восстановлен

Итог

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