move nomination vote bootstrap classes to model

scipost_django/colleges/models/nomination.py

@@ -312,6 +312,12 @@ class FellowshipNominationVote(models.Model):
         (VOTE_DISAGREE, "Disagree"),
         (VOTE_VETO, "Veto"),
     )
+    VOTE_BS_CLASSES = {
+        VOTE_AGREE: "success",
+        VOTE_ABSTAIN: "warning",
+        VOTE_DISAGREE: "danger",
+        VOTE_VETO: "black",
+    }
 
     voting_round = models.ForeignKey(
         "colleges.FellowshipNominationVotingRound",
@@ -331,6 +337,10 @@ class FellowshipNominationVote(models.Model):
 
     objects = FellowshipNominationVoteQuerySet.as_manager()
 
+    @property
+    def get_vote_bs_class(self):
+        return self.VOTE_BS_CLASSES[self.vote]
+
     class Meta:
         constraints = [
             models.UniqueConstraint(

scipost_django/colleges/templates/colleges/_hx_nomination_vote.html

 <h3>Cast your vote:</h3>
 
-{% for vote_option, color in vote_options %}
+{% for vote_option, color in VOTE_BS_CLASSES.items %}
 
   <form hx-post="{% url 'colleges:_hx_nomination_vote' round_id=voting_round.id %}"
         hx-target="#nomination-{{ voting_round.nomination.id }}-vote">

scipost_django/colleges/templates/colleges/_hx_nomination_voter_table.html

@@ -41,16 +41,8 @@
 
             {% if not round.is_unscheduled %}
 
-              {% if voter.vote.vote == "agree" %}
-                <td class="text-success">{{ voter.vote.get_vote_display }}</td>
-              {% elif voter.vote.vote == "abstain" %}
-                <td class="text-warning">{{ voter.vote.get_vote_display }}</td>
-              {% elif voter.vote.vote == "disagree" %}
-                <td class="text-danger">{{ voter.vote.get_vote_display }}</td>
-              {% elif voter.vote.vote == "veto" %}
-                <td>
-                  <span class="badge bg-dark">{{ voter.vote.get_vote_display }}</span>
-                </td>
+              {% if voter.vote %}
+                <td class="text-{{ voter.vote.get_vote_bs_class }}">{{ voter.vote.get_vote_display }}</td>
               {% else %}
                 <td class="text-muted">No vote</td>
               {% endif %}