From a02d237518c401c6b85d7b1f36b56990753c6605 Mon Sep 17 00:00:00 2001
From: Robin Roestenburg <robinroestenburg@gmail.com>
Date: Tue, 6 Mar 2012 23:54:38 +0100
Subject: Removed flag attributes.

---
 .../lib/active_record/attribute_methods.rb         | 51 ++++++++++++++--------
 .../lib/active_record/locking/optimistic.rb        |  2 +-
 activerecord/lib/active_record/persistence.rb      |  4 +-
 3 files changed, 35 insertions(+), 22 deletions(-)

(limited to 'activerecord/lib')

diff --git a/activerecord/lib/active_record/attribute_methods.rb b/activerecord/lib/active_record/attribute_methods.rb
index 9f89f5572a..4b0925a9ec 100644
--- a/activerecord/lib/active_record/attribute_methods.rb
+++ b/activerecord/lib/active_record/attribute_methods.rb
@@ -214,40 +214,53 @@ module ActiveRecord
       value
     end
 
-    # Returns a copy of the attributes hash where all the values have been
-    # typecasted for use in an Arel insert/update method.
-    def arel_attributes_values(pk_attr_allowed = true, readonly_attr_allowed = true, attribute_names = @attributes.keys)
-      attrs      = {}
-      arel_table = self.class.arel_table
-
-      attribute_names.each do |name|
-        if attribute_allowed?(pk_attr_allowed, readonly_attr_allowed, name) 
-          attrs[arel_table[name]] = typecasted_attribute_value(name)
-        end
-      end
+    def arel_attributes_with_values_for_create(pk_attribute_allowed)
+      arel_attributes_with_values(attributes_for_create(pk_attribute_allowed))
+    end
 
-      attrs
+    def arel_attributes_with_values_for_update(attribute_names)
+      arel_attributes_with_values(attributes_for_update(attribute_names))
     end
 
     def attribute_method?(attr_name)
       defined?(@attributes) && @attributes.include?(attr_name)
     end
 
-    private 
+    private
 
-    def attribute_allowed?(pk_attribute_allowed, readonly_attribute_allowed, name)
-      return unless column = column_for_attribute(name)
+    # Returns a Hash of the Arel::Attributes and attribute values that have been
+    # type casted for use in an Arel insert/update method.
+    def arel_attributes_with_values(attribute_names)
+      attrs = {}
+      arel_table = self.class.arel_table
 
-      (pk_attribute_allowed || !pk_attribute?(column)) && 
-        (readonly_attribute_allowed || !readonly_attribute?(name))
+      attribute_names.each do |name|
+        attrs[arel_table[name]] = typecasted_attribute_value(name)
+      end
+      attrs
+    end
+
+    # Filters the primary keys and readonly attributes from the attribute names.
+    def attributes_for_update(attribute_names)
+      attribute_names.select do |name|
+        column_for_attribute(name) && !pk_attribute?(name) && !readonly_attribute?(name)
+      end
+    end
+
+    # Filters out the primary keys, from the attribute names, when the primary
+    # key is to be generated (e.g. the id attribute has no value).
+    def attributes_for_create(pk_attribute_allowed)
+      @attributes.keys.select do |name|
+        column_for_attribute(name) && (pk_attribute_allowed || !pk_attribute?(name))
+      end
     end
 
     def readonly_attribute?(name)
       self.class.readonly_attributes.include?(name)
     end
 
-    def pk_attribute?(column)
-      column.primary
+    def pk_attribute?(name)
+      column_for_attribute(name).primary
     end
 
     def typecasted_attribute_value(name)
diff --git a/activerecord/lib/active_record/locking/optimistic.rb b/activerecord/lib/active_record/locking/optimistic.rb
index 9b2dc096a0..8266427b71 100644
--- a/activerecord/lib/active_record/locking/optimistic.rb
+++ b/activerecord/lib/active_record/locking/optimistic.rb
@@ -84,7 +84,7 @@ module ActiveRecord
               relation.table[self.class.primary_key].eq(id).and(
                 relation.table[lock_col].eq(quote_value(previous_lock_value))
               )
-            ).arel.compile_update(arel_attributes_values(false, false, attribute_names))
+            ).arel.compile_update(arel_attributes_with_values_for_update(attribute_names))
 
             affected_rows = connection.update stmt
 
diff --git a/activerecord/lib/active_record/persistence.rb b/activerecord/lib/active_record/persistence.rb
index c4bce87311..09502bb52c 100644
--- a/activerecord/lib/active_record/persistence.rb
+++ b/activerecord/lib/active_record/persistence.rb
@@ -350,7 +350,7 @@ module ActiveRecord
     # Updates the associated record with values matching those of the instance attributes.
     # Returns the number of affected rows.
     def update(attribute_names = @attributes.keys)
-      attributes_with_values = arel_attributes_values(false, false, attribute_names)
+      attributes_with_values = arel_attributes_with_values_for_update(attribute_names)
       return 0 if attributes_with_values.empty?
       klass = self.class
       stmt = klass.unscoped.where(klass.arel_table[klass.primary_key].eq(id)).arel.compile_update(attributes_with_values)
@@ -360,7 +360,7 @@ module ActiveRecord
     # Creates a record with values matching those of the instance attributes
     # and returns its id.
     def create
-      attributes_values = arel_attributes_values(!id.nil?)
+      attributes_values = arel_attributes_with_values_for_create(!id.nil?)
 
       new_id = self.class.unscoped.insert attributes_values
 
-- 
cgit v1.2.3