在请求上下文外部（生成器内部）设置 Flask 会话变量

Question

我有一个烧瓶视图，执行该视图以在生成器中加载一些信息（我正在使用生成器，以便我可以不断生成进度 - 已加载多少信息）。视图如下所示：

@app.route("/progress", methods=['GET'])
def progress():
     gen = get_user_saved_tracks(session['token'], session['spotify_id'], session)
     return Response(gen, mimetype= 'text/event-stream')

def get_user_saved_tracks(token, id, session):
    #load information and keep yielding
    session['info'] = info

我想存储在生成器内的会话变量中加载的信息（该生成器函数在请求上下文之外的不同文件中定义）。但是当我尝试访问会话变量时，出现以下错误：

运行时错误：在请求上下文之外工作。

那么，有没有办法通过这种方式将信息写入到session中呢？我现在正在使用 FileSystemSessionInterface，但愿意使用 redis 会话（如果这能解决我的问题）。

更新：

根据 Sraw 的建议，我尝试了以下更改：

from flask import current_app
app = current_app._get_current_object()
def get_user_saved_tracks(token, id,session):
    with app.app_context():
        session['info'] = info

但我仍然遇到同样的错误。

更新2：

因此，我需要使用实际的应用程序实例而不是使用 current_app （应用程序对象正在不同的文件中创建 - app.py）

from app import app
def get_user_saved_tracks(token, id,session):
    with app.app_context():
        session['info'] = info

这样做时，我得到同样的错误：

运行时错误：在请求上下文之外工作。

更新3：

以下是 get_user_saved_tracks 的代码：

def get_user_saved_tracks(token, id, session, j, service):
    tracks = []
    for i in range(100):
        a = service.current_user_saved_tracks(limit=50, offset=i*50)
        if len(a['items']) == 0:
            break
        yield "data:" + "lib" + str((float(i+1)/(j))*100) + "\n\n"
        time.sleep(0.5)
        tracks.extend(a)

    session['tracks'] = tracks
    session.modified = True
    yield "data:" + "close" + "\n\n"

Answer 1

在类似的设置中，我遇到了与您相同的错误。但通过使用：

return Response(stream_with_context(....),
                mimetype='text/event-stream')

已解决。所以在你的情况下这意味着：

from flask import Flask, stream_with_context, request, Response, session
....

return Response(stream_with_context(get_user_saved_tracks(session['token'], session['spotify_id'], session),
                mimetype='text/event-stream')

编辑：

一种纯粹的技术方法，通过使用回调函数来完成您想要完成的任务。如果这是您想要的，那么这是另一个问题。您引入了 2 个附加参数j和服务（？）。

def get_user_saved_tracks(cb, token, id, j, service):
    #tracks = []
    for i in range(100):
        a = service.current_user_saved_tracks(limit=50, offset=i*50)
        if len(a['items']) == 0:
            break
        yield "data:" + "lib" + str((float(i+1)/(j))*100) + "\n\n"
        time.sleep(0.5)
        #tracks.extend(a)
        cb(a)  # <<--------

    #self.session['tracks'] = tracks
    #self.session.modified = True
    yield "data:" + "close" + "\n\n"

@app.route("/progress", methods=['GET'])
def progress():
    # add a callback function to update the session for the 'a' you
    # calculate in get_user_saved_tracks(). The callback updates the session
    session['tracks'] = []
    def cb(l):
        session['tracks'].extend(l)
    session.modied = True

    return Response(stream_with_context(get_user_saved_tracks(cb, session['token'], session['spotify_id']),  # j and service ?
            mimetype='text/event-stream')