在Python代码中嵌入psql命令时如何捕获错误信息？

一、背景场景 #

在bash控制台执行以下命令可成功导入数据：

psql -U postgres -d sample -c "copy data(f1,f2) from '/tmp/data.txt'  with delimiter ',' "

执行输出：

Pager usage is off.
Timing is on.
COPY 1
Time: 9.573 ms

移除with delimiter子句后触发错误，bash控制台输出：

psql -U postgres -d sample -c "copy data(f1,f2) from '/tmp/data.txt'  "

错误输出：

Pager usage is off.
Timing is on.
ERROR:  missing data for column "f2"
CONTEXT:  COPY data, line 1: ""x1","y1""
Time: 0.318 ms

尝试将该psql命令嵌入Python代码，试图捕获错误信息写入日志文件/tmp/log.txt，代码如下：

import os 
import logging
logging_file = '/tmp/log.txt' 
logging.basicConfig(filename=logging_file,level=logging.INFO,filemode='a+')
logger = logging.getLogger("import_data")

sql_string ="""
psql -U postgres -d sample -c "copy data(f1,f2) from '/tmp/data.txt'  " 
"""

try:
    os.system(sql_string)
except Exception as e:
    logger.info(e)

但错误信息并未写入日志文件，请问原因是什么？如何在Python代码中嵌入psql命令时捕获错误信息？

二、原因分析 #

• os.system()仅返回命令的退出状态码（0表示成功，非0表示失败），不会主动抛出异常，所以代码里的try-except块根本不会触发。
• psql的错误信息是输出到标准错误流（stderr），而os.system()不会捕获这个流的内容，自然无法写入日志文件。

三、解决办法 #

推荐使用subprocess模块替代os.system()，它能更灵活地控制子进程的输入输出，捕获标准输出和错误流。

方法1：捕获错误信息并写入日志 #

import logging
import subprocess

logging_file = '/tmp/log.txt' 
logging.basicConfig(filename=logging_file, level=logging.INFO, filemode='a+')
logger = logging.getLogger("import_data")

# 用列表形式定义命令，避免引号转义问题
cmd = [
    'psql', '-U', 'postgres', '-d', 'sample',
    '-c', "copy data(f1,f2) from '/tmp/data.txt'  "
]

try:
    # check=True：命令执行失败时抛出CalledProcessError
    # capture_output=True：捕获stdout和stderr内容
    # text=True：将输出转为字符串（而非字节流）
    result = subprocess.run(cmd, check=True, capture_output=True, text=True)
    logger.info(f"数据导入成功：{result.stdout.strip()}")
except subprocess.CalledProcessError as e:
    logger.error(f"数据导入失败，退出码：{e.returncode}")
    logger.error(f"错误详情：{e.stderr.strip()}")

方法2：直接重定向错误到日志文件（简单但灵活性低） #

如果不需要在代码中处理错误内容，也可以在命令里直接将stderr重定向到日志文件：

import os

sql_string = """
psql -U postgres -d sample -c "copy data(f1,f2) from '/tmp/data.txt'  " 2>> /tmp/log.txt
"""

os.system(sql_string)

补充说明 #

• 使用subprocess.run()时，capture_output=True等价于同时设置stdout=subprocess.PIPE和stderr=subprocess.PIPE；
• text=True（或旧版本的universal_newlines=True）会把输出从字节流转为字符串，方便直接处理。

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