Сложные связи в базах данных с использованием SQLAlchemy

2. Реализация Many-to-Many

Связь "Многие-ко-Многим" сложнее. Пример: Пост может иметь много Тегов, а один Тег может висеть на множестве постов. Напрямую связать две таблицы нельзя. Нам нужен посредник — Ассоциативная таблица (Association Table).

Шаг 1. Создание таблицы-посредника

Мы создаем таблицу, которая хранит только пары ID: post_id и tag_id. Обычно для неё не нужен отдельный класс-модель, достаточно объекта Table.

# models/association.py
from sqlalchemy import Table, Column, Integer, ForeignKey
from .database import Base

post_tag_association = Table(
    "post_tag_association",
    Base.metadata,
    Column("post_id", Integer, ForeignKey("posts.id"), primary_key=True),
    Column("tag_id", Integer, ForeignKey("tags.id"), primary_key=True)
)

Обратите внимание: мы делаем оба поля primary_key=True. Это создает составной первичный ключ, гарантируя, что один и тот же тег не привяжется к одному посту дважды.

Шаг 2. Настройка моделей

Теперь нужно "научить" модели Post и Tag использовать эту таблицу. Для этого в relationship добавляется параметр secondary.

class Post(Base):
    __tablename__ = "posts"
    id = Column(Integer, primary_key=True)
    title = Column(String)

    # Указываем таблицу-посредника через secondary
    tags = relationship(
        "Tag",
        secondary=post_tag_association,
        back_populates="posts"
    )

class Tag(Base):
    __tablename__ = "tags"
    id = Column(Integer, primary_key=True)
    name = Column(String, unique=True)

    # Зеркальная связь
    posts = relationship(
        "Post",
        secondary=post_tag_association,
        back_populates="tags"
    )

Шаг 3. Нюансы миграций (Alembic)

Решение: Соберите все модели в models/__init__.py:

# models/__init__.py
from .user import User
from .post import Post
from .tag import Tag
# Обязательно импортируем ассоциативную таблицу!
from .association import post_tag_association

4. Практика и Best Practices

Работа с данными

SQLAlchemy позволяет работать со связями как с обычными списками Python. Добавление тега к посту выглядит очень естественно.

def add_tag_to_post(session, post_id, tag_name):
    post = session.get(Post, post_id)

    # Пытаемся найти тег или создаем новый
    tag = session.scalar(select(Tag).where(Tag.name == tag_name))
    if not tag:
        tag = Tag(name=tag_name)

    # МАГИЯ ORM: просто добавляем в список
    post.tags.append(tag)

    # SQLAlchemy сама сгенерирует INSERT в таблицу post_tag_association
    session.commit()

Транзакции и атомарность

Одна из самых больших ошибок новичков — делать commit() внутри вспомогательных функций. Коммит должен быть один, в самом конце бизнес-операции.

def create_post(data):
    post = Post(...)
    session.add(post)
    session.commit()  # ПЛОХО: пост создался, а теги еще нет

def add_tags(post, tags):
    ...
    session.commit()

Правильный подход: функция выполняет всю логику (создание поста + привязка тегов), и только если всё прошло успешно, фиксирует транзакцию.

def create_full_post(session, title, tag_names):
    try:
        # 1. Создаем пост
        post = Post(title=title)
        session.add(post)

        # 2. Привязываем теги (в памяти)
        for name in tag_names:
            tag = Tag(name=name)
            post.tags.append(tag)

        # 3. Фиксируем ВСЁ сразу
        session.commit()
    except Exception:
        session.rollback()
        raise