load_data_infile 0.1.0 → 0.2.0

data/VERSION

@@ -1 +1 @@
-0.1.0
+0.2.0

data/lib/load_data_infile.rb

@@ -49,7 +49,10 @@ module LoadDataInfile
     # optionally_enclosed_by :: [OPTIONAL] Character
     # 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 = {})
+      disable_keys_option = !options.member?(:disable_keys) || options[:disable_keys]
+
       c = Context.new
 
       if options[:low_priority]
@@ -91,7 +94,11 @@ module LoadDataInfile
         c.mappings = "SET #{s}"
       end
 
+      disable_keys(c.table_name) if disable_keys_option
+
       connection.execute(ERB.new(LOAD_DATA_INFILE_SQL).result(c.binding).gsub(/^\s*\n/, ""))
+    ensure
+      enable_keys(c.table_name) if disable_keys_option
     end
 
     class Context < OpenStruct

data/spec/load_data_infile_spec.rb

@@ -6,14 +6,12 @@ describe LoadDataInfile do
   end
 
   it "loads data from a csv file with headers into an ActiveRecord table" do
-    Thing.with_keys_disabled do
-      Thing.load_data_infile(
-        :path          => FIXTURE_WITH_HEADERS,
-        :columns       => %w|id field_a field_b field_c|,
-        :terminated_by => ",",
-        :ignore        => 1
-      )
-    end
+    Thing.load_data_infile(
+      :path          => FIXTURE_WITH_HEADERS,
+      :columns       => %w|id field_a field_b field_c|,
+      :terminated_by => ",",
+      :ignore        => 1
+    )
     Thing.all.map(&:attributes).should == [{
       "id"      => 71,
       "field_a" => "Hello",
@@ -23,13 +21,11 @@ describe LoadDataInfile do
   end
 
   it "loads data from a csv file without headers into an ActiveRecord table" do
-    Thing.with_keys_disabled do
-      Thing.load_data_infile(
-        :path          => FIXTURE_WITHOUT_HEADERS,
-        :terminated_by => ",",
-        :columns       => %w|id field_a field_b field_c|
-      )
-    end
+    Thing.load_data_infile(
+      :path          => FIXTURE_WITHOUT_HEADERS,
+      :terminated_by => ",",
+      :columns       => %w|id field_a field_b field_c|
+    )
     Thing.all.map(&:attributes).should == [{
       "id"      => 61,
       "field_a" => "live",
@@ -39,18 +35,16 @@ describe LoadDataInfile do
   end
 
   it "loads data from a csv file with mapping" do
-    Thing.with_keys_disabled do
-      Thing.load_data_infile(
-        :path          => FIXTURE_WITHOUT_HEADERS,
-        :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",
-                          }
-      )
-    end
+    Thing.load_data_infile(
+      :path          => FIXTURE_WITHOUT_HEADERS,
+      :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.all.map(&:attributes).should == [{
       "id"      => 61,
       "field_a" => "So live",

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Emmanuel Oga