python教程—DictCursor在psycopg2下似乎不能工作-Python实用宝典

python教程—DictCursor在psycopg2下似乎不能工作

我以前没有使用过psycopg2,但是我尝试将游标工厂更改为DictCursor,这样fetchall或fetchone将返回一个字典而不是列表。

我以前没有使用过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版。

回答

    res = cur.fetchall()

使res成为psycopg2.ext . dictrows的列表。


或者,您可以利用cur是一个可迭代的事实,而不是调用curr .fetchall:

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

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

​Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号:Python实用宝典

本文由 Python实用宝典 作者:Python实用宝典 发表,其版权均为 Python实用宝典 所有,文章内容系作者个人观点,不代表 Python实用宝典 对观点赞同或支持。如需转载,请注明文章来源。
0

发表评论