Contents

require 'csv'

module SpeakyCsv
  # Imports a csv file as attribute hashes.
  class AttrImport
    include Enumerable

    attr_accessor :errors

    def initialize(config, input_io)
      @config = config
      @input_io = input_io
      @errors = ActiveModel::Errors.new(self)
    end

    # yields successive
    def each
      errors.clear
      block_given? ? enumerator.each { |a| yield a } : enumerator
    end

    private

    def enumerator
      Enumerator.new do |yielder|
        begin
          csv = CSV.new @input_io, headers: true

          csv.each do |row|
            attrs = {}

            row.headers.compact.each do |h|
              next unless @config.fields.include?(h.to_sym)
              next if @config.export_only_fields.include?(h.to_sym)
              attrs[h] = row.field h
            end

            headers_length = row.headers.compact.length
            pairs_start_on_evens = headers_length.even?
            (headers_length..row.fields.length).each do |i|
              i.send(pairs_start_on_evens ? :even? : :odd?) || next
              row[i] || next

              m = row[i].match(/^(\w+)_(\d+)_(\w+)$/)
              m || next
              has_many_name = m[1].pluralize
              has_many_index = m[2].to_i
              has_many_field = m[3]
              has_many_value = row[i + 1]

              has_many_config = @config.has_manys[has_many_name.to_sym]

              next unless has_many_config
              next unless has_many_config.fields.include?(has_many_field.to_sym)
              next if has_many_config.export_only_fields.include?(has_many_field.to_sym)

              attrs[has_many_name] ||= []
              attrs[has_many_name][has_many_index] ||= {}
              attrs[has_many_name][has_many_index][has_many_field] = has_many_value
            end

            yielder << attrs
          end

        rescue CSV::MalformedCSVError
          errors.add :csv, "is malformed: #{$!.message}"
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
speaky_csv-0.0.1	lib/speaky_csv/attr_import.rb