如何在Python脚本中输出PostgreSQL存储过程的RAISE NOTICE信息？

嗨，我来帮你搞定这个问题！当用psycopg2调用PostgreSQL存储过程时，默认不会自动显示RAISE NOTICE的输出内容，但我们可以通过设置连接的通知处理回调来轻松捕获并打印这些信息。

具体步骤如下： #

• 定义一个回调函数，用来处理PostgreSQL发送过来的NOTICE消息
• 将这个回调函数绑定到你的数据库连接上
• 正常调用存储过程，此时所有的RAISE NOTICE内容都会被捕获并输出

修改后的完整Python脚本： #

import psycopg2

# 定义处理NOTICE的回调函数
def handle_notice(conn, notice):
    # 提取并打印NOTICE的消息内容
    print(f"PostgreSQL Notice: {notice.message.strip()}")

# 建立数据库连接
conn = psycopg2.connect(host="", database="", user="", password="")
print("Database Connected")

# 将回调函数绑定到连接，开启NOTICE捕获
conn.set_notice_processor(handle_notice)

cur = conn.cursor()
# 调用你的存储过程（注意和你创建的存储过程名称保持一致）
cur.callproc('checktime')

# 如果存储过程涉及写操作，记得提交事务（根据业务需求决定）
# conn.commit()

# 关闭游标和连接
cur.close()
conn.close()

关键点解释： #

• handle_notice函数会在每次PostgreSQL发送NOTICE时被触发，notice.message就是RAISE NOTICE输出的原始内容，用strip()可以去掉多余的换行和空格
• conn.set_notice_processor(handle_notice)是核心步骤，它告诉psycopg2用我们定义的函数来处理所有通知类消息
• 注意存储过程名要和你创建的一致：你提供的存储过程是checktime，但原脚本里写的是rt_visits_function_gz，记得修正成正确的名称

修正后的PostgreSQL存储过程示例（方便对照）： #

create or replace function checktime() returns void language plpgsql as $$ 
DECLARE 
    timestart TIMESTAMP; 
BEGIN
    FOR id in (SELECT id FROM rt) LOOP  -- 补充完整查询语法，原示例的FOR语句存在语法问题
        SELECT timeofday() into timestart; 
        RAISE NOTICE 'Time now : %', timestart; 
    END LOOP; 
END; 
$$ ;

现在运行Python脚本，就能看到存储过程循环中每次输出的RAISE NOTICE内容啦！

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