pollvotemodel.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. from django.db import models
  2. from misago.signals import (rename_user, move_thread,
  3. move_forum_content)
  4. class PollVote(models.Model):
  5. poll = models.ForeignKey('Poll', related_name="vote_set")
  6. forum = models.ForeignKey('Forum')
  7. thread = models.ForeignKey('Thread')
  8. option = models.ForeignKey('PollOption', null=True, blank=True)
  9. user = models.ForeignKey('User', null=True, blank=True, on_delete=models.SET_NULL)
  10. user_name = models.CharField(max_length=255, null=True, blank=True)
  11. user_slug = models.SlugField(max_length=255, null=True, blank=True)
  12. date = models.DateTimeField()
  13. ip = models.GenericIPAddressField()
  14. agent = models.CharField(max_length=255)
  15. class Meta:
  16. app_label = 'misago'
  17. def rename_user_handler(sender, **kwargs):
  18. PollVote.objects.filter(user=sender).update(
  19. user_name=sender.username,
  20. user_slug=sender.username_slug,
  21. )
  22. rename_user.connect(rename_user_handler, dispatch_uid="rename_user_poll_votes")
  23. def move_forum_content_handler(sender, **kwargs):
  24. PollVote.objects.filter(forum=sender).update(forum=kwargs['move_to'])
  25. move_forum_content.connect(move_forum_content_handler, dispatch_uid="move_forum_polls_votes")
  26. def move_thread_handler(sender, **kwargs):
  27. PollVote.objects.filter(thread=sender).update(forum=kwargs['move_to'])
  28. move_thread.connect(move_thread_handler, dispatch_uid="move_thread_polls_votes")