python - row_to_json 和 psycopg2.fetchall() 结果是列表中的列表而不是列表中的字典

Question

我使用 Postgres 的 row_to_json() 函数将数据检索为 json 对象，以便像使用 python 字典一样处理结果。

conn = psycopg2.connect("<My_DB_DSN>")
cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
query_sql = "SELECT row_to_json(row) FROM (SELECT id, name FROM products) row;" 
cur.execute(query_sql)
results = cur.fetchall()
print(results)

这导致:

 [ [{"id": 1, "name": "Bob"}],
   [{"id": 2, "name": "Susan"}]
 ]

我期待这样的结果:

 [ {"id": 1, "name": "Bob"},
   {"id": 2, "name": "Susan"}
 ]

知道为什么我会得到第一个结果吗？我该如何解决这个问题？

在 postgres 的命令行中运行 SQL 查询将按预期返回 json:

{"id": 1, "name": "Bob"},
{"id": 2, "name": "Susan"}

Answer 1

我认为您需要 RealDictCursor，它会将每一行作为字典返回，您不需要修改 SQL 查询:

from psycopg2.extras import RealDictCursor

cur = conn.cursor(cursor_factory=RealDictCursor)
query_sql = "SELECT id, name FROM products where id < 10" 
cur.execute(query_sql)
results = cur.fetchall()
print(results)

返回:

[{'id': 2L, 'name': 'Foo'}, {'id': 4L, 'name': 'Bar'}, ...]