X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/75091532075247ae9c47164ea883f4fe803eddf4..3e0d5a9b6af9ec1283b0c0e8cfae6cca9c2bea3a:/app/models/voter.rb diff --git a/app/models/voter.rb b/app/models/voter.rb index 7139b89..012c795 100644 --- a/app/models/voter.rb +++ b/app/models/voter.rb @@ -1,8 +1,20 @@ class Voter < ActiveRecord::Base belongs_to :election has_one :vote + + def reset_vote + self.vote.destroy + self.reload + end + + def destroy + vote.destroy if vote + super + end + def voted? + vote.confirmed == 1 + end end -