导入 CSV 时，如何处理与关联对应的行中的数据？

Question

我正在关注Import CSV Railscast，它是直截了当的。

问题在于它只处理一个 csv 文件，该文件仅包含 1 个文件中的 1 个模型中的信息。

假设我有一个 CSV 文件，我正尝试将其导入到我的 Listing 模型中。在每一行/列表中，它都有一个名为 Building 的列，其中的值实际上是该列表的建筑属性的名称（即 @listing.building.name）。

我如何处理导入中的这些情况？

这是 Ryan 在 Railscast 中获得的壁橱，在他的案例中是 Product 模型：

def self.import(file)
  CSV.foreach(file.path, headers: true) do |row|
    product = find_by_id(row["id"]) || new
    product.attributes = row.to_hash.slice(*accessible_attributes)
    product.save!
  end
end

所发生的只是他正在检查产品是否存在，如果存在则更新属性。如果没有，则创建一个新的。

不太清楚在这种情况下如何处理关联...特别是考虑到需要发生的情况是如果关联记录不存在，则需要在此过程中创建它。

所以回到我之前的building.name 示例，如果没有Building.find_by_name(name)，那么它应该创建一个新的建筑记录。

想法？

Answer 1

试试这个

def self.import(file)
  CSV.foreach(file.path, headers: true) do |row|
    product = find_by_id(row["id"]) || new
    product.attributes = row.to_hash.slice(*accessible_attributes)
    product.save!

    building = product.buildings.find_by_name(row['building_name'])
    building ||= product.buildings.build
    building.attributes = row.to_hash.slice(*build_accessible_attributes)
    building.save!
  end
end

更新：使用新的 rails 3 方法更新答案

def self.import(file)
  CSV.foreach(file.path, headers: true) do |row|
    product = where(id: row["id"])
      .first_or_create!(row.to_hash.slice(*accessible_attributes))

    product.buildings.where(name: row['building_name'])
      .first_or_create!(row.to_hash.slice(*building_accessible_attributes))
  end
end