mygpo/votes/models.py

   1 from django.db import models
   2 from django.conf import settings
   3 from django.contrib.contenttypes.models import ContentType
   4 from django.contrib.contenttypes.fields import GenericRelation, GenericForeignKey
   5
   6 from mygpo.core.models import UpdateInfoModel
   7
   8
   9 class Vote(UpdateInfoModel):
  10     """A vote by a user for some object"""
  11
  12     # the user who voted
  13     user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
  14
  15     # the object that was voted for
  16     content_type = models.ForeignKey(ContentType, on_delete=models.PROTECT)
  17     # this should suit UUID and integer primary keys
  18     object_id = models.UUIDField()
  19     content_object = GenericForeignKey("content_type", "object_id")
  20
  21     class Meta:
  22         unique_together = [
  23             # a user can only vote once per object
  24             ("user", "content_type", "object_id")
  25         ]
  26
  27
  28 class VoteMixin(models.Model):
  29
  30     votes = GenericRelation(Vote, related_query_name="votes")
  31
  32     class Meta:
  33         abstract = True
  34
  35     def vote(self, user):
  36         """Register a vote from the user for the current object"""
  37         Vote.objects.get_or_create(
  38             user=user,
  39             content_type=ContentType.objects.get_for_model(self),
  40             object_id=obj.pk,
  41         )
  42
  43     def vote_count(self):
  44         return self.votes.count()