DictCursor 似乎在 psycopg2 下不起作用

Question

我以前没有使用过 psycopg2，但我正在尝试将游标工厂更改为 DictCursor，以便 fetchall 或 fetchone 将返回字典而不是列表。

我创建了一个测试脚本来使事情变得简单并且仅测试此功能。这是我认为应该有效的一点代码

import psycopg2
import psycopg2.extras

conn = psycopg2.connect("dbname=%s user=%s password=%s" % (DATABASE, USERNAME, PASSWORD))

cur = conn.cursor(cursor_factory = psycopg2.extras.DictCursor)
cur.execute("SELECT * from review")

res = cur.fetchall()

print type(res)
print res

res 变量始终是一个列表，而不是我所期望的字典。

我当前实现的解决方法是使用此函数构建字典并通过它运行 fetchall 返回的每一行。

def build_dict(cursor, row):
    x = {}
    for key,col in enumerate(cursor.description):
        x[col[0]] = row[key]
    return d

Python 的版本为 2.6.7，psycopg2 的版本为 2.4.2。

Answer 1

res = cur.fetchall()

使 res 成为 psycopg2.extras.DictRow 的列表。

---

或者，您可以利用 cur 是可迭代的这一事实，而不是调用 cur.fetchall：

cur.execute("SELECT * from review")
for row in cur:
    print(row['column_name'])

因此您将能够使用 访问数据类似于 dict 的语法。

Answer 2

另一种解决方案是使用 命名元组游标，因为 Real Dict 游标将破坏任何使用整数索引的查询，如其文档中所述。

使用命名元组游标，您可以使用点语法访问它们，如下所示：

import psycopg2
import psycopg2.extras
cur = conn.cursor(cursor_factory = psycopg2.extras.NamedTupleCursor)
cur.execute("SELECT * from review")
res = cur.fetchone()
res.key1
res.key2

这使事情保持整洁，并且据我所知不会破坏任何内容。

Answer 3

虽然这是一个较老的问题，但它仍然出现在谷歌中，所以我想我会为来自大G的其他人添加我的代码。

对我来说，我有多行，我想返回到字典中并且理想情况下，不想使用循环或类似的方式从数据库中的字段设置键。

因此，使用 dict 理解语法我​​可以执行以下操作。

表行插入字典

---

pgCursor = Conn.cursor(cursor_factory = psycopg2.extras.RealDictCursor)
pgCursor.execute("SELECT * FROM tablename;",([]))
dictRows = {n['id']: n for n in pgCursor}

---

函数和字典调用它

#NOTE this is using a class object hence the self param
def DBTableToDictByID(self, squery):
    self.Pointer.execute(squery,([]))
    return {n['id']: n for n in self.Pointer}

dictRows = self.DBTableToDictByID("SELECT * FROM tablename;")

---

虽然这是使用 for x in y 循环，但据我所知，它是Pythonic....希望这对一些人有帮助。

Answer 4

除了使用 RealDictCursor 功能外，您可能还需要询问所有列
（在选择后使用 * 符号），如答案中所做的那样。

我对结果的某些列不感兴趣，因为它们具有已知的值
已在 WHERE 条件中使用。但是 SELECT (..., ..., ..., ...) FROM ... WHERE ...
变体没有给我字典。

此致 ！
哈雷

Answer 5

因此，要使其像 mysql 版本的 Dictionary 游标一样工作，您必须将其包装在另一个函数或代码中。我将去论坛并向他们建议，以便将来部署他们的代码，以便在 fetchall() 调用与字典游标一起使用时返回字典。下面是一些示例代码，您可以使用它来修复它：

cursor.execute(query)
# Python 2.7 and beyond with dictionary comprehension
results = [{key:value for key,value in row.iteritems()} for row in cursor]
# Python 2.6 and before
# results = [dict((key,value) for key,value in row.iteritems()) for row in cursor]

该代码使用 fetchall() 使其格式与 MySQL 版本的字典游标相同。不知道为什么他们以不同的方式实现它，但这将帮助您获得实际 python 字典的相同输出，而不是 fetchall() 情况下的列表。

Answer 6

使用 RealDictCursor：

import psycopg2.extras

cur = conn.cursor(cursor_factory = psycopg2.extras.RealDictCursor)
cur.execute("SELECT * from review")
res = cur.fetchall()    

这给你一个列表，其中的行是真正的 python 字典，而不是“高级 psycopg2 列表”。