Contents

require 'csv'

module Chartio
  class CSVFormatter < Formatter

    attr_accessor :foreign_keys

    def initialize
      @foreign_keys = []
      @stringio = StringIO.open
      @csv = CSV.new(@stringio)
    end

    def foreign_key(incoming_foreign_key)
      @foreign_keys << incoming_foreign_key
    end

    def output_report
      @csv << Chartio::ForeignKeyRelationship::FIELDS.map(&:to_s)

      @foreign_keys.each do |key|
        @csv << key.to_a
      end

      @csv.string
    end

  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
chartio-rails-0.1.0	lib/chartio/csv_formatter.rb
chartio-rails-0.0.2	lib/chartio/csv_formatter.rb
chartio-rails-0.0.1	lib/chartio/csv_formatter.rb