Contents

module Bmg
  module Writer
    class Csv
      include Writer

      DEFAULT_OPTIONS = {
      }

      def initialize(csv_options, output_preferences = nil)
        @csv_options = DEFAULT_OPTIONS.merge(csv_options)
        @output_preferences = OutputPreferences.dress(output_preferences)
      end
      attr_reader :csv_options, :output_preferences

      def call(relation, string_or_io = nil)
        require 'csv'
        string_or_io, to_s = string_or_io.nil? ? [StringIO.new, true] : [string_or_io, false]
        headers, csv = infer_headers(relation.type), nil
        relation.each do |tuple|
          if csv.nil?
            headers = infer_headers(tuple) if headers.nil?
            csv = CSV.new(string_or_io, csv_options.merge(headers: headers))
          end
          csv << headers.map{|h| tuple[h] }
        end
        to_s ? string_or_io.string : string_or_io
      end

    private

      def infer_headers(from)
        attrlist = if from.is_a?(Type) && from.knows_attrlist?
          from.to_attrlist
        elsif from.is_a?(Hash)
          from.keys
        end
        attrlist ? output_preferences.order_attrlist(attrlist) : nil
      end

    end # class Csv
  end # module Writer
end # module Bmg

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
bmg-0.18.2	lib/bmg/writer/csv.rb
bmg-0.18.1	lib/bmg/writer/csv.rb
bmg-0.18.0	lib/bmg/writer/csv.rb
bmg-0.17.8	lib/bmg/writer/csv.rb