csv_lazy 0.0.3 → 0.0.4

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.0.4

data/csv_lazy.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "csv_lazy"
-  s.version = "0.0.3"
+  s.version = "0.0.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Kasper Johansen"]

data/lib/csv_lazy.rb

@@ -12,7 +12,8 @@ class Csv_lazy
     @args = {
       :quote_char => '"',
       :row_sep => "\n",
-      :col_sep => ";"
+      :col_sep => ";",
+      :headers => false
     }.merge(args)
     
     @io = @args[:io]
@@ -23,7 +24,7 @@ class Csv_lazy
     @mutex = Mutex.new
     #@debug = true
     
-    accepted = [:encode, :quote_char, :row_sep, :col_sep, :io, :debug]
+    accepted = [:encode, :quote_char, :row_sep, :col_sep, :io, :debug, :headers]
     @args.each do |key, val|
       if accepted.index(key) == nil
         raise "Unknown argument: '#{key}'."
@@ -45,6 +46,15 @@ class Csv_lazy
     @regex_read_until_row_sep = /\A(.+?)#{Regexp.escape(@args[:row_sep])}/
     @regex_read_until_end = /\A(.+?)\Z/
     
+    if @args[:headers]
+      headers = []
+      read_row.each do |key|
+        headers << key.to_sym
+      end
+      
+      @headers = headers
+    end
+    
     self.each(&blk) if blk
   end
   
@@ -86,7 +96,16 @@ class Csv_lazy
     if row.empty?
       return false
     else
-      return row
+      if @headers
+        ret = {}
+        row.length.times do |count|
+          ret[@headers[count]] = row[count]
+        end
+        
+        return ret
+      else
+        return row
+      end
     end
   end
   

data/spec/csv_lazy_spec.rb

@@ -83,4 +83,30 @@ describe "CsvLazy" do
     
     lines_found.should eql(1)
   end
+  
+  it "should be able to use headers and return hashes instead" do
+    cont = "\"name\",age\r\n"
+    cont += "\"Kasper Johansen\",27\r\n"
+    cont += "\"Christina Stoeckel\",\"25\"\r\n"
+    
+    line = 0
+    Csv_lazy.new(:col_sep => ",", :io => StringIO.new(cont), :headers => true, :row_sep => "\r\n") do |csv|
+      csv.class.should eql(Hash)
+      line += 1
+      csv.keys.length.should eql(2)
+      csv.length.should eql(2)
+      
+      if line == 1
+        csv[:name].should eql("Kasper Johansen")
+        csv[:age].should eql("27")
+      elsif line == 2
+        csv[:name].should eql("Christina Stoeckel")
+        csv[:age].should eql("25")
+      else
+        raise "Wrong line: #{line}"
+      end
+    end
+    
+    line.should eql(2)
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: csv_lazy
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -111,7 +111,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -779445160721638785
+      hash: 222799097840594694
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: