Contents

require 'csv'

module Spotlight
  class Resources::Csv < Spotlight::Resource
    mount_uploader :url, Spotlight::CsvUploader

    # we want to do this before reindexing
    after_save :update_exhibit_specific_fields, if: :url_changed?, prepend: true

    def to_solr
      csv.map do |row|
        h = {}
        row.each do |k,v|
          if label_to_field[k]
            h[label_to_field[k]] ||= []
            h[label_to_field[k]] << v
          end
        end
        h
      end
    end

    def label_to_field
      @label_to_field ||= begin
        label_to_field = {}
        label_to_field['id'] ||= ::SolrDocument.unique_key
        label_to_field[title_field_name] ||= exhibit.blacklight_config.index.title_field
        label_to_field[public_field_name] ||= Spotlight::SolrDocument.visibility_field(exhibit)
        label_to_field.merge! Hash[exhibit.blacklight_config.index_fields.map { |k,v| [v.label, v.field]}]
        label_to_field.merge! Hash[exhibit.blacklight_config.facet_fields.map { |k,v| [v.label, v.field]}]
        label_to_field
      end
    end

    def csv options={ headers: true }, &block
      CSV.new File.open(url.path, 'r'), options
    end

    private
    def update_exhibit_specific_fields

      csv.map do |row|
        sidecar_updates = {}
        row.each do |label,v|
          if custom_fields[label]
            sidecar_updates[custom_fields[label]] = v
          end
        end

        unless row[public_field_name].blank? and sidecar_updates.empty?
          sidecar = ::SolrDocument.new(id: row['id']).sidecar(exhibit)
          sidecar.update(
            public: row[public_field_name],
            data: sidecar.data.merge(sidecar_updates)
          )
        end
      end
    end

    def custom_fields
      Hash[exhibit.custom_fields.map { |f| [f.label, f.field] }]
    end

    def title_field_name
      "Title"
    end

    def public_field_name
      "Public"
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
blacklight-spotlight-0.2.0	app/models/spotlight/resources/csv.rb