Как реализуется Unit of Work через context manager в Python?

Что такое Unit of Work?

Unit of Work — это паттерн, который собирает все изменения, сделанные с объектами в рамках одной бизнес-операции, и затем единовременно применяет их к базе данных. Это гарантирует атомарность: либо все изменения сохраняются, либо ни одно.

Реализация через context manager в Python

Контекстный менеджер (оператор with) идеально подходит для реализации Unit of Work, так как он автоматически управляет жизненным циклом транзакции. При входе в блок with открывается сессия/транзакция, а при выходе — либо фиксация (commit), либо откат (rollback) в случае исключения.

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

from contextlib import contextmanager
from sqlalchemy.orm import Session

@contextmanager
def unit_of_work(session: Session):
    try:
        yield session
        session.commit()
    except Exception:
        session.rollback()
        raise
    finally:
        session.close()

# Использование
with unit_of_work(db_session) as session:
    user = session.query(User).first()
    user.name = 'New Name'
    # При выходе из with изменения автоматически сохранятся

Где применяется

• В веб-фреймворках (Django, Flask, FastAPI) для обработки запросов.
• В сервисном слое приложений для группировки операций.
• В тестах для изоляции изменений.

Вывод: Используйте Unit of Work через context manager, когда нужно гарантировать атомарность операций с базой данных и упростить управление транзакциями, особенно в приложениях с несколькими изменениями за один запрос.