Question

我是数据库管理的新手，我们正在使用psql。我需要做的就是将csv（大约200个表）迁移到我们的数据库。为每个csv文件手动创建表都有点烦人，所以请帮帮我，有没有办法从csv文件生成表？

Answer 1

由 OP 回答 at DBA Stackexchange。我在这里复制答案，因为这是我的搜索引擎返回的第一个链接。

OP 编写了如下脚本：

DATADIR='data'                      # this directory name


PREFIX='jobd'
DBNAME='divacsv'

function createSchema {
    COLUMNS=`head -n 1 $1 | 
        awk -F, '{for(i=1; i<=NF; i++){out=out $i" text, ";} print out;}' | 
        sed 's/ text, $/MYEXTRA text/' | 
        sed 's/"//g'`
    CMD_CREATE="psql $DBNAME -c \"CREATE TABLE $2 ($COLUMNS);\""

    echo $CMD_CREATE
    sh -c "$CMD_CREATE"

    CMD_COPY="psql divacsv  -c \"COPY $2 FROM '"`pwd`"/$1' DELIMITER ',' CSV;\""
    echo $CMD_COPY
    sh -c "$CMD_COPY"
}

for file in $DATADIR/*.csv; do
    table=$PREFIX"_"`echo $file | sed 's/.*\///' | sed 's/.csv//' `
    createSchema "$file" $table
done

评论建议可能需要使用 HEADER 以避免加载带有标题文本的第一行，这是真的。

我已经测试了这段代码，但无法在 CentOS 下运行。

如何在postgres sql中从csv生成表

1 个答案: