将 csv 导入 ruby​​ 但需要创建关系

Question

我想导入 2 个文件（县和州）。当我导入时，我需要创建与县到州的关系。

我的州 CSV 文件有：

• id
• state 例如“康涅狄格”
• abbreviation 例如“CT”

我的县 CSV 文件有：

• id
• name 例如“哈特福德”
• market_state 例如“康涅狄格”

MarketCounty 的属性是：

create_table "market_counties", force: :cascade do |t|
  t.string   "name"
  t.integer  "market_state_id"
  t.datetime "created_at",      null: false
  t.datetime "updated_at",      null: false
end

MarketState 的属性是：

create_table "market_states", force: :cascade do |t|
  t.string   "name"
  t.string   "abbreviation"
  t.datetime "created_at",   null: false
  t.datetime "updated_at",   null: false
end

我的模型有这些关系：

市场状态：

class MarketState < ActiveRecord::Base
  has_many :market_reports, as: :location
  has_many :market_cities
  has_many :market_zips
  has_many :market_counties  
end

市场县：

class MarketCounty < ActiveRecord::Base
  has_many :market_reports, as: :location
  has_many :market_cities
  has_many :market_zips
  belongs_to :market_state
end

我的耙子任务：

def import_market_states
  MarketState.create!(name: "Connecticut", abbreviation: "CT")
end

def import_market_counties
  path = Rails.root.join("config/csv/locations/counties.csv")
  CSV.foreach(path, headers: true) do |row|
    MarketCounty.create! row.to_hash
  end

我的问题是方法import_market_counties，row.to_hash 不起作用。

ActiveRecord::AssociationTypeMismatch: MarketState(#70157557268540)
expected, got String(#70157439770720)

我手动设置的：

MarketCity.create!(name: "Hartford", market_state: MarketState.find_by(abbreviation: "CT")

导入县文件时如何在循环中进行“查找”（将其与 MarketState 模型相关联）？

澄清一下，问题是当我创建县时，我并没有尝试用 CT 填充州。我需要用关联填充 state_id。

我发现了这个：When importing a CSV, how do I handle data in a row that corresponds to an association? --- 并且认为它可能与我正在尝试做的类似 - 只是不知道如何在这里应用它。

Answer 1

假设您的 MarketState 有一个 name 列是状态的名称：

def import_market_counties


path = Rails.root.join("config/csv/locations/counties.csv")
  CSV.foreach(path, headers: true) do |row|
    MarketState.find_by_name( row['market_state'].strip )
  end

如果你想创建不存在的状态，你也可以使用 find_or_create_by。