load_data_infile 0.3.0 → 0.4.0

data/VERSION

@@ -1 +1 @@
-0.3.0
+0.4.0

data/lib/load_data_infile.rb

@@ -3,6 +3,7 @@ require 'ostruct'
 
 module LoadDataInfile
   module MySql
+    mattr_accessor :load_data_infile_defaults
 
     # Deletes all rows in table very fast, but without calling +destroy+ method
     # nor any hooks.
@@ -29,6 +30,14 @@ module LoadDataInfile
     end
 
     # Load csv from a file using MySql's LOAD DATA INFILE
+    # You can set defaults for all these options using the accesor load_data_infile_defaults:
+    #
+    # class ActiveRecord::Base
+    #   load_data_infile_defaults = {
+    #     :ignore => 1
+    #   }
+    # end
+    #
     # For details see: http://dev.mysql.com/doc/refman/5.1/en/load-data.html
     #
     # Options:
@@ -51,7 +60,9 @@ module LoadDataInfile
     # table                  :: [OPTIONAL] Table name. Defaults to quoted_table_name (won't work if used from an abstract class, e.g. ActiveRecord::Base')
     # terminated_by          :: [OPTIONAL] Character
     # disable_keys           :: [OPTIONAL] true or false. Defaults to true. Disables foreign keys while running the import.
-    def load_data_infile(options = {})
+    def load_data_infile(opt = {})
+      options = (load_data_infile_defaults || Hash.new).merge(opt)
+
       disable_keys_option = !options.member?(:disable_keys) || options[:disable_keys]
 
       c = Context.new

data/spec/load_data_infile_spec.rb

@@ -52,4 +52,25 @@ describe LoadDataInfile do
       "field_c" => 24000
     }]
   end
+
+  it "can use defaults" do
+    ActiveRecord::Base.load_data_infile_defaults = {
+      :terminated_by => ",",
+      :columns       => %w|id @field_a @field_b @field_c|,
+      :mappings      => {
+                          :field_a => "CONCAT('So ', @field_a)",
+                          :field_b => "CONCAT('Much ', @field_b)",
+                          :field_c => "@field_c * 10",
+                        }
+    }
+
+    Thing.load_data_infile(:path => FIXTURE_WITHOUT_HEADERS)
+
+    Thing.all.map(&:attributes).should == [{
+      "id"      => 61,
+      "field_a" => "So live",
+      "field_b" => "Much from",
+      "field_c" => 24000
+    }]
+  end
 end

data/spec/spec_helper.rb

@@ -1,11 +1,11 @@
 SPEC_PATH = File.dirname(__FILE__)
 $LOAD_PATH.unshift(SPEC_PATH)
 $LOAD_PATH.unshift(File.join(SPEC_PATH, '..', 'lib'))
-require 'load_data_infile'
 require 'spec'
 require 'spec/autorun'
 require 'rubygems'
 require 'active_record'
+require 'load_data_infile'
 require 'active_record_helper'
 require File.join(SPEC_PATH, "..", "rails", "init.rb")
 

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 3
+  - 4
   - 0
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Emmanuel Oga
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-24 00:00:00 -03:00
+date: 2010-02-25 00:00:00 -03:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency