我正在从CSV文件中获取数据并将其保存到数据库中。这是简化的代码结构(这是一个小任务):
... loading CSV tools...
car = Car.new
car.tender_load_id = data.element[8]
car.brand = data.element[2]
car.color = 'green'
...
car.build_car_metadata(mbol: help_var[:mbol], ...)
car.first_registration = data.element[21]
car.skip_registration_code_validation = true # for not creating registration_code
if existing_car = Car.where("cars.tender_load_id ILIKE ?", "%#{car.tender_load_id}%").first
# car already exists => update
existing_car.update(car)
puts "Car exists, updating car with ID #{existing_car.id}."
else
car.save!
end
当我运行此任务时,收到以下错误消息:
NoMethodError: undefined method `reject' for #<Car:0x007fa3e2063a78>
,它指出了这一行:
existing_car.update(car)
我如何进行这项工作?不幸的是,我无法将if-else
部分放在rake任务的开始,其结构需要像这样。
谢谢。
答案 0 :(得分:1)
您需要仅使用属性更新现有汽车,而不能传递完整模型。
构建属性,然后根据汽车是否已经存在来创建或更新:
attributes = {}
attributes[:color] = ...
attributes[:brand] = ...
attributes[:tender_load_id] = ...
if (car = Car.where("cars.tender_load_id ILIKE ?", "%#{attributes[:tender_load_id]}%").first)
car.update(attributes)
else
Car.create!(attributes)
end
注1:似乎可以有多辆具有相同tender_load_id
的汽车。这意味着对现有汽车的查询将返回任意记录。也许您想向该查询添加order
。
注2:查询方式似乎很脆弱。 CSV中没有更好的ID吗?
注3:以_id
结尾的属性通常是外键。因此,也许为此属性找到一个更好的名称。