如果引发异常，如何使 Flask-SQLAlchemy 自动回滚会话？

Question

我想设置一个使用Flask-SQLAlchemy 构建的应用，以在视图引发在视图代码之外冒泡的异常（即未捕获到内部）时回滚对数据库所做的所有更改。

即使某些对象在子事务中自动或直接通过session.commit() 刷新到数据库，我也希望它能够工作。

类似于Django's transaction request wrapping。

Answer 1

受dduffy's answer 启发，对我有用的最可靠的解决方案是附加一个错误处理程序，专门捕获 SQLAlchemy 异常并发出回滚。

from sqlalchemy import exc
@app.errorhandler(exc.SQLAlchemyError)
def handle_db_exceptions(error):
    #log the error: app.logger.error(error)
    db.session.rollback()

@app.teardown_request 钩子方法似乎没有及时修复处于不良状态的 db 连接，然后它被后来的请求从 db 连接池中拉出。 errorhandler 方法在重用之前捕获连接，并将其范围限制为 SQLAlchemy 引发的错误类别，感觉更精确。

Answer 2

您可以在自定义错误处理程序中包含回滚...

@app.errorhandler(500)
def internal_error(error):
    db.session.rollback()
    return render_template('500.html'), 500

请参阅“自定义错误页面”部分下的 here。

Answer 3

你可以这样做：

@app.teardown_request
def teardown_request(exception):
    if exception:
        db.session.rollback()
    db.session.remove()

查看here 以获取 teardown_request 信息。如果您处于调试模式，您可能需要设置 PRESERVE_CONTEXT_ON_EXCEPTION 配置变量。