使用subprocess调用psql时如何显示RAISE NOTICE输出？

问题根源 #

PostgreSQL的RAISE NOTICE语句输出的信息，会被psql发送到标准错误流（stderr），而非标准输出流（stdout）。你之前的代码要么只读取stdout，要么没有正确处理stderr内容，所以看不到进度提示。

解决方案1：实时捕获并输出stderr（Popen方式） #

修改你的Popen代码，将stderr合并到stdout后实时读取，或者通过线程分别监听两个流：

方式A：合并stderr到stdout #

import subprocess

# 替换为你的psql执行命令，例如："psql -h 远程地址 -U 用户名 -d 数据库名 -f 批量更新脚本.sql"
process = subprocess.Popen(
    query_command,
    shell=True,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,  # 将stderr输出合并到stdout
    text=True,
    env={"PGPASSWORD": db_pass}
)

# 实时读取并打印输出
while True:
    output = process.stdout.readline()
    if output == '' and process.poll() is not None:
        break
    if output:
        print(output.strip())

# 处理执行结束后剩余的输出
remaining_output = process.communicate()[0]
if remaining_output:
    print(remaining_output.strip())

方式B：线程分别监听stdout和stderr（如需区分流） #

import subprocess
from threading import Thread

def read_stream(stream, print_func):
    for line in iter(stream.readline, ''):
        print_func(line.strip())

process = subprocess.Popen(
    query_command,
    shell=True,
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE,
    text=True,
    env={"PGPASSWORD": db_pass}
)

# 启动线程分别读取两个流
stdout_thread = Thread(target=read_stream, args=(process.stdout, print))
stderr_thread = Thread(target=read_stream, args=(process.stderr, print))

stdout_thread.start()
stderr_thread.start()

# 等待线程结束
stdout_thread.join()
stderr_thread.join()

process.wait()

解决方案2：使用subprocess.run捕获输出 #

如果不需要实时输出，只想在执行完成后查看所有信息，可以修改参数合并流或分别打印：

合并流打印 #

import subprocess

result = subprocess.run(
    query_command,
    shell=True,
    text=True,
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,
    env={"PGPASSWORD": db_pass}
)

print(result.stdout)

区分stdout和stderr打印 #

import subprocess

result = subprocess.run(
    query_command,
    shell=True,
    text=True,
    capture_output=True,
    env={"PGPASSWORD": db_pass}
)

if result.stdout:
    print("标准输出内容：")
    print(result.stdout.strip())

if result.stderr:
    print("\n标准错误内容（含NOTICE进度）：")
    print(result.stderr.strip())

额外注意事项 #

• 确保psql命令没有添加-q（安静模式）参数，该参数会抑制NOTICE这类提示信息。
• 确认批量更新SQL脚本中的RAISE NOTICE语法正确，percentage_complete变量已正确计算并传入。

内容的提问来源于stack exchange，提问作者rerere L.