X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/9129a21b28c88064edfcfc8a8fa23a640f29819c..ee280d9250b5b4c925cc6b4809ce40437d600e7d:/app/models/election.rb diff --git a/app/models/election.rb b/app/models/election.rb index 08ef1a5..ac8d628 100644 --- a/app/models/election.rb +++ b/app/models/election.rb @@ -2,19 +2,8 @@ # Copyright (C) 2007, 2008 Benjamin Mako Hill # Copyright (C) 2007 Massachusetts Institute of Technology # -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU Affero General Public License as -# published by the Free Software Foundation, either version 3 of the -# License, or (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, but -# WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -# Affero General Public License for more details. -# -# You should have received a copy of the GNU Affero General Public -# License along with this program. If not, see -# . +# This program is free software. Please see the COPYING file for +# details. class Election < ActiveRecord::Base has_many :candidates @@ -22,9 +11,12 @@ class Election < ActiveRecord::Base has_many :votes belongs_to :user validates_presence_of :name, :description - + + # enforce constraints associated with dependencies (i.e., a kiosk + # election can't also be unauthenticated) + before_save :enforce_constraints + #validate that method is one of the listed election types - attr_reader :plurality_result attr_reader :approval_result attr_reader :condorcet_result @@ -32,7 +24,7 @@ class Election < ActiveRecord::Base attr_reader :borda_result require 'date' - + def initialize(params={}) super self.enddate = read_attribute( :enddate ) || \ @@ -82,11 +74,11 @@ class Election < ActiveRecord::Base def activate! self.active = 1 - self.save! + self.save end def quickvote? - self.class == 'QuickVote' + self.class == QuickVote end def active? @@ -100,6 +92,10 @@ class Election < ActiveRecord::Base def authenticated? authenticated end + + def kiosk? + kiosk + end def shortdesc shortdesc = description.split(/\n/)[0] @@ -141,11 +137,11 @@ class Election < ActiveRecord::Base # 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 + plurality_tally << voter.vote.rankings.sort[0].candidate_id approval_tally << voter.vote.rankings.sort[0..1].collect \ - { |ranking| ranking.candidate.id } + { |ranking| ranking.candidate_id } preference_tally << voter.vote.rankings.sort.collect \ - { |ranking| ranking.candidate.id } + { |ranking| ranking.candidate_id } end @plurality_result = PluralityVote.new(plurality_tally).result @@ -171,6 +167,40 @@ class Election < ActiveRecord::Base names end -end + def candidate_hash + hash = {} + self.candidates.each {|c| hash[c.id] = c} + return hash + end + + # TODO now that this code is in here, we should go ahead and remove + # date checking from other places in the code + def after_find + if self.active < 2 and self.enddate < Time.now + self.active = 2 + self.save + end + end + + def just_before_midnight(datetime) + if datetime.instance_of? DateTime + off = datetime.offset + else + off = Rational(datetime.utc_offset, 60*60*24) + end + return DateTime.new(datetime.year, datetime.month, datetime.day, + 23, 59, 59, off) + end + + private + def enforce_constraints + # elections end just before midnight + self.enddate = just_before_midnight(self.enddate) + # kiosks can't be authenticated + self.authenticated = false if kiosk? + return true + end + +end