merge.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. from rest_framework.response import Response
  2. from django.core.exceptions import PermissionDenied
  3. from django.utils.translation import ugettext as _
  4. from misago.acl import add_acl
  5. from misago.threads.serializers import MergePostsSerializer, PostSerializer
  6. def posts_merge_endpoint(request, thread):
  7. if not thread.acl['can_merge_posts']:
  8. raise PermissionDenied(_("You can't merge posts in this thread."))
  9. serializer = MergePostsSerializer(
  10. data=request.data,
  11. context={
  12. 'thread': thread,
  13. 'user': request.user,
  14. },
  15. )
  16. serializer.is_valid(raise_exception=True)
  17. posts = serializer.validated_data['posts']
  18. first_post, merged_posts = posts[0], posts[1:]
  19. for post in merged_posts:
  20. post.merge(first_post)
  21. post.delete()
  22. if first_post.pk == thread.first_post_id:
  23. first_post.set_search_document(thread.title)
  24. else:
  25. first_post.set_search_document()
  26. first_post.save()
  27. first_post.update_search_vector()
  28. first_post.save(update_fields=['search_vector'])
  29. first_post.postread_set.all().delete()
  30. thread.synchronize()
  31. thread.save()
  32. thread.category.synchronize()
  33. thread.category.save()
  34. first_post.thread = thread
  35. first_post.category = thread.category
  36. add_acl(request.user, first_post)
  37. return Response(PostSerializer(first_post, context={'user': request.user}).data)