e9a52a115be4970db2abdeb921ca0a790ed30809,app/server/views.py,ProjectDocsAPI,get_queryset,#ProjectDocsAPI#,156

Before Change

        project = get_object_or_404(Project, pk=project_id)
        isnull = self.request.query_params.get("is_checked") == "true"
        if project.is_type_of(Project.DOCUMENT_CLASSIFICATION):
            queryset = queryset.filter(doc_annotations__isnull=isnull).distinct()
        elif project.is_type_of(Project.SEQUENCE_LABELING):
            queryset = queryset.filter(seq_annotations__isnull=isnull).distinct()
        elif project.is_type_of(Project.Seq2seq):
            queryset = queryset.filter(seq2seq_annotations__isnull=isnull).distinct()
        else:
            queryset = queryset

        return queryset


class AnnotationsAPI(generics.ListCreateAPIView):

After Change

        project = get_object_or_404(Project, pk=project_id)
        is_null = self.request.query_params.get("is_checked") == "true"
        queryset = Factory.get_documents(project, is_null).distinct()

        return queryset

Instances