csv_party 0.0.1.pre3 → 0.0.1.pre7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1e5823d9eec07faf638ab898d7082be890c73373
-  data.tar.gz: f3b115f8fd4055e141629fc3fc92ac39278dbb2f
+  metadata.gz: 4a03ef93214f6d5974e16e881c4ccab9b2543cfe
+  data.tar.gz: 3179789242dc149e98129bf06e6a18dafb1b25cf
 SHA512:
-  metadata.gz: 3c89169637d692dec361b79b87e5e392050224e43869f0d7a6f66decb3ca1edaed8e7dab777a288ac17c5557ef2964bc5f51a80a841f73601158e22021bf44e5
-  data.tar.gz: 4b677735ca562a25d30fd1a46b999d2803b4f73cf187db340f33f373c7acda071f96ffe1614dc6c51dbde7f2ceddc40c88a4738afb44f97ba40b3703f31accce
+  metadata.gz: 8fd9938a9dfe1fe7cf0fd2f52cb94c2bf210c4f3fab1e7789e14e0ad5a10b999895489193ba32d6df97327c475c6ce37b3f116f30990f91c8d0d27e3b271cd3e
+  data.tar.gz: 58b12e98902a2e676dfb32475922f6a36eb1b4535900b782d601d248cab785bbd5c8a4bd5fbd63eed527649870a4b5947b4dcbf06ac5fb2f96d2fe0fb3b1a919

data/lib/csv_party.rb

@@ -3,65 +3,59 @@ require 'bigdecimal'
 require 'ostruct'
 
 class CSVParty
+  attr_accessor :columns, :row_importer, :importer, :error_processor
+
+  attr_reader :imported_rows, :skipped_rows, :aborted_rows,
+              :abort_message
+
   def initialize(csv_path, options = {})
+    initialize_import_settings
+    initialize_counters_and_statuses
+
     options[:headers] = true
+    dependencies = options.delete(:dependencies)
     @headers = CSV.new(File.open(csv_path)).shift
     @csv = CSV.new(File.open(csv_path), options)
 
+    setup_dependencies(dependencies)
     raise_unless_named_parsers_are_valid
     raise_unless_csv_has_all_headers
   end
 
   def import!
+    if importer
+      instance_exec(&importer)
+    else
+      import_rows!
+    end
+  rescue AbortedImportError => error
+    @aborted = true
+    @abort_message = error.message
+  end
+
+  def import_rows!
     loop do
       begin
         row = @csv.shift
         break unless row
-        import_row(row)
+        import_row!(row)
+        imported_rows << @csv.lineno
+      rescue SkippedRowError
+        skipped_rows << @csv.lineno
+        next
+      rescue AbortedImportError => error
+        raise AbortedImportError, error.message
       rescue StandardError => error
         process_error(error, @csv.lineno + 1)
+        aborted_rows << @csv.lineno
         next
       end
     end
   end
 
-  def parse_row(row)
-    unparsed_row = OpenStruct.new
-    parsed_row = OpenStruct.new
-
-    columns.each do |name, options|
-      header = options[:header]
-      unparsed_value = row[header]
-      parser = options[:parser]
-
-      unparsed_row[name] = unparsed_value
-      parsed_row[name] = if options[:blanks_as_nil] && is_blank?(unparsed_value)
-                           nil
-                         elsif parser.is_a? Symbol
-                           send(parser, unparsed_value)
-                         else
-                           instance_exec(unparsed_value, &parser)
-                         end
-    end
-
-    parsed_row['unparsed'] = unparsed_row
-    parsed_row['csv_string'] = row.to_csv
-
-    return parsed_row
-  end
-
-  def import_row(row)
-    parsed_row = parse_row(row)
-    instance_exec(parsed_row, &importer)
-  end
-
-  def process_error(error, line_number)
-    instance_exec(error, line_number, &error_handler)
-  end
-
-  def self.column(name, options, &block)
-    raise_if_duplicate_column(name)
-    raise_if_missing_header(name, options)
+  def self.column(column, options, &block)
+    raise_if_duplicate_column(column)
+    raise_if_missing_header(column, options)
 
     options = {
       blanks_as_nil: (options[:as] == :raw ? false : true),
@@ -74,60 +68,116 @@ class CSVParty
                "#{options[:as]}_parser".to_sym
              end
 
-    columns[name] = {
+    columns[column] = {
       header: options[:header],
       parser: parser,
       blanks_as_nil: options[:blanks_as_nil]
     }
   end
 
-  def self.columns
-    @columns ||= {}
-  end
-
-  def columns
-    self.class.columns
+  def self.rows(&block)
+    @row_importer = block
   end
 
   def self.import(&block)
     @importer = block
   end
 
-  def self.importer
-    @importer
+  def self.errors(&block)
+    @error_processor = block
   end
 
-  def importer
-    self.class.importer
+  def self.columns
+    @columns ||= {}
   end
 
-  def self.error(&block)
-    @error = block
+  def self.row_importer
+    @row_importer ||= nil
   end
 
-  def self.error_handler
-    @error
+  def self.importer
+    @importer ||= nil
   end
 
-  def error_handler
-    self.class.error_handler
+  def self.error_processor
+    @error_processor ||= nil
+  end
+
+  def aborted?
+    @aborted
   end
 
   def self.raise_if_duplicate_column(name)
     return unless columns.has_key?(name)
 
     raise DuplicateColumnError, "A column named :#{name} has already been \
-            defined, choose a different name"
+              defined, choose a different name"
   end
+  private_class_method :raise_if_duplicate_column
 
   def self.raise_if_missing_header(name, options)
     return if options.has_key?(:header)
 
     raise MissingHeaderError, "A header must be specified for #{name}"
   end
+  private_class_method :raise_if_missing_header
 
   private
 
+  def import_row!(row)
+    parsed_row = parse_row(row)
+    instance_exec(parsed_row, &row_importer)
+  end
+
+  def parse_row(row)
+    unparsed_row = OpenStruct.new
+    columns.each do |column, options|
+      header = options[:header]
+      unparsed_row[column] = row[header]
+    end
+
+    parsed_row = OpenStruct.new
+    columns.each do |column, options|
+      value = row[options[:header]]
+      parsed_row[column] = parse_column(
+        value,
+        options[:parser],
+        options[:blanks_as_nil]
+      )
+    end
+
+    parsed_row[:unparsed] = unparsed_row
+    parsed_row[:csv_string] = row.to_csv
+
+    return parsed_row
+  end
+
+  def parse_column(value, parser, blanks_as_nil)
+    if blanks_as_nil && is_blank?(value)
+      nil
+    elsif parser.is_a? Symbol
+      send(parser, value)
+    else
+      instance_exec(value, &parser)
+    end
+  end
+
+  def process_error(error, line_number)
+    instance_exec(error, line_number, &error_processor)
+  end
+
+  def skip_row
+    raise SkippedRowError
+  end
+
+  def abort_row(message)
+    raise AbortedRowError, message
+  end
+
+  def abort_import(message)
+    raise AbortedImportError, message
+  end
+
   def is_blank?(value)
     value.nil? || value.strip.empty?
   end
@@ -190,7 +240,30 @@ class CSVParty
     columns = missing_columns.join("', '")
     raise MissingColumnError,
           "CSV file is missing column(s) with header(s) '#{columns}'. \
-          File has these headers: #{@headers.join(', ')}."
+              File has these headers: #{@headers.join(', ')}."
+  end
+
+  def setup_dependencies(dependencies)
+    return unless dependencies
+
+    dependencies.each do |dependency, value|
+      self.class.class_eval { attr_accessor dependency }
+      send("#{dependency}=", value)
+    end
+  end
+
+  def initialize_import_settings
+    @columns = self.class.columns
+    @row_importer = self.class.row_importer
+    @importer = self.class.importer
+    @error_processor = self.class.error_processor
+  end
+
+  def initialize_counters_and_statuses
+    @imported_rows = []
+    @skipped_rows = []
+    @aborted_rows = []
+    @aborted = false
   end
 end
 
@@ -205,3 +278,12 @@ end
 
 class MissingColumnError < ArgumentError
 end
+
+class SkippedRowError < RuntimeError
+end
+
+class AbortedRowError < RuntimeError
+end
+
+class AbortedImportError < RuntimeError
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: csv_party
 version: !ruby/object:Gem::Version
-  version: 0.0.1.pre3
+  version: 0.0.1.pre7
 platform: ruby
 authors:
 - Rico Jones
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-04-27 00:00:00.000000000 Z
+date: 2017-05-01 00:00:00.000000000 Z
 dependencies: []
 description: A gem for making CSV imports a little more fun.
 email: rico@toasterlovin.com
@@ -29,7 +29,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '2.0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">"