csv_party 0.0.1.pre

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 475350af4d86e4c6586ea1752a731a888ddb336d
+  data.tar.gz: 679da67bcacd832eb394f0bd93f483babcf44ecc
+SHA512:
+  metadata.gz: 9c91fdd4277ba7e80b9561333879386c115dbaa58df621705cdbfae8d517ab7a74b2298efdc8f45e32be014f2718d1c2fa8fcaba3adbb17db13b0e38ffda934b
+  data.tar.gz: 58597567ec2ab2e1950bbc32a7526fe57ebe632437074661dcfc7c3390a90f570ce90ce203acfd622f84dea4aedea5e9d429206953bd21eb59621e986cfd596e

data/lib/csv_party.rb

@@ -0,0 +1,149 @@
+require 'csv'
+require 'bigdecimal'
+require 'ostruct'
+
+class CSVParty
+  def initialize(csv_path)
+    @headers = CSV.new(File.open(csv_path)).shift
+    @csv = CSV.new(File.open(csv_path), headers: true)
+    raise_unless_named_parsers_are_valid
+    raise_unless_csv_has_all_headers
+  end
+
+  def import!
+    @csv.each do |row|
+      parsed_row = parse_row(row)
+      import_row(parsed_row)
+    end
+  end
+
+  def parse_row(row)
+    parsed_row = OpenStruct.new
+    parsed_row[:values] = OpenStruct.new
+
+    columns.each do |name, options|
+      header = options[:header]
+      parser = options[:parser]
+      parsed_row[name] = instance_exec(row[header], &parser)
+      parsed_row[:values][name] = row[header]
+    end
+
+    return parsed_row
+  end
+
+  def import_row(parsed_row)
+    importer.call(parsed_row)
+  end
+
+  def self.column(name, options, &block)
+    if columns.has_key?(name)
+      raise DuplicateColumnError, "A column named :#{name} has already been defined, choose a different name"
+    end
+    unless options.has_key?(:header)
+      raise MissingHeaderError, "A header must be specified for #{name}"
+    end
+
+    if block_given?
+      columns[name] = { header: options[:header], parser: block }
+    else
+      if options.has_key?(:as)
+        parser_method = "#{options[:as]}_parser".to_sym
+      else
+        parser_method = :string_parser
+      end
+      columns[name] = {
+        header: options[:header],
+        parser: Proc.new { |value| send(parser_method, value) },
+        parser_method: parser_method
+      }
+    end
+  end
+
+  def self.import(&block)
+    @importer = block
+  end
+
+  def self.columns
+    @columns ||= {}
+  end
+
+  def columns
+    self.class.columns
+  end
+
+  def self.importer
+    @importer
+  end
+
+  def importer
+    self.class.importer
+  end
+
+  private
+
+
+  def raw_parser(value)
+    value
+  end
+
+  def string_parser(value)
+    value.to_s.strip
+  end
+
+  def boolean_parser(value)
+    ['1', 't', 'true'].include? value.to_s.strip.downcase
+  end
+
+  def integer_parser(value)
+    value.to_i
+  end
+
+  def decimal_parser(value)
+    cleaned_value = value.to_s.strip.gsub(/[^0-9.]/, "")
+    BigDecimal.new(cleaned_value)
+  end
+
+  def named_parsers
+    (private_methods + methods).grep(/_parser$/)
+  end
+
+  def columns_with_named_parsers
+    columns.select { |name, options| options.has_key?(:parser_method) }
+  end
+
+  # This error has to be raised at runtime because, when the class body
+  # is being executed, the parser methods won't be available unless
+  # they are defined above the column definitions in the class body
+  def raise_unless_named_parsers_are_valid
+    columns_with_named_parsers.each do |name, options|
+      parser = options[:parser_method]
+      unless named_parsers.include? parser
+        raise UnknownParserError,
+          "You're trying to use the :#{parser.to_s.gsub('_parser', '')} parser for the :#{name} column, but it doesn't exist. Available parsers are: :#{named_parsers.map { |p| p.to_s.gsub('_parser', '') }.join(', :')}."
+      end
+    end
+  end
+
+  def defined_headers
+    columns.map { |name, options| options[:header] }
+  end
+
+  def raise_unless_csv_has_all_headers
+    missing_columns = defined_headers - @headers
+    unless missing_columns.empty?
+      raise MissingColumnError, "CSV file is missing column(s) with header(s) '#{missing_columns.join("', '")}'."
+    end
+  end
+end
+
+class UnknownParserError < ArgumentError
+end
+
+class MissingHeaderError < ArgumentError
+end
+
+class DuplicateColumnError < ArgumentError
+end
+
+class MissingColumnError < ArgumentError
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: csv_party
+version: !ruby/object:Gem::Version
+  version: 0.0.1.pre
+platform: ruby
+authors:
+- Rico Jones
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-03-30 00:00:00.000000000 Z
+dependencies: []
+description: A gem for making CSV imports a little more fun.
+email: rico@toasterlovin.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/csv_party.rb
+homepage: http://rubygems.org/gems/csv_party
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">"
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: CSV Party
+test_files: []