如何在psycopg2中同时执行多个PostgreSQL SELECT查询并获取全部结果？

方案1：利用psycopg2的结果集切换（先确认版本） #

首先检查你的psycopg2版本，执行print(psycopg2.__version__)。较新版本的psycopg2 cursor原生支持nextset()方法，可能你当前使用的版本过低。如果允许升级，执行pip install --upgrade psycopg2-binary（注意是psycopg2-binary，不是psycopg）。

升级后可用以下代码获取所有结果集：

import psycopg2

conn = psycopg2.connect("你的数据库连接参数")
cur = conn.cursor()

# 执行多查询语句
cur.execute("SELECT * FROM table1; SELECT * FROM table2;")

all_results = []
# 获取第一个结果集
all_results.append(cur.fetchall())
# 切换到下一个结果集，直到没有更多
while cur.nextset():
    all_results.append(cur.fetchall())

# 处理结果
for idx, result in enumerate(all_results):
    print(f"第{idx+1}个查询结果: {result}")

cur.close()
conn.close()

方案2：无nextset()时的兼容处理 #

如果确实无法升级psycopg2，可通过检查cursor.description判断是否还有下一个结果集：

import psycopg2

conn = psycopg2.connect("你的数据库连接参数")
cur = conn.cursor()

cur.execute("SELECT * FROM table1; SELECT * FROM table2;")

all_results = []
while cur.description is not None:
    # 获取当前结果集
    all_results.append(cur.fetchall())
    # 触发获取下一个结果集的元数据，若没有下一个结果集，cur.description会变为None
    cur.fetchone()

# 移除最后一次fetchone返回的无效结果（如果存在）
if all_results and all_results[-1] == [None]:
    all_results.pop()

# 处理结果
for idx, result in enumerate(all_results):
    print(f"第{idx+1}个查询结果: {result}")

cur.close()
conn.close()

方案3：用临时表中转结果 #

如果上述方法都不适用，可以将每个查询的结果存入临时表，再依次取出：

import psycopg2

conn = psycopg2.connect("你的数据库连接参数")
cur = conn.cursor()

# 执行创建临时表+查询的语句
cur.execute("""
CREATE TEMP TABLE temp_table1 AS SELECT * FROM table1;
CREATE TEMP TABLE temp_table2 AS SELECT * FROM table2;
SELECT * FROM temp_table1;
SELECT * FROM temp_table2;
""")

# 按方案1的方式获取结果
all_results = []
all_results.append(cur.fetchall())
while cur.nextset():
    all_results.append(cur.fetchall())

# 清理临时表（可选，会话结束后临时表会自动删除）
cur.execute("DROP TABLE temp_table1; DROP TABLE temp_table2;")
conn.commit()

cur.close()
conn.close()

内容的提问来源于stack exchange，提问作者Quantitative