queryable_pstore 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 42e7acf8a72dd7e9de68bbe669c101bb9beee30d
-  data.tar.gz: eb8fa9814c3e75a29be7635205df0dcc7b2a4231
+  metadata.gz: 4ea6d2cc0b46d969b9696c064351b8f6ac70438c
+  data.tar.gz: 843721ce93a69ae6868a78d973411b71c99774b0
 SHA512:
-  metadata.gz: 92a03d212cae19ea54199225aa4914f20e13ba3190c920d1973516b13b42ae929bd36fb5efaf422ab0d50c0a560405cbb81b51aa81c0e13a4dc975590c04676e
-  data.tar.gz: 905e16e37c196b7882537b3f4215a1ebd9f9884346c618ffe7287a1d6e53a2b9d0cd3c27fc8dd75410d54f0d67e8382b679f179d9253ac0a3bc2d757389057ac
+  metadata.gz: 8aa31e61c7ed278c590adafb22ce818cfcc4baad60746384153c2a579ddf2e5f0a39a679211fc7150dbc87ec6f8b16b715ec04bb7d3c7696124cd459e73cbd5b
+  data.tar.gz: 146000712f4ae9b0912d7c5c19d712c77b6bbaebace3b3bbceb91e52be93b738896d2e07cc8b7ba3baea41ad8b77c7c63d07b4d5f30b6216b4378bf9f64f2b0f

data/lib/csv_importer.rb

@@ -1,8 +1,27 @@
+require 'pry'
 class CSVImporter
-  def self.import_csv(filename, opts = {})
-    csv_converters = opts.fetch(:convert, []).map { |conversion| CSVConverter.new(conversion) }
+  attr_reader :original_headers, :headers_and_type
 
+  def import_csv(filename, opts = {})
     csv = CSV.read(File.open(filename), headers: true, header_converters: -> (header) { convert_header_to_methodable_name(header) })
+    store = create_pstore(csv, opts, filename)
+    store.csv_importer = self
+    set_headers_and_type(@original_headers, store)
+
+    store
+  end
+
+  def import_csv_from_string(string, opts = {})
+    csv = CSV.parse(string, headers: true, header_converters: -> (header) { convert_header_to_methodable_name(header) })
+    store = create_pstore(csv, opts)
+    store.csv_importer = self
+    set_headers_and_type(@original_headers, store)
+
+    store
+  end
+
+  def create_pstore(csv, opts = {}, filename = SecureRandom.uuid)
+    csv_converters = opts.fetch(:convert, []).map { |conversion| CSVConverter.new(conversion) }
 
     store = QueryablePStore.new("#{filename}.pstore")
     store.transaction do
@@ -14,7 +33,24 @@ class CSVImporter
     store
   end
 
-  def self.convert_header_to_methodable_name(header)
+  def convert_header_to_methodable_name(header)
+    @original_headers ||= []
+    @original_headers << header
+
     header.downcase.gsub(/[^a-z]/, "_").to_sym
   end
+
+  private
+
+  def set_headers_and_type(headers, store)
+    headers_and_type = {}
+    store.results.first.each_with_index do |key_value, index|
+      value = key_value.last
+
+      original_header = headers[index]
+      headers_and_type[original_header] = value.class
+    end
+
+    @headers_and_type = headers_and_type
+  end
 end

data/lib/queryable_pstore.rb

@@ -8,8 +8,20 @@ require_relative 'csv_converter'
 require_relative 'csv_importer'
 
 class QueryablePStore < PStore
-  extend SingleForwardable
-  def_delegator :CSVImporter, :import_csv
+  class << self
+    def import_csv(filename, opts = {})
+      CSVImporter.new.import_csv(filename, opts)
+    end
+
+    def import_csv_from_string(string, opts = {})
+      CSVImporter.new.import_csv_from_string(string, opts)
+    end
+  end
+
+  extend  Forwardable
+  attr_accessor :csv_importer
+  def_delegator :@csv_importer, :original_headers
+  def_delegator :@csv_importer, :headers_and_type
 
   def initialize(store_name)
     super(store_name)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: queryable_pstore
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - Jacob Smith