如何从 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: No module named JSONEncoder

Answer 1

使用Flask内置的jsonify函数，因为它是already extended to work with dates：

from Flask import jsonify

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

数据将作为具有单个键 (data) 的对象返回，其中包含行数组（根据 fetchall 返回行的形式，将表示为数组或对象）。

如果您需要序列化更多类型（如您的情况，您将返回 date 而不是 datetime 实例，您将需要使用 JSONEncoder 的子类覆盖 Flask 的 json_encoder 属性，它知道如何处理你的类型：

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。