Запрос многие ко многим в sqlalchemy

Oct 16 2020

Есть таблицы для моего вопроса.

class TemplateExtra(ExtraBase, InsertMixin, TimestampMixin):
    __tablename__ = 'template_extra'

    id = Column(Integer, primary_key=True, autoincrement=False)
    name = Column(Text, nullable=False)
    roles = relationship(
        'RecipientRoleExtra',
        secondary='template_to_role',
    )


class RecipientRoleExtra(
    ExtraBase, InsertMixin, TimestampMixin,
    SelectMixin, UpdateMixin,
):
    __tablename__ = 'recipient_role'

    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(Text, nullable=False)
    description = Column(Text, nullable=False)


class TemplateToRecipientRoleExtra(ExtraBase, InsertMixin, TimestampMixin):
    __tablename__ = 'template_to_role'

    id = Column(Integer, primary_key=True, autoincrement=True)
    template_id = Column(Integer, ForeignKey('template_extra.id'))
    role_id = Column(Integer, ForeignKey('recipient_role.id'))

Я хочу выбрать все шаблоны с предварительно выбранными ролями в двух sql-запросах, как Django ORM с prefetch_related. Могу ли я это сделать? Это моя текущая попытка.

def test_custom():
    # creating engine with echo=True
    s = DBSession()

    for t in s.query(TemplateExtra).join(RecipientRoleExtra, TemplateExtra.roles).all():
        print(f'id = {t.id}')
        for r in t.roles:
            print(f'-- {r.name}')

Но..

  1. он генерирует запрос выбора для каждого шаблона, чтобы выбрать его роли. Могу ли я заставить sqlalchemy выполнять только один запрос?
  2. сгенерированные запросы для ролей без соединения, только FROM recipient_role, template_to_roleс WHERE %(param_1)s = template_to_role.template_id AND recipient_role.id = template_to_role.role_id. Это правильно?

Вы можете мне помочь?

Ответы

1 ky_aaaa Oct 16 2020 at 21:46

Основываясь на этом ответе: flask many to many join, как это сделано prefetch_related из django

Может быть, что-то вроде этого:

roles = TemplateExtra.query.options(db.joinedload(TemplateExtra.roles)).all

Сообщите мне, сработало ли это.