checkpointmodel.py 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. from django.db import models
  2. from misago.signals import (merge_post, merge_thread, move_forum_content,
  3. move_post, move_thread, rename_forum, rename_user)
  4. class Checkpoint(models.Model):
  5. forum = models.ForeignKey('Forum')
  6. thread = models.ForeignKey('Thread')
  7. action = models.CharField(max_length=255)
  8. extra = models.CharField(max_length=255, 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)
  11. user_slug = models.CharField(max_length=255)
  12. target_user = models.ForeignKey('User', null=True, blank=True, on_delete=models.SET_NULL, related_name='+')
  13. target_user_name = models.CharField(max_length=255, null=True, blank=True)
  14. target_user_slug = models.CharField(max_length=255, null=True, blank=True)
  15. old_forum = models.ForeignKey('Forum', null=True, blank=True, related_name='+')
  16. old_forum_name = models.CharField(max_length=255, null=True, blank=True)
  17. old_forum_slug = models.CharField(max_length=255, null=True, blank=True)
  18. date = models.DateTimeField()
  19. ip = models.GenericIPAddressField()
  20. agent = models.CharField(max_length=255)
  21. deleted = models.BooleanField(default=False)
  22. class Meta:
  23. app_label = 'misago'
  24. def rename_forum_handler(sender, **kwargs):
  25. Checkpoint.objects.filter(old_forum=sender).update(
  26. old_forum_name=sender.name,
  27. old_forum_slug=sender.slug,
  28. )
  29. rename_forum.connect(rename_forum_handler, dispatch_uid="rename_forum_checkpoints")
  30. def rename_user_handler(sender, **kwargs):
  31. Checkpoint.objects.filter(user=sender).update(
  32. user_name=sender.username,
  33. user_slug=sender.username_slug,
  34. )
  35. rename_user.connect(rename_user_handler, dispatch_uid="rename_user_checkpoints")
  36. def move_forum_content_handler(sender, **kwargs):
  37. Checkpoint.objects.filter(forum=sender).update(forum=kwargs['move_to'])
  38. move_forum_content.connect(move_forum_content_handler, dispatch_uid="move_forum_checkpoints")
  39. def move_thread_handler(sender, **kwargs):
  40. Checkpoint.objects.filter(thread=sender).update(forum=kwargs['move_to'])
  41. move_thread.connect(move_thread_handler, dispatch_uid="move_thread_checkpoints")
  42. def merge_thread_handler(sender, **kwargs):
  43. Checkpoint.objects.filter(thread=sender).delete()
  44. merge_thread.connect(merge_thread_handler, dispatch_uid="merge_threads_checkpoints")