Как добавить новый столбец в строку CSV в рельсах - PullRequest
Новая
       5

Как добавить новый столбец в строку CSV в рельсах

0 голосов
/ 03 марта 2019

Я хочу добавить новый столбец и обновить существующие значения в ответе CSV.Как я могу сделать более простой и лучший способ сделать следующие преобразования?

Ввод 

id,name,country
1,John,US
2,Jack,UK
3,Sam,UK

Я использую следующий метод для анализа строки CSV и добавления нового столбца 

# Parse original CSV
rows = CSV.parse(csv_string, headers: true).collect do |row|
  hash = row.to_hash
  # Merge additional data as a hash.
  hash.merge('email' => 'sample@gmail.com')
end

# Extract column names from first row of data
column_names = rows.first.keys
# Generate CSV after transformation of csv
csv_response = CSV.generate do |csv|
  csv << column_names
  rows.each do |row|
    # Extract values for row of data
    csv << row.values_at(*column_names)
  end
end

Я использую следующий метод для анализа CSV и обновлениясуществующие значения

name_hash = {"John" => "Johnny", "Jack" => "Jackie"}

Разобрать оригинальный CSV 

rows = CSV.parse(csv_string, headers: true).collect do |row|
  hash = row.to_hash
  hash['name'] = name_hash[hash['name']] if name_hash[hash['name']] != nil
  hash
end

# Extract column names from first row of data
column_names = rows.first.keys
# Generate CSV after transformation of csv
csv_response = CSV.generate do |csv|
  csv << column_names
  rows.each do |row|
    # Extract values for row of data
    csv << row.values_at(*column_names)
  end
end

1 Ответ

0 голосов
/ 03 марта 2019

Один из возможных вариантов, учитывая следующие справочные данные, которые будут использоваться для изменения таблицы: 

name_hash = {"John" => "Johnny", "Jack" => "Jackie"}
sample_email = {'email' => 'sample@gmail.com'}

Просто сохраняйте в строках таблицу, преобразованную в хэш: 

rows = CSV.parse(csv_string, headers: true).map(&:to_h)
#=> [{"id"=>"1", "name"=>"John", "country"=>"US"}, {"id"=>"2", "name"=>"Jack", "country"=>"UK"}, {"id"=>"3", "name"=>"Sam", "country"=>"UK"}]

Затем измените хэш на основе справочных данных (я использовал Object # затем для псевдонима Ruby 2.6.1 Object # yield_self для Ruby 2.5): 
rows.each { |h| h.merge!(sample_email).then {|h| h['name'] = name_hash[h['name']] if name_hash[h['name']] } }
#=> [{"id"=>"1", "name"=>"Johnny", "country"=>"US", "email"=>"sample@gmail.com"}, {"id"=>"2", "name"=>"Jackie", "country"=>"UK", "email"=>"sample@gmail.com"}, {"id"=>"3", "name"=>"Sam", "country"=>"UK", "email"=>"sample@gmail.com"}]

Наконец восстановите таблицу: 
csv_response = CSV.generate(headers: rows.first.keys) { |csv| rows.map(&:values).each { |v| csv << v } }

Итак, теперь у вас есть: 

puts csv_response

# id,name,country,email
# 1,Johnny,US,sample@gmail.com
# 2,Jackie,UK,sample@gmail.com
# 3,Sam,UK,sample@gmail.com
...