【python】postgresql查询数据返回成字典格式

悠悠 2024-03-23 16:42 83阅读 0赞

postgresql查询数据返回成字典格式

### 确定版本 ###

`psycopg2.extras` 模块最早出现在 psycopg2 的版本 2.5.0 中，因此如果您的 psycopg2 版本低于 2.5.0，就无法使用 `psycopg2.extras` 模块。

为了确定您当前的 psycopg2 版本，请在 Python shell 中运行以下代码：

import psycopg2
    
    print(psycopg2.__version__)

输出的版本号将告诉您当前安装的 psycopg2 版本。如果您的版本低于 2.5.0，请升级到最新版本。如果您的版本高于等于 2.5.0，但仍然遇到 `AttributeError: module 'psycopg2' has no attribute 'extras'` 错误，请确保您的 psycopg2 安装正确，并尝试重新安装它。

### 安装方式 ###

解决此问题的方法是升级您的 psycopg2 版本。您可以使用以下命令在命令行中升级 psycopg2：

pip install --upgrade psycopg2

如果您使用的是 Anaconda，可以使用以下命令升级 psycopg2：

conda install -c anaconda psycopg2

升级完 psycopg2 后，您应该能够正常使用 `psycopg2.extras` 模块了。

如果您的 psycopg2 版本是 2.9.0，但仍然遇到 `AttributeError: module 'psycopg2' has no attribute 'extras'` 错误，这可能是因为您在导入 psycopg2 模块时使用了错误的语法。

在 psycopg2 2.9.0 中，`psycopg2.extras` 模块的导入方式已经改变。如果您使用旧的导入语法（例如 `from psycopg2.extras import DictCursor`），就会遇到上述错误。

在 psycopg2 2.9.0 中，您应该使用以下导入语法：

from psycopg2 import extras
    
    # 然后可以使用 extras 模块中的各种功能，例如
    cur = conn.cursor(cursor_factory=extras.DictCursor)

注意，这里使用 `from psycopg2 import extras`，而不是 `from psycopg2.extras import ...`。这是因为在 psycopg2 2.9.0 中，`extras` 不再是一个子模块，而是一个独立的模块。

如果您仍然遇到 `AttributeError: module 'psycopg2' has no attribute 'extras'` 错误，请确认您的 psycopg2 版本正确，并检查您的导入语法是否正确。

### 开始使用 ###

在使用 `extras.DictCursor` 时，查询结果返回的确实是元组（tuple）类型，其中每个元组的元素都是字段名和对应的值。

如果您需要将查询结果转换为字典类型，可以手动将查询结果转换为字典。例如，对于每个查询结果元组，可以使用以下代码将其转换为字典：

import psycopg2
    from psycopg2 import extras
    
    conn = psycopg2.connect(database="mydatabase", user="myusername", password="mypassword", host="localhost", port="5432")
    
    cur = conn.cursor(cursor_factory=extras.DictCursor)
    cur.execute("SELECT * FROM mytable")
    
    result = []
    for row in cur.fetchall():
        row_dict = dict(row)
        result.append(row_dict)
    
    print(result)

这样，查询结果就会以列表的形式存储在 `result` 变量中，每个列表元素都是一个字典，其中键是字段名，值是对应的值。

如果您希望在查询时立即将结果转换为字典类型，而不是在查询后手动转换，请使用 `extras.RealDictCursor`。例如：

import psycopg2
    from psycopg2 import extras
    
    conn = psycopg2.connect(database="mydatabase", user="myusername", password="mypassword", host="localhost", port="5432")
    
    cur = conn.cursor(cursor_factory=extras.RealDictCursor)
    cur.execute("SELECT * FROM mytable")
    
    result = cur.fetchall()
    
    print(result)

这样，查询结果将以列表的形式存储在 `result` 变量中，每个列表元素都是一个字典，其中键是字段名，值是对应的值。

第二种方法：RealDictCursor使用还存在一定问题，没有得到期望的字典形数据，还是元祖形数据，那就先用第一种方式转化一下，也能满足需求了，如果你找到第二种方法问题欢迎在留言区展现你的才华哟。