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

时间:2014-02-11 18:23:36

标签: python mysql flask

我有以下代码:

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的模块

1 个答案:

答案 0 :(得分:15)

使用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个实例,则需要覆盖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 s。