使用Python导出PostgreSQL查询结果至CSV文件时遇语法错误

你遇到的这个错误很典型——\copy其实是psql客户端专属的命令，不是PostgreSQL数据库能直接执行的SQL语句，所以用cur.execute()去跑它肯定会报语法错误。

下面给你两种可行的解决方案：

方案一：使用psycopg2内置的copy_to方法 #

这是最推荐的方式，直接通过psycopg2的API来处理导出，不需要依赖psql客户端。代码示例如下：

#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import psycopg2

# 数据库连接参数
db_params = {
    "dbname": "dbname",
    "user": "user",
    "password": "password"
}

try:
    conn = psycopg2.connect(**db_params)
    cur = conn.cursor()
    
    # 打开本地文件准备写入
    with open('/mnt/results/month/table.csv', 'w', encoding='utf-8') as f:
        # 执行查询并导出为CSV，分隔符用;
        cur.copy_to(f, "(SELECT * FROM table WHERE month=6)", sep=';', format='csv')
    
    conn.commit()
    print("导出成功！")
except Exception as e:
    print(f"导出失败：{e}")
    conn.rollback()
finally:
    cur.close()
    conn.close()

方案二：通过subprocess调用psql命令执行\copy #

如果你坚持想用\copy逻辑，可以直接在Python里调用系统的psql命令，相当于在终端执行psql命令行。代码示例：

#!/usr/bin/python
# -*- coding: utf-8 -*-
import subprocess
import os

# 设置数据库密码环境变量（避免命令行明文暴露密码）
os.environ['PGPASSWORD'] = 'password'

# 构造psql命令
psql_cmd = [
    'psql',
    '-d', 'dbname',
    '-U', 'user',
    '-c', r"\copy (SELECT * FROM table WHERE month=6) TO '/mnt/results/month/table.csv' WITH CSV DELIMITER ';';"
]

try:
    result = subprocess.run(psql_cmd, check=True, capture_output=True, text=True)
    print("导出成功！")
except subprocess.CalledProcessError as e:
    print(f"导出失败：{e.stderr}")

注意事项 #

• 方案一中，copy_to默认不会包含表头，如果需要表头的话，你需要先手动写入表头行，或者查询表的字段名来生成表头。
• 方案二中，要确保服务器上的Python进程有执行psql的权限，并且psql客户端所在用户有目标文件路径的读写权限（因为\copy是在客户端侧处理文件的）。

内容的提问来源于stack exchange，提问作者Raúl Casado