doccano/backend/examples/views/comment.py

from django_filters.rest_framework import DjangoFilterBackend
from rest_framework import filters, generics, status
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response

from members.permissions import IsInProjectOrAdmin
from examples.models import Comment
from examples.permissions import IsOwnComment
from examples.serializers import CommentSerializer


class CommentList(generics.ListCreateAPIView):
    permission_classes = [IsAuthenticated & IsInProjectOrAdmin]
    serializer_class = CommentSerializer
    filter_backends = (DjangoFilterBackend, filters.SearchFilter)
    filterset_fields = ['example']
    search_fields = ('text',)

    def get_queryset(self):
        queryset = Comment.objects.filter(
            example__project_id=self.kwargs['project_id']
        )
        return queryset

    def perform_create(self, serializer):
        serializer.save(
            example_id=self.request.query_params.get('example'),
            user=self.request.user
        )

    def delete(self, request, *args, **kwargs):
        delete_ids = request.data['ids']
        Comment.objects.filter(user=request.user, pk__in=delete_ids).delete()
        return Response(status=status.HTTP_204_NO_CONTENT)


class CommentDetail(generics.RetrieveUpdateDestroyAPIView):
    queryset = Comment.objects.all()
    serializer_class = CommentSerializer
    lookup_url_kwarg = 'comment_id'
    permission_classes = [IsAuthenticated & IsInProjectOrAdmin & IsOwnComment]