module UsefulRenderers::CsvRenderable

Public Instance Methods

to_csv(options = {}) click to toggle source

Converts an array to CSV formatted string Options include: :only => [:col1, :col2] # Specify which columns to include :except => [:col1, :col2] # Specify which columns to exclude :translate => boolean # Translate headers

# File lib/useful_renderers/csv_renderable.rb, line 12
def to_csv(options = {})
  klass = first.class
  return '' if empty?
  return join(',') unless klass.respond_to? :column_names

  columns = options[:only] ? options[:only].map(&:to_s) : klass.column_names
  columns -= options[:except].map(&:to_s) if options[:except]

  headers = columns.dup
  headers.map!{|col| klass.human_attribute_name col } if options[:translate]

  csv_options = {
    encoding: 'utf-8',
    headers: headers,
    write_headers: true
  }

  CSV.generate(csv_options) do |row|
    self.each do |obj|
      row << columns.map do |column|
        column.to_s.split('.').inject(obj) { |o, m| o.try(m) }
      end
    end
  end
end