CODEWITHSUNDEEP
ruby-on-railsrubyactiverecordactiverecord-import
TheGuy
TheGuy

Reputation: 349

Ruby CSV | How to skip few columns while reading CSV?

I am bulk importing data from a CSV file into database using Active record Import gem. But I want to skip few CSV columns when importing.

For example, my xaa.csv has headers like name, author, author_id, rating. While importing, I want to skip the 'author_id' column values and import all the other columns.

books = CSV.read("/public/xaa.csv") //What more should I do here to skip the 3rd column
columns = [:name, :author, :rating]
Book.import columns, books, :validate => false

Upvotes: 1

Views: 2579

Answers (2)

andynu
andynu

Reputation: 722

books = CSV.readlines(path, headers:true).map{|row| row.values_at('name','rating') }

vegetaras's version is slightly more memory efficient.

Upvotes: 2

vegetaras
vegetaras

Reputation: 119

books = []

CSV.foreach('/public/xaa.csv', headers: true) do |row|
  books << [row['name'], row['author'], row['rating']]
end

columns = [:name, :author, :rating]
Book.import(columns, books, validate: false)

Upvotes: 7

Related Questions