]> git.openstreetmap.org Git - rails.git/blobdiff - app/models/relation_member.rb
Fix up case statement syntax for ruby 1.9 compatibility
[rails.git] / app / models / relation_member.rb
index fbaf89dafd937d30871f30fed7c3bb5ab87e1d20..389e2ce5234cded24bba7548a567e6b7b9dee124 100644 (file)
@@ -1,24 +1,26 @@
 class RelationMember < ActiveRecord::Base
 class RelationMember < ActiveRecord::Base
-  set_table_name 'current_relation_members'
-  
-  set_primary_keys :id, :sequence_id
+  set_table_name 'current_relation_members'  
+  set_primary_keys :relation_id, :sequence_id
+
+  belongs_to :relation
   belongs_to :member, :polymorphic => true
   belongs_to :member, :polymorphic => true
-  belongs_to :relation, :foreign_key => :id
 
 
-  def after_find
-    self[:member_class] = self.member_type.classify
+  after_find :set_class_from_type
+  after_initialize :set_class_from_type
+  before_save :set_type_from_class
+
+  def member_type=(type)
+    self[:member_type] = type
+    self[:member_class] = type.capitalize
   end
 
   end
 
-  def after_initialize
+private
+
+  def set_class_from_type
     self[:member_class] = self.member_type.classify unless self.member_type.nil?
   end
 
     self[:member_class] = self.member_type.classify unless self.member_type.nil?
   end
 
-  def before_save
+  def set_type_from_class
     self.member_type = self[:member_class].classify
   end
     self.member_type = self[:member_class].classify
   end
-
-  def member_type=(type)
-    self[:member_type] = type
-    self[:member_class] = type.capitalize
-  end
 end
 end