如何从Flask应用程序中的MySQL查询返回数据？

Question

我有以下代码:

import flask as fk
import MySQLdb
import JSONEncoder


class SpecializedJSONEncoder(JSONEncoder):
    def default(o):
        if isinstance(o, date):
            return date.strftime("%Y-%m-%d")
        else:
            super(SpecializedJSONEncoder, self).default(o)

app = fk.Flask(__name__)
app.json_encoder = SpecializedJSONEncoder
app.debug = True

@app.route("/")
def home():
   return "Hello world"

@app.route("/temp")
def temp():
    db = MySQLdb.connect("localhost", "root", "","test")
    cur = db.cursor()
    query = "SELECT DATE(DTM), POM, ROUND(MIN(TMP),1) FROM dados_meteo WHERE POM = %s AND      DATE(DTM) >= %s AND DATE(DTM) <= %s"
    param = ("Faro", "2013-12-01", "2013-12-05")
    cur.execute(query, param)
    data = cur.fetchall()

    return data.json_encoder()

 if __name__ == "__main__":
    app.run()

返回的错误是:ImportError:没有名为JSONEncoder的模块

Answer 1

使用Flask的内置jsonify函数,因为它已经扩展为使用日期:

from Flask import jsonify

@app.route('/temp')
def temp():
    # Load database results
    # and then ...
    return jsonify(data=cur.fetchall())

数据将作为一个对象返回,data其中包含一个包含行数组的key()(根据fetchall返回的行,它将表示为数组或对象).

如果你需要序列化更多类型(在你的情况下,你回来date而不是datetime实例,你需要覆盖Flask的json_encoder属性,其子类JSONEncoder知道如何处理你的类型:

class SpecializedJSONEncoder(JSONEncoder):
    def default(o):
        if isinstance(o, date):
            return date.strftime("%Y-%m-%d")
        else:
            super(SpecializedJSONEncoder, self).default(o)

然后你可以在你的Flask实例上设置它:

app.json_encoder = SpecializedJSONEncoder

您现在可以处理dates以及datetimes.