X-Git-Url: https://projects.mako.cc/source/selectricity/blobdiff_plain/1660465f3e072cbddc19b19f7f8869affb0694c5..905cf65f6b3e8acc894c268c3bf729e7b0c76292:/app/models/election.rb diff --git a/app/models/election.rb b/app/models/election.rb index be828d0..a574139 100644 --- a/app/models/election.rb +++ b/app/models/election.rb @@ -1,12 +1,142 @@ class Election < ActiveRecord::Base has_many :candidates + has_many :voters + has_many :votes + belongs_to :user validates_presence_of :name, :description + + #validate that method is one of the listed election types + + attr_reader :plurality_result + attr_reader :approval_result + attr_reader :condorcet_result + attr_reader :ssd_result + attr_reader :borda_result + + require 'date' + + def initialize(params={}) + super + self.enddate = read_attribute( :enddate ) || \ + Time.now + 14.days - 1.second + end + + def other_methods + if election_method + @other_methods = ELECTION_TYPES.reject {|i| i == election_method} + else + @other_methods = nil + end + @other_methods + end + + def startdate + read_attribute( :startdate ) || Time.now + 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 self.candidates.each do |candidate| candidate.destroy end - super destroy + super + end + + def start_blockers + reasons = [] + debugger + if self.candidates.length <= 1 + reasons << "You must have at least two candidates." + end + + if self.voters.length <= 1 + reasons << "You must have at least two voters." + end + + reasons + end + + def activate! + self.active = 1 + self.save! + end + + def quickvote? + self.class == 'QuickVote' + end + + def active? + active == 1 + end + + def done? + active == 2 + end + + def shortdesc + shortdesc = description.split(/\n/)[0] + end + + def longdesc + 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 + + { 'plurality' => @plurality_result, + 'approval' => @approval_result, + 'condorcet' => @condorcet_result, + 'ssd' => @ssd_result, + 'borda' => @borda_result } + 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 + +