csvmapper 0.0.3 → 0.0.4

data/Changes

@@ -1,3 +1,11 @@
+0.04
+  - Added #columns method
+      columns :first_name, :last_name, :age
+  - Added #grouped_by method
+    grouped by field key
+  - Added #where method
+    arel style finder (testing)
+
 0.03
   - Added CSVMapper#each method
   - Added CSVMapper#before_filter method

data/examples/columns.rb

@@ -0,0 +1,16 @@
+# encoding: utf-8
+require 'csvmapper'
+require 'cgi'
+
+class OmgCSV < CSVMapper
+  columns :first_name, :last_name, :age
+end
+
+csv = OmgCSV.load <<EOT
+takumi,kimoto,99
+super,urutoraman,11
+EOT
+
+p csv.map(&:age)
+# => ["99", "11"]
+

data/examples/no_dsl.rb

@@ -0,0 +1,24 @@
+# encoding: utf-8
+require 'csvmapper'
+require 'cgi'
+
+class MyCSV < CSVMapper
+end
+
+MyCSV.before_filter do |v|
+  CGI.unescapeHTML(v) if v
+end
+
+MyCSV.delimiter '<>'
+MyCSV.column :name, 0, :string
+MyCSV.column :hash, 1
+MyCSV.column :host, 2
+MyCSV.column :contact, 3, :uri
+MyCSV.column :genre, 4
+MyCSV.column :description, 5
+MyCSV.column :active_viewers, 6, :numeric
+MyCSV.column :total_viewers, 7, :numeric
+
+csv = MyCSV.load_file("http://temp.orz.hm/yp/index.txt")
+p csv
+

data/examples/yp.rb

@@ -34,3 +34,5 @@ p csv.sort_by{ |e|
   -e.active_viewers
 }
 
+puts csv.to_s
+

data/examples/yp_find.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+require 'csvmapper'
+require 'cgi'
+require 'hashie'
+
+class YPIndex < CSVMapper
+  before_filter do |v|
+    CGI.unescapeHTML(v) if v
+  end
+
+  delimiter '<>'
+  column :name, 0, :string
+  column :hash, 1
+  column :host, 2
+  column :contact, 3, :uri
+  column :genre, 4
+  column :description, 5
+  column :active_viewers, 6, :numeric
+  column :total_viewers, 7, :numeric
+  column :bitrate, 8, :numeric
+  column :filetype, 9
+  column :artist, 10
+  column :title, 11
+  column :album, 12
+  column :reserved, 13
+  column :hash2, 14
+  column :time, 15
+  column :type, 16
+  column :reservee, 17
+  column :nullfield, 18
+end
+
+p YPIndex.load_file("http://temp.orz.hm/yp/index.txt").where(:name => 'あくえり').where(:genre => 'DQ3 SFC').first
+
+

data/lib/csvmapper.rb

@@ -9,7 +9,7 @@ require 'uri'
 require 'open-uri'
 
 class CSVMapper
-  @@data = []
+  @@schema = []
   @@ignore_header_line = false
   @@on_error_go_to_next_line = false
   @@has_header = nil
@@ -18,13 +18,18 @@ class CSVMapper
 
   def self.column(field, *args, &block)
     value = args.shift
-    #value = block.call(value) if block_given?
     record = {
       field => value,
       :options => args,
       :filter => block
     }
-    @@data << record
+    @@schema << record
+  end
+
+  def self.columns(*fields)
+    fields.each_with_index{ |field, index|
+      self.column(field, index, :string)
+    }
   end
 
   def self.has_header(at=0)
@@ -47,6 +52,10 @@ class CSVMapper
 
     records = []
     buf.each_with_index{ |line, index|
+      if line.empty?
+        next
+      end
+
       if @@has_header == index
         line.each_with_index{ |name, i|
           self.column(name.gsub(" ", "_").downcase, i, :string)
@@ -62,7 +71,7 @@ class CSVMapper
       }
 
       hash = {}
-      @@data.each{ |opt|
+      @@schema.each{ |opt|
         key = opt.keys.first
         options = opt[:options]
 
@@ -155,17 +164,56 @@ class CSVMapper
     @records[index]
   end
 
+  ## size
+  def rows
+    @records.size
+  end
+
+  def cols
+    if @records.nil? or @records.empty?
+      return 0
+    else
+      return @records.first.size
+    end
+  end
+
   def [](index)
     row(index)
   end
 
   include Enumerable
   def each(&block)
-    @records.each{ |record|
-      block.call(record)
+    matched = []
+    if @where
+      @records.each{ |record|
+        @where.each{ |key, value|
+          if record.send(key) == value
+            matched << record
+          end
+        }
+      }
+    else
+      matched = @records
+    end
+
+    matched.each{ |e|
+      block.call(e)
     }
   end
 
+  ## arel style finder
+  def where(options={})
+    @where = Hashie::Clash.new(options) if @where.nil?
+    @where.merge! options
+    self
+  end
+
+  def order(options={})
+    @order = Hashie::Clash.new(options) if @order.nil?
+    @order.merge! options
+    self
+  end
+
   def to_hash
     @records.map{ |record|
       record.to_hash
@@ -175,5 +223,15 @@ class CSVMapper
   def to_json
     @records.to_json
   end
+
+  def grouped_by(field)
+    hash = {}
+    each{|e|
+      k = e[field]
+      hash[k] ||= []
+      hash[k] << e
+    }
+    hash
+  end
 end
 

data/lib/csvmapper/version.rb

@@ -1,3 +1,3 @@
 class CSVMapper
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: csvmapper
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-01 00:00:00.000000000 Z
+date: 2012-08-03 00:00:00.000000000 Z
 dependencies: []
 description: CSV to Ruby class mapper
 email:
@@ -27,7 +27,10 @@ files:
 - csvmapper.gemspec
 - examples/basic.rb
 - examples/basic2.rb
+- examples/columns.rb
+- examples/no_dsl.rb
 - examples/yp.rb
+- examples/yp_find.rb
 - lib/csvmapper.rb
 - lib/csvmapper/version.rb
 homepage: ''