python - 如何获得由 zope.transaction 管理的 SQLAlchemy session ，它具有与 http 请求相同的范围但不会在提交时自动关闭？

Question

我有一个 Pyramid 网络应用程序，其中包含一些表单页面，可以从数据库中读取数据并写入其中。

该应用程序将 SQLAlchemy 与 PostgreSQL 数据库结合使用，以下是我如何设置 SQLAlchemy session :

from sqlalchemy.orm import scoped_session
from sqlalchemy.orm import sessionmaker
from zope.sqlalchemy import ZopeTransactionExtension

DBSession = scoped_session(sessionmaker(extension=ZopeTransactionExtension()))

当我处理一个表单时，我需要执行一个包含在 try 中的显式提交，并查看该提交是否有效。我需要这个显式提交，因为我有 deferrable triggers在 PostgreSQL 数据库中(在提交时执行的检查)，在某些情况下，没有错误是不可预测的。

一旦我成功提交了一个事务，例如添加了一个 MyClass 的实例，我想获得这个实例的一些属性以及链接实例的一些属性。事实上，我无法在提交之前获取这些数据，因为它们是由数据库本身计算的。

我的问题是，当我使用 transaction.commit()(在 transaction 包中)时， session 自动关闭，我不能再使用该实例，因为它是处于分离 状态。 Documentation证实了这一点。

因此，如文档中所述，我尝试改用以下 session 设置:

DBSession = scoped_session(sessionmaker(extension=ZopeTransactionExtension(keep_session=True)))

但是，现在 session 范围不再与 http 请求范围相同:在我仅执行读取查询的 http 请求结束时不会发送 ROLLBACK。

那么，有没有办法让 session 与 http 请求具有相同的范围，但不会在提交时自动关闭？

Answer 1

您可以通过中兴通讯上的keep_session=True 将 session 从请求中分离出来。但是，如果是这种情况，您可能还想在 session 提交后使用这些对象，否则您会对新 session 感到满意。因此，您还需要在 session 中使用 expire_on_commit=False。在那之后，您已经成功地将 session 从 pyramid_tm 的生命周期中分离出来，您可以随意提交/中止。那么我们如何在不使用 pyramid_tm 的情况下将其重新附加回请求生命周期？好吧，如果您将 DBSession 包装在不太全局的东西中，这将使它作为请求范围的变量更易于管理，那会有所帮助。从那里，很明显 session 何时创建，何时应该通过请求完成的回调销毁。以下是我的散文摘要:

def get_db(request):
    session = request.registry['db_session_factory']()
    def _closer(request):
        session.close()
    request.add_finished_callback(_closer)
    return session

def main(global_conf, **settings):
    config = Configurator()

    DBSession = sessionmaker(expire_on_commit=False, extension=ZopeTransactionExtension(keep_session=True))
    # look we don't even need a scoped_session anymore because it's not global, it's local to the request!

    config.registry['db_session_factory'] = DBSession

    config.add_request_method('db', get_db, reify=True)

def myview(request):
    db = request.db # creates, or reuses the session created for this request
    model = db.query(MyModel).first();
    transaction.commit()
    # model is still valid here
    return {}

当然，如果我们正在做这一切，ZTE 可能根本帮不了您，您只想使用 db.commit() 并自己处理事情。如果发生异常，完成的回调仍将被调用，因此您不需要 pyramid_tm 来清理。