Python flask中mysql查询返回json格式数据

在进行web项目开发时，经常需要查询数据库并返回json格式给前端，Java中可以使用JSONObject或mybatis直接就可以映射成map结构，而python中返回的是元组的形式(('a',1,2), ('b',2,3), ...)。下面提供几种在python flask中转换为json格式的方式。

循环遍历

最简单粗暴的方式就是遍历结果集，使用字典进行包装，最后使用flask的jsonify将其转换为json格式，参考代码如下：

def index():
   cur = mysql.connection.cursor()
   cur.execute('''SELECT * FROM Users WHERE id=1''')
   rv = cur.fetchall()
   payload = []
   content = {}
   for result in rv:
       content = {'id': result[0], 'username': result[1], 'password': result[2]}
       payload.append(content)
       content = {}
   return jsonify(payload)

直接设置Mysqldb返回字典结构

db  = MySQLdb.connect(host='localhost', user='root', passwd='123456', db='test')
cur = db.cursor(cursorclass=MySQLdb.cursors.DictCursor)

只需要在最后设置cursorclass可选参数即可

在flask中设置

我使用的是flask-mysqldb扩展，只需要在配置中设置MYSQL_CURSORCLASS='DictCursor'即可。

参考

• https://stackoverflow.com/questions/43796423/python-converting-mysql-query-result-to-json
• http://chenxiaoyu.org/2009/11/10/python-mysqldb-return-dict/
• https://stackoverflow.com/questions/38980044/looking-for-the-equivalent-of-dictcursor-in-flaskext-mysql
• https://flask-mysqldb.readthedocs.io/en/latest/