csv_class_maker 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c337d2fc536792c333afc49688af5945e8e7892d
-  data.tar.gz: dc35299f89d9048ce49ffb867cedcb139e7fbd49
+  metadata.gz: 1b8b85b814a7b26331ffd7677f375a0eb8f9c5bb
+  data.tar.gz: 90ee267cefb5291d54b77c6753f2423091126aa3
 SHA512:
-  metadata.gz: 80c3f72db74a092de09ffcec7c361304f4a5f5ee13a18440b6f18a43109d95b7c1ad50cde9937f3c378d4381de6b114e963b8b7203a8540cfc47838a82efa8ae
-  data.tar.gz: fb6bb09b68875cf5cbc2c0bc763254eee695a6f905c6f129e82d4ccdee896eb5789326aa884cb694879bc52bec073bb054da2317e4f631c8cf213e6c5e8fafa5
+  metadata.gz: a5b71f57eb29c3a6eb1cbef8b5308e717f73f07766d491f4facaa65603d1111c5d31c36e48716250f88605d7747c4acb58fb34b4742ae7252bf00a65719204dc
+  data.tar.gz: ae2ec65f7486a935e00102b6437044c9e972f59b69021907282d549082aa03b82cf03f496949ad419c6a9829529c51ca57869e5617260261f207b7e1fe989307

data/lib/csv_class_maker/csv_class_maker.rb

@@ -21,24 +21,26 @@ module CsvClassMaker
       @@file = CSV.new(File.open(file_name,'r'), headers: true, header_converters: :symbol, return_headers: false)
       @@first_line = 2
       @@last_line = `wc -l #{file_name}`.split(' ').first.to_i
+      @@middle_line = (@@last_line/2)+1
       @line_number = nil
 
       attr_accessor :line_number
 
       def self.file; return @@file; end
       def self.first_line; return @@first_line; end
+      def self.middle_line; return @@middle_line; end
       def self.last_line; return @@last_line; end
 
       # End of class definition.
 
     }
   end
-  
+
   private
 
   def self.extract_headers(file_name)
     @csv_headers = []
-    CSV.new(File.open(file_name,'r'), headers: true, header_converters: :symbol, return_headers: true).first.each do |headers, values| 
+    CSV.new(File.open(file_name,'r'), headers: true, header_converters: :symbol, return_headers: true).first.each do |headers, values|
       @csv_headers << headers
     end
     @csv_headers

data/lib/csv_class_maker/csv_find.rb

@@ -23,14 +23,13 @@ module CsvClassMaker::CsvFind
 
   def find(line_number)
     if (first_line..last_line).include? line_number
-      row = CSV.new(`head -n 1 #{file.path} && head -n #{line_number} #{file.path} | tail -n 1`, headers: true, header_converters: :symbol, return_headers: false).first
-      build_instance row, line_number
-    elsif ((last_line/2+1)..last_line).include? line_number
-      row = CSV.new(`head -n 1 #{file.path} && tail -n #{last_line - line_number} #{file.path} | head -n 1`, headers: true, header_converters: :symbol, return_headers: false).first
-      build_instance row, line_number
+      row = front_find(line_number, file.path)
+    elsif (middle_line..last_line).include? line_number
+      row = back_find(line_number, file.path)
     else
-      nil
+      row = nil
     end
+    row.nil? ? nil : build_instance(row, line_number)
   end
 
   def first
@@ -93,4 +92,11 @@ module CsvClassMaker::CsvFind
     @accumulator
   end
 
+  def front_find(line_number, file_path)
+    CSV.new(`head -n 1 #{file_path} && head -n #{line_number} #{file_path} | tail -n 1`, headers: true, header_converters: :symbol, return_headers: false).first
+  end
+  def back_find(line_number, file_path)
+    CSV.new(`head -n 1 #{file_path} && tail -n #{last_line - line_number} #{file_path} | head -n 1`, headers: true, header_converters: :symbol, return_headers: false).first
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: csv_class_maker
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Mark Platt