使用 ruby​​ CSV 模块导入 csv 文件时出现问题

Question

我正在尝试使用 Ruby 的 csv 模块将 csv 文件中包含的记录导入到 Ruby on Rails 3 应用程序中的本地表中。

该表是通过创建模型 Movie 来创建的。

以下是我在控制台中执行的操作：

require 'csv'
CSV.foreach('public/uploads/VideoTitles2.csv') do |row|
    record = Movie.new(
        :media_format   => row[0], 
        :title          => row[1], 
        :copies_at_home => row[2], 
        :order          => row[3]
    )
    record.save
end

csv 文件的行与它们传递到的列相匹配（在数据类型中）。这是我尝试导入的 csv 文件 (VideoTitles2.csv) 的缩短版本：

"DVD","LEAP OF FAITH",1,1
"DVD","COCOON",1,2
"DVD","TITANIC",1,3

我相信每条记录均以 \n 分隔。此 csv 文件是从 Access 导出的，其原始文件扩展名为 .txt。为了导入，我已手动将其更改为 .csv。

问题是，在 Rails 控制台中执行上述行后，我得到以下输出：

=> nil

导入似乎没有发生。如果有人知道如何解决这个问题，我将非常感激。

Answer 1

我没有看到问题所在。此代码片段返回 nil，因为 CSV.foreach 返回 nil，但这并不指示循环是否运行。您是否检查过是否创建了任何Movie？您是否包含任何调试行来跟踪该过程？

您可能想要检查 record.save 的输出（或调用 record.save!），也许验证错误阻止了记录的创建。另外，如果你想让循环返回创建的记录，你可以这样写(Ruby >= 1.8.7)：

require 'csv'

records = CSV.foreach('public/uploads/VideoTitles2.csv').map do |media_format, title, copies_at_home, order|     
    Movie.create!({
        media_format: media_format, 
        title: title, 
        copies_at_home: copies_at_home, 
        order: order,
    })
end

Answer 2

好吧，我有两件事错了：

1. 导出的 csv 文件不应在字符串周围包含引号 - 我刚刚删除了它们。

2. 感谢 tokland，记录。保存！是必要的（与我正在执行的 record.save 相反） - 验证错误阻止了创建记录。

总而言之，可以在创建模型/表 Movie 后创建以下函数：

class Movie < ActiveRecord::Base
  attr_accessible :media_format, :title, :copies_at_home, :order
  require 'csv'

  def self.import_movies()
    CSV.foreach('public/uploads/movies.csv') do |row|
        record = Movie.new(
            :media_format   => row[0], 
            :title          => row[1], 
            :copies_at_home => row[2], 
            :order          => row[3]
        )
        record.save!  
    end
  end
end

其中 movie.csv 如下所示：

Blu-ray, Movie 1, 1, 1
DVD, Movie 2, 1, 2
Blu-ray, Movie 3, 1, 3

然后在控制台中调用此函数：

Movie.import_movies()

正如预期的那样，所有内容都将在控制台中返回将是：

=> nil

检查您的索引视图（如果您已经创建了索引视图），您应该看到记录已成功导入到电影表中。