Rails:CSV.generate和重命名标头

时间:2015-06-25 01:19:27

标签: ruby-on-rails csv ruby-on-rails-4

我将一堆记录导出到CSV,如下所示:

def self.to_csv
  attributes = %w(full_name first_name last_name created_at status)

  CSV.generate(headers: true) do |csv|
    csv << attributes
    all.each do |lead|
      csv << attributes.map { |attr| lead.send(attr) }
    end
  end
end

但我需要标题为Full Name而不是full_name。我需要一个哈希来匹配名称,但是如何将新的头名称写入 CSV.generate中的

更新查找哈希。

def lookup
  {
    full_name: 'Full Name', first_name: 'First Name', last_name: 'Last Name', created_at: 'Generation Date', status: 'Status'
  }
end

1 个答案:

答案 0 :(得分:1)

在属性数组上调用titalize可以帮助您实现目标。

def self.to_csv
  CSV.generate(headers: true) do |csv|
    csv << lookup.values
    all.each do |lead|
      csv << lookup.keys.map { |attr| lead.send(attr) }
    end
  end
end