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

循环遍历

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

1
2
3
4
5
6
7
8
9
10
11
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返回字典结构

1
2
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'即可。

参考