queryable_pstore 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f84e3e92baf6549c12c1bec54a36a155461b11aa
-  data.tar.gz: d831ec8f2ab52d5911f70b06203857d85f5a8f7f
+  metadata.gz: 42e7acf8a72dd7e9de68bbe669c101bb9beee30d
+  data.tar.gz: eb8fa9814c3e75a29be7635205df0dcc7b2a4231
 SHA512:
-  metadata.gz: 6cfb2ca00c225b53ca498ce78a250701e8b33c203d21d00b66e97399d0a11cb5a5e48499e98b486987f3c88a39335af98adfafa0600af349f9102261ac4f5526
-  data.tar.gz: 2b71d3e531872a06a80b2a41f257fd50ae6a8a94a6c0b4a17f550010171a1ef3148df427538d35187aecaafcf08692a6fdf6d502e8f469df40d95602176c30b5
+  metadata.gz: 92a03d212cae19ea54199225aa4914f20e13ba3190c920d1973516b13b42ae929bd36fb5efaf422ab0d50c0a560405cbb81b51aa81c0e13a4dc975590c04676e
+  data.tar.gz: 905e16e37c196b7882537b3f4215a1ebd9f9884346c618ffe7287a1d6e53a2b9d0cd3c27fc8dd75410d54f0d67e8382b679f179d9253ac0a3bc2d757389057ac

data/lib/csv_converter.rb

@@ -1,20 +1,20 @@
-  class CSVConverter
-    def initialize(convert)
-      @key = convert.keys.first
-      @conversion = convert[@key]
-    end
-
-    def convert(hash)
-      case @conversion
-      when :integer
-        hash[@key] = hash[@key].to_i
-      when :float
-        hash[@key] = hash[@key].to_f
-      else
-        raise ArgumentError.new("Unknown converter: `#{@conversion}`")
-      end
+class CSVConverter
+  def initialize(convert)
+    @key = convert.keys.first
+    @conversion = convert[@key]
+  end
 
-      hash
+  def convert(hash)
+    case @conversion
+    when :integer
+      hash[@key] = hash[@key].to_i
+    when :float
+      hash[@key] = hash[@key].to_f
+    else
+      raise ArgumentError.new("Unknown converter: `#{@conversion}`")
     end
 
+    hash
   end
+
+end

data/lib/csv_importer.rb

@@ -0,0 +1,20 @@
+class CSVImporter
+  def self.import_csv(filename, opts = {})
+    csv_converters = opts.fetch(:convert, []).map { |conversion| CSVConverter.new(conversion) }
+
+    csv = CSV.read(File.open(filename), headers: true, header_converters: -> (header) { convert_header_to_methodable_name(header) })
+
+    store = QueryablePStore.new("#{filename}.pstore")
+    store.transaction do
+      csv.each do |row|
+        store[SecureRandom.uuid] = csv_converters.inject(row.to_h) { |hash, converter| converter.convert(hash) }
+      end
+    end
+
+    store
+  end
+
+  def self.convert_header_to_methodable_name(header)
+    header.downcase.gsub(/[^a-z]/, "_").to_sym
+  end
+end

data/lib/queryable_pstore.rb

@@ -5,23 +5,12 @@ require 'csv'
 
 require_relative 'query'
 require_relative 'csv_converter'
+require_relative 'csv_importer'
 
 class QueryablePStore < PStore
-  def self.import_csv(filename, opts = {})
-    csv_converters = opts.fetch(:convert, []).map { |conversion| CSVConverter.new(conversion) }
+  extend SingleForwardable
+  def_delegator :CSVImporter, :import_csv
 
-    csv = CSV.read(File.open(filename), headers: true, header_converters: -> (header) { header.downcase.to_sym })
-
-    store = new("#{filename}.pstore")
-    store.transaction do
-      csv.each do |row|
-        store[SecureRandom.uuid] = csv_converters.inject(row.to_h) { |hash, converter| converter.convert(hash) }
-      end
-    end
-    
-    store
-  end
-  
   def initialize(store_name)
     super(store_name)
     @queries = []
@@ -29,7 +18,7 @@ class QueryablePStore < PStore
 
   def records
     transaction do
-      roots.map { |root| fetch(root) } 
+      roots.map { |root| fetch(root) }
     end
   end
 
@@ -37,7 +26,7 @@ class QueryablePStore < PStore
     attribute = method.to_s.split("_")[0..-2].join("_").to_sym
     modifier = method.to_s.split("_")[-1].to_sym
 
-    query = Query.new(attribute, modifier, argument || blk) 
+    query = Query.new(attribute, modifier, argument || blk)
     @queries << query if query.valid?(records)
     self
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: queryable_pstore
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Jacob Smith
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-02-25 00:00:00.000000000 Z
+date: 2017-03-04 00:00:00.000000000 Z
 dependencies: []
 description: 'This provides a very simple way of querying PStores. It allows you to
   write queries like: `pstore.age_gt(40).height_lt(1.8).results` Additional documentation
@@ -19,6 +19,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/csv_converter.rb
+- lib/csv_importer.rb
 - lib/query.rb
 - lib/queryable_pstore.rb
 homepage: https://github.com/jacobsmith/queryable_pstore