lib/serialization_helper.rb in yaml_db-0.1.0 vs lib/serialization_helper.rb in yaml_db-0.2.0

@@ -77,13 +77,15 @@
 
     def self.load_records(table, column_names, records)
       if column_names.nil?
         return
       end
+      columns = column_names.map{|cn| ActiveRecord::Base.connection.columns(table).detect{|c| c.name == cn}}
       quoted_column_names = column_names.map { |column| ActiveRecord::Base.connection.quote_column_name(column) }.join(',')
       quoted_table_name = SerializationHelper::Utils.quote_table(table)
       records.each do |record|
-        ActiveRecord::Base.connection.execute("INSERT INTO #{quoted_table_name} (#{quoted_column_names}) VALUES (#{record.map { |r| ActiveRecord::Base.connection.quote(r) }.join(',')})")
+        quoted_values = record.zip(columns).map{|c| ActiveRecord::Base.connection.quote(c.first, c.last)}.join(',')
+        ActiveRecord::Base.connection.execute("INSERT INTO #{quoted_table_name} (#{quoted_column_names}) VALUES (#{quoted_values})")
       end
     end
 
     def self.reset_pk_sequence!(table_name)
       if ActiveRecord::Base.connection.respond_to?(:reset_pk_sequence!)
\ No newline at end of file