Added data model class

cykod · Nov 4, 2010 · 48ffc1e · 48ffc1e
1 parent 0b3fb77
commit 48ffc1e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/app/models/market_segment.rb b/app/models/market_segment.rb
@@ -123,6 +123,21 @@ def self.push_everyone_segment
      MarketSegment.first(:conditions => {:segment_type => 'everyone'}) || MarketSegment.create(:name => 'Everyone', :segment_type => 'everyone', :options => {})
    end
 
+   def data_model_class
+     case self.segment_type
+     when 'subscription'
+       UserSubscriptionEntry
+     when 'user_segment'
+       EndUser
+     when 'everyone'
+       EndUser
+     when 'content_model'
+       self.content_model.content_model
+     else
+       nil
+     end
+   end
+
    private
 
    def subscription_target_count(options={})