python - SqlAlchemy 中的本地 "merge"

Question

我想知道是否有一种方法可以将新映射与数据库数据合并，例如使用 session.merge，但无需更新数据库？就像我用 git 执行 pull 时，获取本地状态，该状态是远程和先前本地状态(可能包含未推送的提交)的合并，但不更新远程状态。在这里，我想获取执行 session.merge 后产生的状态的本地“ View ”。

也许创建保存点(使用 session.begin_nested)，然后执行 session.merge 和稍后的 session.rollback 将完成此操作，但是有没有一种方法不需要这种事务管理(这可能意味着对数据库的实际撤消操作，所以对于我的目的来说效率不是很高)？

使用 session.no_autoflush 会成功吗？

我想做的示例代码:

local_merge = session.merge_local(Model(...))
# do stuff with local_merge...
remotes = session.query(Model).all()
# remotes should remain "old" db versions, as no data was pushed
return something 

编辑:所以我认为我在 rollback 方法效率低下方面可能是错误的。至少，只要不发出 commit，就不应该有昂贵的撤消操作，只会丢弃事务。

Answer 1

由于自动刷新，合并只会更新数据库。您可以使用 session.no_autoflush 上下文管理器暂时关闭它，或者只是在您的 session 中设置 autoflush=False。您还可以将 autoflush=False 传递给您的 sessionmaker。

需要注意的一件事是 session.query(Model).all() 的结果将如何与未刷新的、已更改的本地对象交互。

因为 session 在 Identity Map 中维护唯一对象的记录(相对于主键)，您将无法在同一 session 中拥有同一对象的两个版本。

这是一个示例，显示了本地更改如何与 autoflush=False 交互:

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///:memory:', echo=True)
Base = declarative_base()


class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String)

    def __repr__(self):
        return "<User(name='%s')>" % self.name


Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

ed_user = User(name='ed')
session.add(ed_user)
session.commit()

ed_again = session.query(User).get(1)
ed_again.name = "not ed"

with session.no_autoflush:
    ed_three = session.query(User).get(1)
    all_eds = session.query(User).all()
    print(ed_again, id(ed_again))
    print(ed_three, id(ed_three))
    print(all_eds, id(all_eds[0]))

<User(name='not ed')> 139664151068624
<User(name='not ed')> 139664151068624
[<User(name='not ed')>] 139664151068624

是的，它无法从数据库中获取原始 Ed，即使使用 no_autoflush - 这对于 get() 来说是意料之中的，因为它将检查身份映射首先在数据库之前，如果在身份映射中找到它，则不会查询数据库。但是使用 query.all()，它查询数据库，发现它返回的对象之一已经在标识映射中，并返回该引用来代替，以保持对象的唯一性session(这也是我的直觉，但我找不到在文档中明确说明的内容)。

你可以做类似 expunge 的事情来自 session 的对象，但我认为拥有合并对象的旧副本和新副本的最简单方法是使用两个单独的 session ，一个将合并并可能提交更改，另一个是您可用于检查数据库中对象的现有状态。