NDOJ/judge/models/pagevote.py

63 lines
2 KiB
Python
Raw Normal View History

2022-11-16 15:43:03 +00:00
from django.db import models
from django.db.models import CASCADE
from django.utils.translation import gettext_lazy as _
from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
2022-11-16 15:43:03 +00:00
from judge.models.profile import Profile
2023-10-10 22:38:48 +00:00
from judge.caching import cache_wrapper
2022-11-16 15:43:03 +00:00
__all__ = ["PageVote", "PageVoteVoter"]
class PageVote(models.Model):
page = models.CharField(
max_length=30,
verbose_name=_("associated page"),
db_index=True,
) # deprecated
2022-11-16 15:43:03 +00:00
score = models.IntegerField(verbose_name=_("votes"), default=0)
content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
object_id = models.PositiveIntegerField()
linked_object = GenericForeignKey("content_type", "object_id")
2022-11-16 15:43:03 +00:00
class Meta:
verbose_name = _("pagevote")
verbose_name_plural = _("pagevotes")
indexes = [
models.Index(fields=["content_type", "object_id"]),
]
unique_together = ("content_type", "object_id")
2022-11-17 00:48:32 +00:00
2023-10-10 22:38:48 +00:00
@cache_wrapper(prefix="PVvs")
2022-11-16 15:43:03 +00:00
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.linked_object}"
2022-11-16 15:43:03 +00:00
2022-11-17 00:48:32 +00:00
2022-11-16 15:43:03 +00:00
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")
2022-11-17 00:48:32 +00:00
verbose_name_plural = _("pagevote votes")
class PageVotable:
def get_or_create_pagevote(self):
if self.pagevote.count():
return self.pagevote.first()
new_pagevote = PageVote()
new_pagevote.linked_object = self
new_pagevote.save()
return new_pagevote