python - 如何从 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())

数据将作为具有单个键(数据)的对象返回，该对象包含一个行数组(根据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

您现在可以处理 date 以及 datetime。