X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/ac5e3489e8b886b7b8a7252ab56ddeb3d3bdadc7..1544cbab02fbc3333bc53a5ac98d38e377640d4a:/app/models/election.rb diff --git a/app/models/election.rb b/app/models/election.rb index eaf0727..d76c13f 100644 --- a/app/models/election.rb +++ b/app/models/election.rb @@ -4,15 +4,36 @@ class Election < ActiveRecord::Base has_many :votes belongs_to :user validates_presence_of :name, :description + + attr_reader :plurality_result + attr_reader :approval_result + attr_reader :condorcet_result + attr_reader :ssd_result + attr_reader :borda_result require 'date' def startdate - read_attribute( :startdate ) || DateTime.now + read_attribute( :startdate ) || Time.now end - + def enddate - read_attribute( :enddate ) || DateTime.now + 14 + date = read_attribute( :enddate ) || Time.now + 14 + date - 1.second + end + + def enddate=(date) + date += 1.day + date = Time.gm(*date) + super(date) + end + + def votes + votes = Array.new + self.voters.each do |voter| + votes << voter.vote + end + return votes end def destroy @@ -38,10 +59,19 @@ class Election < ActiveRecord::Base def activate! self.active = 1 + self.save! end def quickvote? - quickvote.to_i == 1 + self.class == 'QuickVote' + end + + def active? + active == 1 + end + + def done? + active == 2 end def shortdesc @@ -52,4 +82,45 @@ class Election < ActiveRecord::Base longdesc = description.split(/\n/)[1..-1].join("") longdesc.length > 0 ? longdesc : nil end + + #Calculate Election Results + def results + # initalize the tallies to empty arrays + preference_tally = Array.new + plurality_tally = Array.new + approval_tally = Array.new + + self.voters.each do |voter| + # skip if the voter has not voted or has an unconfirmed vote + next unless voter.voted? + + plurality_tally << voter.vote.rankings.sort[0].candidate.id + approval_tally << voter.vote.rankings.sort[0..1].collect \ + { |ranking| ranking.candidate.id } + preference_tally << voter.vote.rankings.sort.collect \ + { |ranking| ranking.candidate.id } + end + @plurality_result = PluralityVote.new(plurality_tally).result + @approval_result = ApprovalVote.new(approval_tally).result + @condorcet_result = PureCondorcetVote.new(preference_tally).result + @ssd_result = CloneproofSSDVote.new(preference_tally).result + @borda_result = BordaVote.new(preference_tally).result + #@runoff_result = InstantRunoffVote.new(preference_tally).result + + nil # to stay consistent + end + + def names_by_id + names = Hash.new + + competitors = self.candidates.sort.collect {|candidate| candidate.id} + competitors.each do |candidate| + names[candidate] = Candidate.find(candidate).name + end + + names + end + end + +