Disable Vote Button if they already voted

TexasLAN · Feb 27, 2017 · 957dc1f · 957dc1f
1 parent 7eac2a8
commit 957dc1f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/texaslan/templates/voting/candidate_list.html b/texaslan/templates/voting/candidate_list.html
@@ -23,7 +23,7 @@ <h1 class="panel-title">Officer Panel</h1>
                     </div>
                 </div>
             {% endif %}
-            {% if voting_open %}
+            {% if voting_open and has_not_voted %}
                 <div class="panel panel-default">
                     <div class="panel-heading">
                         <h1 class="panel-title">Actions</h1>

diff --git a/texaslan/voting/views.py b/texaslan/voting/views.py
@@ -5,7 +5,7 @@
 from django.views.generic import ListView, CreateView, DetailView, FormView
 from django.forms import ValidationError
 
-from .models import Candidate, VoteBallot, CANDIDATE_POSITIONS, VoteService
+from .models import Candidate, VoteBallot, CANDIDATE_POSITIONS, VoteService, VoteStatus
 from .forms import StartElectionForm, CreateCandidateApplicationForm, VoteForm
 from texaslan.utils.utils import ActiveRequiredMixin, HasNotAppliedRequiredMixin, HasNotVotedRequiredMixin
 from texaslan.site_settings.models import SiteSettingService
@@ -24,6 +24,13 @@ def get_context_data(self, **kwargs):
 
         context['voting_open'] = SiteSettingService.is_voting_currently()
 
+        context['has_not_voted'] = True
+        try:
+            VoteStatus.objects.get(voter__username=self.request.user.username)
+            context['has_not_voted'] = False
+        except VoteStatus.DoesNotExist:
+            pass
+
         positions_list = []
         for (position_code, position_name) in CANDIDATE_POSITIONS:
             has_winner = False