PostgreSQL - 使用函数从csv导入 - csv文件具有带空格的列名

时间:2017-06-14 07:45:29

标签: postgresql postgresql-9.1 postgresql-9.3 postgresql-9.2 postgresql-9.4

我复制了函数(来自其中一个webportal并相应地修改)以将数据从csv文件复制到表。我的Csv文件是从报告工具动态生成的,列名中有空格(例如:" Date Desc"," item No")

是否可以使用以下函数用_替换空格:

create or replace function public.load_csv_file
(
target_table text,
csv_path text,
col_count integer
)

returns void as $$

declare

iter integer; -- dummy integer to iterate columns with
col text; -- variable to keep the column name at each iteration
col_first text; -- first column name, e.g., top left corner on a csv file or     spreadsheet

begin
set schema 'public';

create table insert_from_csv ();

-- add just enough number of columns
for iter in 1..col_count
loop
    execute format('alter table insert_from_csv add column col_%s text;', iter);
end loop;

-- copy the data from csv file
execute format('copy insert_from_csv from %L with delimiter '','' quote ''"'' csv ', csv_path);

iter := 1;
col_first := (select col_1 from insert_from_csv limit 1);

-- update the column names based on the first row which has the column names
for col in execute format('select unnest(string_to_array(trim(temp_table::text, ''()''), '','')) from temp_table where col_1 = %L', col_first)
loop
    execute format('alter table insert_from_csv rename column col_%s to %s', iter, col);
    iter := iter + 1;
end loop;

-- delete the columns row
execute format('delete from insert_from_csv where %s = %L', col_first, col_first);

-- change the temp table name to the name given as parameter, if not blank
if length(target_table) > 0 then
    execute format('alter table insert_from_csv rename to %I', target_table);
end if;

end;

$$ language plpgsql;

和 收到错误:

ERROR:  syntax error at or near "DESC"
LINE 1: delete from insert_from_csv where Date DESC = 'Date DESC'
                                                ^
QUERY:  delete from insert_from_csv where Date DESC = 'Date DESC'
CONTEXT:  PL/pgSQL function load_csv_file(text,text,integer) line 34 at EXECUTE

任何帮助表示感谢。

0 个答案:

没有答案