add pages vote

judge/models/__init__.py

@@ -54,6 +54,7 @@ from judge.models.submission import (
 )
 from judge.models.ticket import Ticket, TicketMessage
 from judge.models.volunteer import VolunteerProblemVote
+from judge.models.pagevote import PageVote, PageVoteVoter
 
 revisions.register(Profile, exclude=["points", "last_access", "ip", "rating"])
 revisions.register(Problem, follow=["language_limits"])
@@ -76,5 +77,6 @@ revisions.register(ProblemData)
 revisions.register(ProblemTestCase)
 revisions.register(ContestParticipation)
 revisions.register(Rating)
+revisions.register(PageVoteVoter)
 revisions.register(VolunteerProblemVote)
 del revisions

judge/models/pagevote.py

@@ -0,0 +1,40 @@
+from django.db import models
+from django.db.models import CASCADE
+from django.utils.translation import gettext_lazy as _
+
+from judge.models import Profile
+
+__all__ = ["PageVote", "PageVoteVoter"]
+
+
+class PageVote(models.Model):
+    page = models.CharField(
+        max_length=30,
+        verbose_name=_("associated page"),
+        db_index=True,
+    )
+    score = models.IntegerField(verbose_name=_("votes"), default=0)
+
+    class Meta:
+        verbose_name = _("pagevote")
+        verbose_name_plural = _("pagevotes")
+    
+    def vote_score(self, user):
+        page_vote = PageVoteVoter.objects.filter(pagevote=self, voter=user)
+        if page_vote.exists():
+            return page_vote.first().score
+        else:
+            return 0
+
+    def __str__(self):
+        return f"pagevote for {self.page}"
+
+class PageVoteVoter(models.Model):
+    voter = models.ForeignKey(Profile, related_name="voted_page", on_delete=CASCADE)
+    pagevote = models.ForeignKey(PageVote, related_name="votes", on_delete=CASCADE)
+    score = models.IntegerField()
+
+    class Meta:
+        unique_together = ["voter", "pagevote"]
+        verbose_name = _("pagevote vote")
+        verbose_name_plural = _("pagevote votes")