rcsv_loader 0.0.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f581428c6e050728dbc4aa92a115ecd9c1441621
-  data.tar.gz: cd70140fdf7f62279bb6b28e01dd36ab0c9411f3
+  metadata.gz: f16383a5089660aaa8623b243f05a9cdc72dc363
+  data.tar.gz: 592ee577893e3f81f4ba869bf1c227f1e7cba056
 SHA512:
-  metadata.gz: 8e0da230d509d4dca60d489d23551c8fda507d672ef33f372cfba747fe2b2ea904f1e1c531d67e1db72461bbc89452293cb0346a033f6a4da8702b5de25fdcba
-  data.tar.gz: 904038d9043c618727974778f5b93ea265abb6d8ac9f042ccec4a8867e60dabe5b0531a3403aa47ff8e1997292dfb2d8445e2cf5f768eb575aacca762c980de6
+  metadata.gz: ac77ebaaa41fe9c176b736953842de5c66a18752d000e3a312d5711b1d4094aa16d0432109b1de659bc24a064c3ae1cfd87736585da9281796885cb4c76385b4
+  data.tar.gz: af321f0a281ca5dd9d30a6443e1d349df33cf6f5e29f1f6678858be06ca37f6d4a34610eb07ab05451da89d179741f3b54cf56bab5929640497210a6026aab95

data/.travis.yml

@@ -5,4 +5,6 @@ rvm:
   - 2.0.0
   - 2.1.0
   - 2.2.0
+before_script:
+  - rake install
 script: bundle exec rspec spec

data/lib/rcsv_loader/base.rb

@@ -0,0 +1,11 @@
+require 'csv'
+require "rcsv_loader/version"
+require 'rcsv_loader/core'
+
+module RCsvLoader
+
+  class Base
+    include Core
+  end
+
+end

data/lib/rcsv_loader/core.rb

@@ -0,0 +1,137 @@
+require 'rcsv_loader/where'
+
+module RCsvLoader
+  module Core
+    extend Forwardable
+
+    include Where
+
+    def self.included base
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      #
+      # Laod a csv file
+      #
+      #   options: The options for 'csv' which is Ruby's built-in library.
+      #
+      def load_csv file, options = {}
+
+        rows = CSV.readlines file, options
+
+        rows = yield rows if block_given?
+
+        self.new rows.map { |row| self::Row.new row }
+      end
+
+      attr_reader :headers
+
+      #
+      # Define accessor names for csv column.
+      #
+      def column column_name, alias_name = nil
+        alias_name = column_name.to_sym unless alias_name
+        @headers = all_headers
+        @headers.merge!(alias_name => column_name)
+        define_row(@headers)
+
+        @has_headers = true
+      end
+
+      #
+      # Define headers for no headers csv.
+      #
+      def insert_headers headers = []
+        @headers = Hash[headers.map.with_index { |e, i| [e, i] }]
+        define_row(@headers)
+
+        @has_headers = false
+      end
+
+      def headers?
+        !!@has_headers
+      end
+
+      private
+      #
+      # Get headers that include super class's one.
+      #
+      def all_headers current_class = nil
+        current_class ||= self
+        return {} if current_class == Object
+        current_header = current_class.headers || {}
+        return current_header.merge all_headers(current_class.superclass)
+      end
+
+      #
+      # Define a class which represent a row of csv.
+      #
+      def define_row headers
+        headers ||= {}
+
+        # get the superclass for Row
+        baseclass = if superclass.const_defined? :Row
+                      superclass::Row
+                    else
+                      Object
+                    end
+
+        # remove class definition
+        if const_defined? :Row and baseclass != self::Row
+          self.send :remove_const, :Row
+        end
+
+        # define Row class
+        const_set :Row, (Class.new(baseclass) do
+
+          attr_accessor *headers.keys
+
+          define_method :initialize do |line = {}|
+            headers.each do |k, v|
+              instance_variable_set "@#{k.to_s}", line[v]
+            end
+          end
+
+          define_method :to_csv do
+            CSV.generate_line headers.map { |k, v| self.send k }
+          end
+
+        end)
+      end
+    end
+
+    def_delegators :@rows, :+, :<<
+
+    def initialize rows = []
+      @rows = rows
+    end
+
+    def all
+      @rows
+    end
+
+    #
+    # Convert to csv string
+    #
+    # options: {
+    #   headers: boolean
+    # }
+    #
+    def to_csv options = {}
+      csv = ""
+      csv += header_line if options[:headers].nil? or options[:headers]
+      csv += @rows.map(&:to_csv).join
+    end
+
+    private
+
+    #
+    # generate header line
+    #
+    def header_line
+      CSV.generate_line(self.class.headers.map { |k, v| self.class.headers? ? v.to_s : k.to_s })
+    end
+
+  end
+end

data/lib/rcsv_loader/version.rb

@@ -1,3 +1,3 @@
 module RCsvLoader
-  VERSION = "0.0.1"
+  VERSION = "0.2.0"
 end

data/lib/rcsv_loader/where.rb

@@ -0,0 +1,50 @@
+module RCsvLoader
+  module Where
+
+    #
+    # params = {
+    #   :#{accessor_name} => expected_value,
+    #   ...
+    # }
+    #
+    # compare with '=='
+    #
+    # options = {
+    #   :regexp => true
+    # }
+    #
+    # return empty array if the condition is invalid or couldn't find any rows.
+    #
+    def where conditions = {}, options = {}
+      rows = @rows || []
+      return rows if conditions.empty?
+
+      # filter the conditions.
+      c = conditions.select do |k, v|
+            self.class.headers.keys.any? { |key| key.to_s == k.to_s }
+          end
+
+      return [] if c.empty?
+
+      if options[:regexp]
+        where_with_regexp c
+      else
+        rows.select { |row| c.all? { |attr, con| (row.send attr) == con } }
+      end
+
+    end
+
+    private
+
+    #
+    # evaluate expression with #match
+    #
+    def where_with_regexp conditions = {}
+      rows = @rows || []
+      return rows if conditions.empty?
+      rows.select { |row| conditions.all? { |attr, con| (row.send attr).match con } }
+    end
+
+  end
+end
+

data/lib/rcsv_loader.rb

@@ -2,139 +2,5 @@ require "rcsv_loader/version"
 require 'csv'
 
 module RCsvLoader
-
-  attr_reader :headers
-
-  #
-  # Define accessor names for csv column.
-  #
-  def column column_name, alias_name = nil
-    alias_name = column_name.to_sym unless alias_name
-    @headers = all_headers
-    @headers.merge!(alias_name => column_name)
-    define_row(@headers)
-  end
-
-  #
-  # Define headers for no headers csv.
-  #
-  def insert_headers headers = []
-    @headers = Hash[headers.map.with_index { |e, i| [e, i] }]
-    define_row(@headers)
-  end
-
-  class Base
-    extend RCsvLoader
-    extend Forwardable
-
-    #
-    # Laod a csv file
-    #
-    #   options: The options for 'csv' which is Ruby's built-in library.
-    #
-    def self.load_csv file, options = {}
-
-      rows = CSV.readlines file, options
-
-      rows = yield rows if block_given?
-
-      self.new rows.map { |row| self::Row.new row }
-    end
-
-    def_delegators :@rows, :+, :<<
-
-    def initialize rows = []
-      @rows = rows
-    end
-
-    def all
-      @rows
-    end
-
-    #
-    # params = {
-    #   :#{accessor_name} => expected_value,
-    #   ...
-    # }
-    #
-    # compare with '=='
-    #
-    # return empty array if the condition is invalid or couldn't find any rows.
-    #
-    def where params = {}
-      return @rows if params.empty?
-
-      # filter the conditions.
-      params.select! do |k, v|
-        self.class.headers.keys.any? { |key| key.to_s == k.to_s }
-      end
-
-      return [] if params.empty?
-
-      @rows.select do |row|
-        params.all? { |attr, con| (row.send attr) == con }
-      end
-
-    end
-
-    #
-    # Convert to csv string
-    #
-    # options: {
-    #   headers: boolean
-    # }
-    #
-    def to_csv options = {}
-      csv = ""
-      csv += CSV.generate_line(self.class.headers.map { |k, v| v } ) if options[:headers]
-      csv += @rows.map(&:to_csv).join
-    end
-  end
-
-  private
-  #
-  # Get headers that include super class's one.
-  #
-  def all_headers current_class = nil
-    current_class ||= self
-    return {} if current_class == Base
-    current_header = current_class.headers || {}
-    return current_header.merge all_headers(current_class.superclass)
-  end
-
-  #
-  # Define a class which represent a row of csv.
-  #
-  def define_row headers
-    headers ||= {}
-
-    # get the superclass for Row
-    baseclass = if superclass.const_defined? :Row
-                  superclass::Row
-                else
-                  Object
-                end
-
-    # remove class definition
-    if const_defined? :Row and baseclass != self::Row
-      self.send :remove_const, :Row
-    end
-
-    # define Row class
-    const_set :Row, (Class.new(baseclass) do
-
-      attr_accessor *headers.keys
-
-      define_method :initialize do |line = {}|
-        headers.each do |k, v|
-          instance_variable_set "@#{k.to_s}", line[v]
-        end
-      end
-
-      define_method :to_csv do
-        CSV.generate_line headers.map { |k, v| self.send k }
-      end
-
-    end)
-  end
+  autoload :Base, 'rcsv_loader/base'
 end

data/spec/core_spec.rb

@@ -0,0 +1,63 @@
+require 'spec_helper'
+require 'pathname'
+
+require 'models/sample_csv'
+require 'models/sample_csv_no_headers'
+
+describe RCsvLoader do
+
+  describe RCsvLoader::Core do
+
+    shared_examples 'to_csv' do |expected|
+      subject { data.to_csv }
+      it { is_expected.to eq expected }
+    end
+
+    shared_examples 'to_csv without headers' do |expected|
+      subject { data.to_csv headers: false }
+      it { is_expected.to eq expected }
+    end
+
+    context "sample"do
+      let(:path) { File.join Pathname(__FILE__).dirname, "fixtures/sample.csv" }
+      let(:data) { SampleCsv.load_csv path, headers: true }
+
+      expected_with_headers = [
+        'id,file name,extension,url',
+        '1,file_01.zip,zip,https://example.com/file_01.zip',
+        '2,file_02.zip,zip,https://example.com/file_02.zip',
+        ''
+      ].join($/)
+      it_behaves_like 'to_csv', expected_with_headers
+
+      expected_without_headers = [
+        '1,file_01.zip,zip,https://example.com/file_01.zip',
+        '2,file_02.zip,zip,https://example.com/file_02.zip',
+        ''
+      ].join($/)
+      it_behaves_like 'to_csv without headers', expected_without_headers
+    end
+
+    context "sample_without_headers" do
+      let(:path) { File.join Pathname(__FILE__).dirname, "fixtures/no_headers_sample.csv" }
+      let(:data) { SampleCsvNoHeaders.load_csv path, headers: false }
+
+      expected_with_headers = [
+        'id,file_name,extension,url',
+        '1,file_01.zip,zip,https://example.com/file_01.zip',
+        '2,file_02.zip,zip,https://example.com/file_02.zip',
+        ''
+      ].join($/)
+      it_behaves_like 'to_csv', expected_with_headers
+
+      expected_without_headers = [
+        '1,file_01.zip,zip,https://example.com/file_01.zip',
+        '2,file_02.zip,zip,https://example.com/file_02.zip',
+        ''
+      ].join($/)
+      it_behaves_like 'to_csv without headers', expected_without_headers
+    end
+
+  end
+
+end

data/spec/rcsv_loader_spec.rb

@@ -84,6 +84,19 @@ describe RCsvLoader do
           }
           it_behaves_like "where", expected
         end
+
+        context ':regexp => true, :file_name => ".*?\.zip"' do
+          let(:results) { sample.where( { :file_name => ".*?\.zip" }, :regexp => true) }
+          expected = {
+            :data_count => 2,
+            :first_row  => {
+              :file_name => "file_01.zip",
+              :extension => "zip",
+              :url       => "https://example.com/file_01.zip"
+            }
+          }
+          it_behaves_like "where", expected
+        end
       end
     end
 
@@ -130,6 +143,20 @@ describe RCsvLoader do
           }
           it_behaves_like "where", expected
         end
+
+        context ':regexp => true, :file_name => ".*?\.zip"' do
+          let(:results) { sample.where( { :file_name => ".*?\.zip" }, :regexp => true) }
+          expected = {
+            :data_count => 2,
+            :first_row  => {
+              :file_name => "file_01.zip",
+              :extension => "zip",
+              :url       => "https://example.com/file_01.zip"
+            }
+          }
+          it_behaves_like "where", expected
+        end
+
       end
 
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rcsv_loader
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.2.0
 platform: ruby
 authors:
 - osadake212
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-10 00:00:00.000000000 Z
+date: 2015-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -67,8 +67,12 @@ files:
 - README.md
 - Rakefile
 - lib/rcsv_loader.rb
+- lib/rcsv_loader/base.rb
+- lib/rcsv_loader/core.rb
 - lib/rcsv_loader/version.rb
+- lib/rcsv_loader/where.rb
 - rcsv_loader.gemspec
+- spec/core_spec.rb
 - spec/fixtures/no_headers_sample.csv
 - spec/fixtures/sample.csv
 - spec/models/sample_csv.rb
@@ -100,6 +104,7 @@ signing_key:
 specification_version: 4
 summary: A simple csv loader.
 test_files:
+- spec/core_spec.rb
 - spec/fixtures/no_headers_sample.csv
 - spec/fixtures/sample.csv
 - spec/models/sample_csv.rb