python-3.x - 从 Matplotlib 填充 Flask 图像

Question

这就是我想要做的。

我有一个名为“index.html”的模板，其中包含这样的图像:

<img class="img-circle" src="/image?imei={{i}}&s={{s}}">

在 flask 中，我有一个/image url 的处理程序，如下所示:

@app.route('/image', methods=['GET'])
def image():
    imei = request.args.get('imei') 
    dt = request.args.get('s').split("/") 
    return someFunction(imei,dt) <-- SHOULD THIS BE RETURN?

someFunction 生成一个 Matplotlib 图像。图像在保存到磁盘时看起来是正确的，但我不知道如何“返回”这个图像，以便它在 image.html 中正确呈现。这是我目前正在做的事情( Python Matplotlib to smtplib ):

def somefunction(imei,dt)
    ...
    #plt is valid and can be saved to disk
    plt.tight_layout()
    #plt.show() <-- this looks correct when uncommented

    buf = io.BytesIO()
    plt.savefig(buf, format = 'png')
    buf.seek(0)
    print(buf.read()) <-- prints a bunch of stuff
    return buf.read() <-- this does not work

前面提到的“一堆东西”看起来像这样:

b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x03 \x00\x00\x02X\x08\x06\x00\x00\x00

我该怎么做呢？目前 Flask 炸弹有:ValueError: View function did not return a response

Answer 1

尝试:

@app.route("/image")
def image():
    ret = # create raw image data w/matplotlib

    resp = Response(response=ret,
                    status=200,
                    mimetype="image/png")

    return resp

这个我没试过，可能不行。