0

我遵循 railscasts #396 Importing CSV 并在我的 rails 项目中实现了 CSV 上传。

这是我的视图文件：

<%= form_tag import_customers_path, multipart: true do %>
  <%= file_field_tag :file %>
  <%= submit_tag "Import" %>
<% end %>

这是我的控制器动作：

def import
  current_user.customers.import(params[:file])
  redirect_to customers_path, notice: "Users imported."
end

这些是我的模型方法：

def self.to_csv(options = {})
    CSV.generate(options) do |csv|
        csv << column_names
        all.each do |customer|
            csv << customer.attributes.values_at(*column_names)
        end
    end
end

def self.import(file)
  CSV.foreach(file.path, headers: true) do |row|
    Customer.create! row.to_hash
  end
end

在这里，我不希望用户在 CSV 中包含标题。当我替换headers: true为 时headers: false，出现错误：

客户控制器中的 NoMethodError#import

["abc@wer.com"]:Array 的未定义方法 `to_hash'

谁能告诉如何在不需要标题行的情况下上传 CSV 文件？

4

2 回答 2

1

就 CSV 文件的上传和处理而言，您已经非常非常接近了。Customer.create!您只是在通过调用读取数据行以填充数据库时遇到问题

看起来您一直在使用只有一行数据的 CSV 文件进行测试。使用headers: true，该单行被转换为标题，随后在CSV.foreach迭代器中被忽略。因此，实际上，文件中没有数据，也没有发生迭代。如果输入文件中有两行数据，无论如何你都会遇到错误。

现在，当您使用 时headers: false，该行数据被视为数据。这就是问题所在：处理数据不正确。

由于您的问题中没有模式，因此我假设在字段上有一点余地；您应该能够很容易地推断以使其适用于您的情况。这段代码展示了它是如何工作的：

CSV.parse(csv_data, headers: false) do |row|
  hash = {
    first_name: row[0],
    last_name: row[1],
    age: row[2],
    phone: row[3],
    address: row[4]
  }
  Customer.create!(hash)
end

如果您想要一个带有标题的 CSV 版本，这在这种情况下会很好用，并且具有不允许任意访问不应从外部源分配的列的好处：

CSV.parse(csv_data, headers: true, header_converters: :symbol) do |row|
  hash = {
    first_name: row[:first_name],
    surname: row[:last_name],
    birth_year: Date.today - row[:age],
    phone: row[:phone],
    street_address: row[:address]
  }
  Customer.create!(hash)
end

请注意，Customer#to_csv您的模型中的 也不完全正确。首先，它创建带有标题的 CSV 文件，因此您将无法使用此实现导出然后再次导入。接下来，标题字段变量column_names实际上并未在此代码中定义。最后，代码不控制写入 CSV 的列的顺序，这意味着标题和值可能会不同步。一个正确的（非标题）版本非常简单：

csv_data = CSV.generate do |csv|
  csv.each do |customer|
    csv << [customer.first_name, customer.last_name, customer.age, customer.phone, customer.address]
  end
end

基于标头的版本是这样的：

csv_data = CSV.generate do |csv|
  csv << ["First Name","Last Name","Age","Phone","Address"]
  csv.each do |customer|
    csv << [customer.first_name, customer.last_name, customer.age, customer.phone, customer.address]
  end
end

就个人而言，我会使用基于标题的版本，因为它更加健壮，而且很容易理解哪些列是哪些列。如果您曾经收到过无标题的 CSV 文件，并且不得不弄清楚如何在没有任何键的情况下理解它，那么您就会知道为什么标题很重要。

于 2016-05-06T00:39:08.237 回答
-1

您可以将 CSV 文件加载到数组数组中并删除第一行：

data = CSV.read("path/to/file.csv")
data = data[1..-1]

但是，这只会将数据存储为值数组。

当您使用headers: true它时，它使用一个哈希，其中键是列标题名称。

于 2013-07-27T21:22:46.287 回答