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

使用RealDictCursor:

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

这将为您提供一个列表,其中行为真正的python词典,而不是"高级psycopg2列表".

Answer 2

res = cur.fetchall()

使得res列表psycopg2.extras.DictRow秒.

或者,不是调用cur.fetchall你可以利用cur可迭代的事实:

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

因此,您将能够使用dict类似语法访问数据.

Answer 3

另一个解决方案是使用命名元组游标,因为Real Dict Cursor将破坏任何使用整数指示的查询,如文档中所述.

使用Named Tuple Cursors,您可以使用点语法访问它们,如下所示:

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 4

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

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

因此，使用dict comprehension syntax我可以执行以下操作。

表行到字典中

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;")

虽然这是在 y 循环中使用 for x ，但据我所知，它是 pythonic ......希望这对那里的一些人有所帮助。