我有一个简单的 Rails 应用程序,用户需要通过 csv 或 Excel 上传内容。我并不总是想假设用户知道数据库中所有必填字段,我只想让用户知道第一列是必需的,并且第一列上的任何内容都会进入mobile_number
数据库中的列。我正在谈论的代码示例如下。我正在使用 roo gem。这是一个简单的应用程序,因此联系人表中只使用了两个数据库列。
def load_imported_contacts
spreadsheet = open_spreadsheet
header = spreadsheet.row(1)
(2..spreadsheet.last_row).map do |i|
row = Hash[[header, spreadsheet.row(i)].transpose]
contact = Contact.find_by_id(row["id"]) || Contact.new
contact.name = "content of the second field(optional)"
contact.mobile_number = "content in the first(required and must be the first column)"
contact.save!
end
end
def open_spreadsheet
case File.extname(file.original_filename)
when ".csv" then Csv.new(file.path, nil, :ignore)
when ".xls" then Excel.new(file.path, nil, :ignore)
when ".xlsx" then Excelx.new(file.path, nil, :ignore)
else raise "Unknown file type: #{file.original_filename}"
end
end