如何在Flask中通过SQLAlchemy/psycopg2从Postgres游标获取存储过程数据

这种情况我之前做Flask+Postgres项目时也碰到过，核心问题是Postgres存储过程的返回结果处理逻辑和普通查询不一样，不管是SQLAlchemy还是psycopg2，都得针对性调整代码，下面给你具体的解决方案：

一、先确认存储过程的定义是否正确 #

首先得确保你的存储过程是能正确返回数据集的，Postgres里的PROCEDURE和FUNCTION有明显区别：

• 如果用PROCEDURE，要返回数据的话，需要声明RETURNS TABLE或者在过程体内用RETURN NEXT/RETURN QUERY语句
• 更推荐用FUNCTION来返回数据集，处理起来更省心，比如：

CREATE OR REPLACE FUNCTION get_home_data()
RETURNS TABLE (id INT, title VARCHAR, content TEXT) AS $$
BEGIN
    RETURN QUERY SELECT id, title, content FROM home_content;
END;
$$ LANGUAGE plpgsql;

二、SQLAlchemy中解决"unnamed portal 1"的问题 #

你看到的"unnamed portal 1"是Postgres对未捕获的结果集的标识，这时候别用CALL语句调用，改用SELECT * FROM 函数名()的方式，就能直接获取结构化结果：

from sqlalchemy import text
from your_flask_app import db

# 用engine连接执行
with db.engine.connect() as conn:
    # 如果有参数，直接通过字典传递
    result = conn.execute(text("SELECT * FROM get_home_data(:user_id)"), {"user_id": 123})
    # 转成字典列表，方便在Flask视图中直接使用
    home_data = result.mappings().all()
    # 如果是单列结果，可以用scalars().all()简化
    # home_data = result.scalars().all()

如果一定要用PROCEDURE，那调用后需要手动捕获portal的结果：

result = conn.execute(text("CALL your_procedure(); FETCH ALL IN \"unnamed portal 1\";"))
home_data = result.fetchall()

但这种方式不够优雅，还是用FUNCTION更稳妥。

三、psycopg2中解决返回None的问题 #

psycopg2调用PROCEDURE后，默认不会自动获取结果集，需要手动切换到结果集再读取：

import psycopg2

conn = psycopg2.connect(dbname="your_db", user="your_user", password="your_pwd", host="localhost")
cur = conn.cursor()

# 调用存储过程（带参数的话传元组）
cur.callproc("your_procedure_name", (param1, param2))
# 切换到结果集
cur.nextset()
# 现在就能正常获取数据了
home_data = cur.fetchall()

cur.close()
conn.commit()
conn.close()

同样，换成FUNCTION用SELECT调用会更简单直接：

cur.execute("SELECT * FROM get_home_data(%s)", (param1,))
home_data = cur.fetchall()

这样就不会返回None了。

关键总结 #

• 优先用FUNCTION替代PROCEDURE来返回数据集，Postgres对FUNCTION的结果处理更符合常规查询逻辑
• 不要用CALL调用返回数据的存储对象，改用SELECT * FROM ...的方式
• 不管是SQLAlchemy还是psycopg2，获取结果时要确保调用对应的方法（比如fetchall()、mappings().all()）

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