SQLAlchemy：使用 id 而不是对象添加关系？

Question

是否可以使用 ids 而不是对象来添加 SQLAlchemy 关系？

例如，考虑两个声明性 SQLAlchemy 类，Review 和 Artist，它们之间存在关系：

class Review(Base):
    artist_id = Column(Integer, ForeignKey('artist.id'))
    artist = relationship(Artist, backref=backref('reviews', order_by=id))
    # etc.

class Artist(Base):
    # etc.

通过要添加到艺术家的评论 id 列表，我似乎需要从 id 查找艺术家，然后将艺术家对象添加到评论如下：

for review_id in review_ids:
    review = session.query(Review).filter(Review.id==review_id).first()
    artist.reviews.append(review)

我确信跳过查找并仅添加 ids 会更有效，但这可能吗？

Answer 1

您最好的选择可能是针对支持表编写一个 update 表达式。否则，如果没有实际查询，您就无法真正修改 Review（这就是您所做的；您实际上根本没有修改 Artist）。

假设 Review.id 是主键，大致如下：

conn = session.connection()
conn.execute(Review.__table__
                .update()
                .values(artist_id=artist_id)
                .where(Review.id.in_(review_ids))
            )

Answer 2

我认为如果您想坚持使用纯 ORM 解决方案，您将不得不忍受效率较低的查询模式。

@TokenMacGuy 提供了一个很好的选择。您可以通过将 add_reviews() 添加到 Artist() 类来集成该方法。这将增强“标准”orm api，因此您可以根据情况使用其中之一。

from sqlalchemy.orm.session import object_session

class Artist(Base):
    def add_reviews(self, review_ids):
        sess = object_session(self)
        if isinstance(review_ids, int): review_ids = [review_ids]
        sess.execute(
            Review.__table__
            .update()
            .values(artist_id=self.artist_id)
            .where(Review.id.in_(review_ids))
        )
        sess.refresh(self)

review_ids = [123, 556, 998, 667, 111]
artist.add_reviews(review_ids)
review_id = 333
artist.add_reviews(review_id)

Answer 3

我不确定我是否理解了。我猜你想要这样的东西：

reviews = session.query(Review).filter(Review.id.in_(review_ids)).all()

artist.reviews.extend(reviews)