如何在Flask中使用Python的qrcode服务生成的二维码图像

Question

我有一个函数可以生成二维码图片：

import qrcode
def generateRandomQR():
    qr = qrcode.QRCode(version=1,
            error_correction=qrcode.constants.ERROR_CORRECT_L,
            box_size=10,
            border=4,
            )

    qr.add_data("Huehue")
    qr.make(fit=True)
    img = qr.make_image()
    return img

接下来，我的想法是生成这个图片，然后把它放到flask上，作为一张图片来提供服务，这是我在flask中的函数：

@app.route("/qrgenerator/image.jpg")
def generateQRImage():
    response = make_response(qrWrapper.generateRandomQR())
    response.headers["Content-Type"] = "image/jpeg"
    response.headers["Content-Disposition"] = "attachment; filename=image.jpg"
    return response

但是它似乎没有正常工作... 我遇到了500错误，所以我不太确定我哪里出错了。

Answer 1

编辑：我看到你提到不想在回答后保存到临时文件。不过，如果你决定保存到临时位置，这里有一种方法。

你可以把二维码图片保存在一个临时位置，然后使用 send_file 来提供这个文件。

send_file 的使用方法可以在这个链接找到：http://flask.pocoo.org/docs/0.10/api/#flask.send_file

我没有测试过这段代码，但类似的代码应该可以正常工作。

from flask import send_file

@app.route("/qrgenerator/image.jpg")
def generateQRImage():
    response = make_response(qrWrapper.generateRandomQR())

    temp_location = '/tmp/image.jpg'

    # Save the qr image in a temp location
    image_file = open(temp_location, 'wb')
    image_file.write(response)
    image_file.close

    # Construct response now
    response.headers["Content-Type"] = "image/jpeg"
    response.headers["Content-Disposition"] = "attachment; filename=image.jpg"
    return send_file(temp_location)

Answer 2

在Google+上，有位绅士给我提供了一个解决方案，虽然他没有StackOverflow的账号，但我决定分享他的回答：

#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-
"""Example of Flask and qrcode.

NOTE: by requirements image in memory!
"""

__author__ = 'Daniel Leybovich <setarckos@gmail.com>'
__version__ = (0, 0, 1)


import os
import sys
import flask
import qrcode
import cStringIO


app = flask.Flask(__name__)


def random_qr(url='www.google.com'):
    qr = qrcode.QRCode(version=1,
                       error_correction=qrcode.constants.ERROR_CORRECT_L,
                       box_size=10,
                       border=4)

    qr.add_data(url)
    qr.make(fit=True)
    img = qr.make_image()
    return img


@app.route('/get_qrimg')
def get_qrimg():
    img_buf = cStringIO.StringIO()
    img = random_qr(url='www.python.org')
    img.save(img_buf)
    img_buf.seek(0)
    return flask.send_file(img_buf, mimetype='image/png')


if __name__ == '__main__':
    app.run(host='0.0.0.0', debug=True)