mirror of
https://github.com/ENSL/ensl.org.git
synced 2024-11-15 17:31:27 +00:00
4199062a9c
Added model annotations
68 lines
1.8 KiB
Ruby
68 lines
1.8 KiB
Ruby
# == Schema Information
|
|
#
|
|
# Table name: votes
|
|
#
|
|
# id :integer not null, primary key
|
|
# user_id :integer
|
|
# votable_id :integer
|
|
# poll_id :integer
|
|
# votable_type :string(255)
|
|
#
|
|
|
|
class Vote < ActiveRecord::Base
|
|
include Extra
|
|
|
|
attr_protected :id, :updated_at, :created_at, :user_id
|
|
|
|
validates_uniqueness_of :user_id, :scope => :votable_id
|
|
validates_presence_of :user_id, :votable_id, :votable_type
|
|
|
|
belongs_to :user
|
|
belongs_to :votable, :polymorphic => true
|
|
|
|
after_create :increase_votes
|
|
after_destroy :decrease_votes
|
|
|
|
def increase_votes
|
|
if votable_type == "Option"
|
|
votable.poll.increment :votes
|
|
votable.poll.save
|
|
end
|
|
votable.increment :votes
|
|
votable.save
|
|
end
|
|
|
|
def decrease_votes
|
|
if votable_type == "Option"
|
|
votable.poll.decrement :votes
|
|
votable.poll.save
|
|
end
|
|
votable.decrement :votes
|
|
votable.save
|
|
end
|
|
|
|
def can_create? cuser
|
|
return false unless cuser
|
|
if votable_type == "Option"
|
|
if votable.poll.voted?(cuser)
|
|
return false
|
|
end
|
|
elsif votable_type == "Gatherer" or votable_type == "GatherMap" or votable_type == "GatherServer"
|
|
return false unless votable.gather.users.exists? cuser
|
|
|
|
case votable_type
|
|
when "Gatherer" then
|
|
return false if votable.gather.status != Gather::STATE_VOTING
|
|
return false if votable.gather.gatherer_votes.count(:conditions => {:user_id => user.id}) > 1
|
|
when "GatherMap" then
|
|
return false if votable.gather.status == Gather::STATE_FINISHED
|
|
return false if votable.gather.map_votes.count(:conditions => {:user_id => user.id}) > 1
|
|
when "GatherServer" then
|
|
return false if votable.gather.status == Gather::STATE_FINISHED
|
|
return false if votable.gather.server_votes.first :conditions => {:user_id => user.id}
|
|
end
|
|
end
|
|
|
|
return true
|
|
end
|
|
end
|