Module: UsefulRenderers::CsvRenderable

Defined in:
lib/useful_renderers/csv_renderable.rb

Instance Method Summary collapse

Instance Method Details

#to_csv(options = {}) ⇒ Object

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



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# 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