从python django中使用copy_from删除csv头?

时间:2016-01-14 09:25:14

标签: python django postgresql csv

我正在使用以下代码:

COPY Table_Name FROM 'wheat_crop_data.csv' DELIMITER ',' CSV HEADER;

出现以下错误:

  

类型为timestamp的输入语法无效:“Order Date”

这意味着应该有一些日期而不是订单日期

此处订单日期为标题。

在postgresql中:

BusServiceProvider

我怎么能在python django中做到这一点?

1 个答案:

答案 0 :(得分:2)

然后你应该砍掉第一行并将剩余部分输入光标:

from StringIO import StringIO

with open('uploaded_inventory_sheet.csv') as f: 
    next(f) # skip the first line
    content = StringIO('\n'.join(line for line in f))
    self.cur.copy_from(content, ...)

请注意,此解决方案将整个文件保存在内存中。如果这不是预期的行为,您可以使用临时中间文件。