流式传输 Flask 响应时保持 SQLAlchemy 会话处于活动状态

Question

我正在尝试从使用 Flask-SQLAlchemy 的 Flask 服务器将大型 CSV 流式传输到客户端。

配置应用时（使用工厂模式），每次请求后都会调用db.session.close()：

@app.after_request
def close_connection(r):
    db.session.close()
    return r

到目前为止，此配置运行良好，因为所有请求都是短暂的。但是在流式传输响应时，SQLAlchemy 会话过早关闭，在调用生成器时抛出以下错误：

sqlalchemy.orm.exc.DetachedInstanceError: Parent instance <Question> is not bound to a Session; lazy load operation of attribute 'offered_answers' cannot proceed

伪代码：

@app.route('/export')
def export_data():
    answers = Answer.query.all()
    questions = Question.query.all()
    def generate():
        Iterate through answers questions and write out various relationships to csv

    response = Response(stream_with_context(generate()), mimetype='text/csv')
    return response

我已经尝试了使用/不使用stream_with_context 和def close_connection 中的全局标志的多种配置，以不自动关闭连接，但同样的错误仍然存​​在。

Answer 1

@app.after_request 在调用流文件的生成器之前关闭数据库会话。

解决方案是将db.session.close() 迁移到@app.teardown_request。实例化Response时也必须使用stream_with_context。