module Formatter

Public Class Methods

format(data, type = 'csv') click to toggle source

# File lib/formatter.rb, line 7
def self.format(data, type = 'csv')
  if !['csv', 'json', 'pretty_print'].include?(type)
    raise 'Formatter: invalid format'
  end

  self.send(type.to_sym, data)
end

Private Class Methods

csv(data) click to toggle source

# File lib/formatter.rb, line 17
def self.csv(data)
  CSV.generate do |csv|
    csv << data.first.to_h.keys

    data.each do |d|
      csv << d.to_h.values
    end
  end
end

json(data) click to toggle source

# File lib/formatter.rb, line 27
def self.json(data)
  JSON.generate(data)
end