theoooo-tally 0.0.7 → 0.0.8

data/app/models/tally_sheet.rb

@@ -6,6 +6,9 @@ class TallySheet < ActiveRecord::Base
   validates_uniqueness_of :tallyable_id, :scope => [:tallyable_type, :voter_type, :voter_id]
   validates_inclusion_of :for, :in => [true, false]
   
+  after_save :update_tally_score!
+  after_destroy :update_tally_score!
+  
   attr_protected :voter, :tallyable
   
   # TODO: Figure out why i can't just reference :tallyable => tallyable
@@ -18,4 +21,8 @@ class TallySheet < ActiveRecord::Base
     { :conditions => { :voter_id => voter.id, :voter_type => voter.class.to_s } }
   }
   
+  def update_tally_score!
+    self.tallyable.update_tally_score!
+  end
+  
 end

data/generators/tally/templates/tally_migration.rb.erb

@@ -14,8 +14,8 @@ class <%= migration_name %> < ActiveRecord::Migration
     add_index :tally_sheets, [ :tallyable_id, :tallyable_type ]
 
 <% tallyables.each do |t| -%>
-    add_column :<%= t.underscore.camelize.tableize %>, :tally_sheets_count, :integer
-    add_column :<%= t.underscore.camelize.tableize %>, :tally_score, :integer
+    add_column :<%= t.underscore.camelize.tableize %>, :tally_sheets_count, :integer, :default => 0
+    add_column :<%= t.underscore.camelize.tableize %>, :tally_score, :float, :default => 0.0
 <% end -%>
   end
 

data/lib/tally/tallyable.rb

@@ -8,7 +8,7 @@ module Tally
     module ClassMethods
       def has_tally      
         include InstanceMethods
-        has_many :votes, :as => :tallyable, :class_name => "TallySheet", :after_add => :update_tally_score!, :after_remove => :update_tally_score!
+        has_many :votes, :as => :tallyable, :class_name => "TallySheet"
         has_many :votes_for, :as => :tallyable, :class_name => "TallySheet", :conditions => {:for => true}
         has_many :votes_against, :as => :tallyable, :class_name => "TallySheet", :conditions => {:for => false}
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: theoooo-tally
 version: !ruby/object:Gem::Version 
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors: 
 - Theo Cushion