无超级用户权限且无法用Shell命令时，如何在PostgreSQL中导入CSV数据？

方案1：Python + psycopg2 客户端批量导入 #

psycopg2的copy_from是客户端侧的COPY实现，无需超级用户权限，纯代码层面操作，完全脱离Shell依赖。

示例代码：

import psycopg2
from psycopg2 import sql

# 数据库连接配置
conn_config = {
    "dbname": "your_database",
    "user": "your_username",
    "password": "your_password",
    "host": "db_host",
    "port": "5432"
}

# 读取CSV并执行导入
with open('target_data.csv', 'r', encoding='utf-8') as csv_file, psycopg2.connect(**conn_config) as conn:
    cursor = conn.cursor()
    # 适配你的CSV格式和表结构
    cursor.copy_from(
        file=csv_file,
        table=sql.Identifier('your_target_table'),  # 避免SQL注入风险
        sep=',',
        columns=('col1', 'col2', 'col3'),  # 与CSV列顺序对应，可省略（需与表列顺序完全一致）
        header=True  # CSV首行为表头时开启
    )
    conn.commit()

方案2：Java + PostgreSQL JDBC 驱动导入 #

PostgreSQL JDBC驱动提供的CopyManager类支持客户端侧COPY操作，同样无需超级权限，纯应用层执行。

示例代码片段：

import org.postgresql.copy.CopyManager;
import org.postgresql.core.BaseConnection;

import java.io.FileReader;
import java.sql.Connection;
import java.sql.DriverManager;

public class CsvDbImporter {
    public static void main(String[] args) throws Exception {
        String dbUrl = "jdbc:postgresql://db_host:5432/your_database";
        String user = "your_username";
        String pwd = "your_password";

        try (Connection conn = DriverManager.getConnection(dbUrl, user, pwd);
             FileReader csvReader = new FileReader("target_data.csv")) {

            CopyManager copyManager = new CopyManager((BaseConnection) conn);
            // 构造适配CSV格式的COPY指令
            String copyCmd = "COPY your_target_table(col1, col2, col3) FROM STDIN WITH (FORMAT csv, HEADER true, DELIMITER ',')";
            copyManager.copyIn(copyCmd, csvReader);
        }
    }
}

核心逻辑说明 #

上述方案均基于客户端侧COPY实现：通过数据库连接将CSV数据流式传输至服务器，区别于服务器端COPY FROM（需超级权限访问本地文件），完全满足非Shell环境、无超级权限的导入需求。

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