acts_as_importable 0.3.0 → 0.4.0

data/README.rdoc

@@ -16,7 +16,15 @@ If model name is not specified, it will use the controller name to find the mode
 
   acts_as_importable :product, :scoped => :current_store
 
-This will import all material records within the current_store object returned by the controller#current_store method. Product should have a belongs_to association to Store for this feature to work.
+This will import all products within the current_store object returned by the controller#current_store method. Product should have a belongs_to association to Store for this feature to work.
+
+==== Updating existing objects:
+
+  acts_as_importable :product, :find_existing_by => :name
+
+This will update existing products whose name matches with the value of name column in the csv row. However, if :find_by_existing is not specified, no attempt will be made to lookup existing products. e.g. the following will always create new products
+
+  acts_as_importable :product
 
 === Add the following to your Model:
 

data/VERSION

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

data/acts_as_importable.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{acts_as_importable}
-  s.version = "0.3.0"
+  s.version = "0.4.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jagdeep Singh"]
-  s.date = %q{2011-01-13}
+  s.date = %q{2011-01-15}
   s.description = %q{Use this gem to add import/export to .csv functionality to your activerecord models}
   s.email = %q{jagdeepkh@gmail.com}
   s.extra_rdoc_files = [

data/lib/import_export/controller_methods.rb

@@ -28,16 +28,14 @@ module ImportExport
 
       def import
         @new_objects = []
-        filename = "#{UPLOADS_PATH}/#{self.controller_name}.csv"
-        context = {}
-        self.class.context.each_pair do |key, value|
-          context[key] = self.send(value)
-        end
+        filename = upload_file_name
+        context = self.class.context.clone
+        context[:scoped] = self.send(context[:scoped]) if context[:scoped]
 
         if File.exists? filename
           begin
             @new_objects = self.class.model_class.import(filename, context)
-            flash[:notice] = "Import Successful - #{@new_objects.length} New #{self.class.model_class.name.underscore.humanize.pluralize}"
+            flash[:notice] = "Import Successful - Imported #{@new_objects.length} #{self.class.model_class.name.underscore.humanize.pluralize}"
           rescue Exception => e
             logger.error "Error: Unable to process file. #{e}"
             flash[:error] = "Error: Unable to process file. #{e}"
@@ -54,8 +52,7 @@ module ImportExport
         require 'ftools'
         if params[:csv_file] && File.extname(params[:csv_file].original_filename) == '.csv'
           File.makedirs "#{UPLOADS_PATH}"
-          File.open("#{UPLOADS_PATH}/#{self.controller_name}.csv", "wb") { |f| f.write(params[:csv_file].read)}
-          flash[:notice] = "Successful import of #{self.controller_name} data file"
+          File.open(upload_file_name, "wb") { |f| f.write(params[:csv_file].read)}
         else
           flash[:error] = "Error! Invalid file, please select a csv file."
         end
@@ -64,8 +61,12 @@ module ImportExport
 
       private
 
+      def upload_file_name
+        @upload_file_name ||= "#{UPLOADS_PATH}/#{self.controller_name}_#{request.session_options[:id]}.csv"
+      end
+
       def export_file_name
-        "#{Time.now.to_formatted_s(:number)}_#{self.controller_name}_export.csv"
+        @export_file_name ||= "#{Time.now.to_formatted_s(:number)}_#{self.controller_name}_export.csv"
       end
 
     end # end of module InstanceMethods

data/lib/import_export.rb

@@ -13,9 +13,8 @@ module ModelMethods
       self.export_fields = options[:export_fields]
       send :include, InstanceMethods
     end
-    
-    def import(filename, context)
 
+    def import(filename, context)
       collection = []
       headers, *data  = self.read_csv(filename)
       scope_object = context[:scoped]
@@ -26,40 +25,18 @@ module ModelMethods
 
           begin
             class_or_association = scope_object ? scope_object.send(self.table_name) : self
-            key_field = self.import_fields.first
-            key_value = data_row[0]
-            finder_method = "find_by_#{key_field}"
-            existing_element = class_or_association.send(finder_method, key_value)
-            element = existing_element || class_or_association.new
+            if key_field = context[:find_existing_by]
+              key_value = data_row[index_of(key_field)]
+              element = class_or_association.send("find_by_#{key_field}", key_value) || class_or_association.new
+            else
+              element = class_or_association.new
+            end
 
             Rails.logger.info "#{element.new_record? ? "Creating new" : "Updating existing"} record from #{data_row.inspect}"
 
             self.import_fields.each_with_index do |field_name, field_index|
               if field_name.include?('.')
-                method_name = 'assign_' + field_name.split('.')[0]
-                if element.respond_to?(method_name)
-                  element.send method_name, data_row
-                elsif element.respond_to?("#{field_name.split('.')[0]}_id")
-                  create_record = field_name.include?('!')
-                  split_field = field_name.gsub(/!/,'').split('.').compact
-                  initial_class = split_field[0].classify.constantize
-                  if initial_class and initial_class.respond_to?("find_by_#{split_field[1]}")
-                    begin
-                      e = initial_class.send("find_by_#{split_field[1]}", data_row[field_index])
-                    rescue
-                      e = nil
-                    end
-                    # try to create a new record if not found in database and if '!' is present
-                    if e.nil? and create_record and !data_row[field_index].blank?
-                      e = initial_class.create!("#{split_field[1]}" => data_row[field_index])
-                    end
-                    if e.kind_of?(ActiveRecord::Base)
-                      element["#{split_field[0]}_id"] = e.id
-                    end
-                  else
-                    e = nil
-                  end
-                end
+                assign_association(element, field_name, field_index, scope_object, data_row)
               else
                 element.send "#{field_name}=", data_row[field_index]
               end
@@ -130,10 +107,45 @@ module ModelMethods
         raise ArgumentError, "File does not exist."
       end
     end
+
+    def index_of(fieldname)
+      @import_field_indices ||= {}
+      @import_field_indices[fieldname] ||= self.import_fields.index{ |f| f.to_s == fieldname.to_s }
+    end
+
+    protected
+
+    def assign_association(element, field_name, field_index, scope_object, data_row)
+      create_record = field_name.include?('!')
+      association_name, association_attribute = field_name.gsub(/!/,'').split('.')
+      assign_association_method = "assign_#{association_name}"
+      association_fk = "#{association_name}_id"
+
+      if element.respond_to?(assign_association_method)
+        element.send assign_association_method, data_row
+      elsif element.respond_to?(association_fk)
+        association_class = association_name.classify.constantize
+
+        if scope_object && scope_object.respond_to?(association_class.table_name)
+          association_class = scope_object.send(association_class.table_name)
+        end
+
+        finder_method = "find_by_#{association_attribute}"
+        if association_class and association_class.respond_to?(finder_method)
+          e = association_class.send(finder_method, data_row[field_index])
+          if e.nil? and create_record and !data_row[field_index].blank?
+            e = association_class.create!(association_attribute => data_row[field_index])
+          end
+          element[association_fk] = e.id if e
+        end
+      end
+    end
   end
 
   module InstanceMethods
-    # any method placed here will apply to instaces, like @hickwall
+    def index_of(fieldname)
+      self.class.index_of(fieldname)
+    end
   end
 
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: acts_as_importable
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 15
   prerelease: false
   segments: 
   - 0
-  - 3
+  - 4
   - 0
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Jagdeep Singh
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-13 00:00:00 +05:30
+date: 2011-01-15 00:00:00 +05:30
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency