Module: ActiveRecord::AttributeMethods::Write

Defined in:
lib/composite_primary_keys/attribute_methods/write.rb

Instance Method Summary collapse

Instance Method Details

#write_attribute(attr_name, value) ⇒ Object



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# File 'lib/composite_primary_keys/attribute_methods/write.rb', line 4

def write_attribute(attr_name, value)
  # CPK
  if attr_name.kind_of?(Array)
    unless value.length == attr_name.length
      raise "Number of attr_names and values do not match"
    end
    [attr_name, value].transpose.map {|name,val| write_attribute(name, val)}
    value
  else
    attr_name = attr_name.to_s
    # CPK
    # attr_name = self.class.primary_key if attr_name == 'id' && self.class.primary_key
    attr_name = self.class.primary_key if attr_name == 'id' && self.class.primary_key && !self.composite?
    @attributes_cache.delete(attr_name)
    column = column_for_attribute(attr_name)

    if column || @attributes.has_key?(attr_name)
      @attributes[attr_name] = type_cast_attribute_for_write(column, value)
    else
      raise ActiveModel::MissingAttributeError, "can't write unknown attribute `#{attr_name}'"
    end
  end
end