Merge pull request #35 from Fivell/csv_replacements

Csv replacements

Author: Fivell

README.md

@@ -178,8 +178,25 @@ Tool                    | Description
         }
     end
 ```    
-    
-#### Example7 dynamic CSV options, template overriding
+
+#### Example7 change csv values before import (find each 'Author name' column and replace it with authors_id before insert )
+
+   ```ruby
+     ActiveAdmin.register Post  do
+             active_admin_import validate: true,
+              headers_rewrites: { :'Author name' => :author_id },
+              before_batch_import: ->(importer) {
+                authors_names = importer.values_at(:author_id)
+                # replacing author name with author id
+                authors   = Author.where(name: authors_names).pluck(:name, :id)
+                options = Hash[*authors.flatten] # #{"Jane" => 2, "John" => 1}
+                importer.batch_replace(:author_id, options)
+              }
+         end
+  ```
+
+
+#### Example8 dynamic CSV options, template overriding
 
  -  put overrided template to ```app/views/import.html.erb```
 

lib/active_admin_import/importer.rb

@@ -49,13 +49,30 @@ def import_options
       @import_options ||= options.slice(:validate, :on_duplicate_key_update, :ignore, :timestamps)
     end
 
+    def batch_replace(header_key, options)
+      index = header_index(header_key)
+      csv_lines.map! do |line|
+        from = line[index]
+        line[index] = options[from] if options.has_key?(from)
+        line
+      end
+    end
+
+    def values_at(header_key)
+      csv_lines.collect { |line| line[header_index(header_key)] }.uniq
+    end
+
+    def header_index(header_key)
+      headers.values.index(header_key)
+    end
+
     protected
 
     def process_file
       lines, batch_size = [], options[:batch_size].to_i
       File.open(file.path) do |f|
         # capture headers if not exist
-        prepare_headers{ CSV.parse(f.readline, @csv_options).first }
+        prepare_headers { CSV.parse(f.readline, @csv_options).first }
         f.each_line do |line|
           lines << line if line.present?
           if lines.size == batch_size || f.eof?

spec/fixtures/files/posts.csv

@@ -0,0 +1,4 @@
+"Title","Body","Author Name"
+"title1","some body","John"
+"title2","some body","Jane"
+

spec/import_spec.rb

@@ -44,6 +44,41 @@ def upload_file!(name, ext='csv')
     find_button('Import').click
   end
 
+  context "posts index" do
+    before do
+        Author.create!(name: 'John', last_name: 'Doe')
+        Author.create!(name: 'Jane', last_name: 'Roe')
+        add_post_resource({
+          validate: true,
+          headers_rewrites: {
+              :'Author Name' => :author_id
+          },
+          before_batch_import: ->(importer) do
+            authors_names = importer.values_at(:author_id)
+            # replacing author name with author id
+            authors   = Author.where(name: authors_names).pluck(:name, :id)
+            #{"Jane" => 2, "John" => 1}
+            options = Hash[*authors.flatten]
+            importer.batch_replace(:author_id, options)
+          end
+        })
+        visit '/admin/posts/import'
+        upload_file!(:posts)
+    end
+
+    it "should resolve author_id by author name" do
+      Post.all.each do |post|
+        expect(Author.where(id: post.author.id)).to be_present
+      end
+    end
+
+    it "should be imported" do
+      expect(Post.count).to eq(2)
+      expect(page).to have_content "Successfully imported 2 posts"
+    end
+
+  end
+
   context "authors index" do
     before do
       add_author_resource

spec/support/admin.rb

@@ -7,3 +7,14 @@ def add_author_resource(options = {}, &block)
   Rails.application.reload_routes!
 
 end
+
+
+def add_post_resource(options = {}, &block)
+
+  ActiveAdmin.register Post do
+     config.filters = false
+     active_admin_import(options, &block)
+  end
+  Rails.application.reload_routes!
+
+end

spec/support/rails_template.rb

@@ -1,9 +1,11 @@
 # Rails template to build the sample app for specs
 
 generate :model, 'author name:string{10}:uniq last_name:string birthday:date'
+generate :model, 'post title:string:uniq body:text author:references'
 
 #Add validation
 inject_into_file "app/models/author.rb", "  validates_presence_of :name\n", after: "Base\n"
+inject_into_file "app/models/post.rb", "   validates_presence_of :author\n", after: ":author\n"
 
 # Configure default_url_options in test environment
 inject_into_file "config/environments/test.rb", "  config.action_mailer.default_url_options = { :host => 'example.com' }\n", after: "config.cache_classes = true\n"