将查询结果转换为 JSON

Question

我有一个返回查询结果的python函数：

class Arc_Nodes_Data(Resource):
  def get(self):
        #Connect to databse
        conn = connectDB()
        cur = conn.cursor()
        #Perform query and return JSON data
        try:
          cur.execute("SELECT start_location FROM trips UNION SELECT end_location FROM trips")
        except:
          print("Error executing select")
        NodeList = list (i[0] for i in cur.fetchall())
        return jsonify ({'id':NodeList})

此函数以下列格式返回结果：

{
  "id": [
    "a",
    "b",
    "c"
  ]
}

我在上面尝试了这个函数，所以输出可能是这样的：

[{ 
  "id": "a"
  }, {
  "id": "b"
  }, {
  "id": "c"
}]

显然我做错了什么。我似乎不能很好地操作 Python 字典，所以也许这就是问题所在。我怎样才能得到正确的输出？

Answer 1

你可以使用列表推导来做你想做的事：

return jsonify ([{'id': x} for x in Nodelist])

这种理解产生了 3 个字典，每个字典都包含一个 id 元素。

您还可以将最后一行合并到一行中：

return jsonify ([{'id': i[0]} for i in cur.fetchall()])