mirror of https://github.com/doccano/doccano.git
pythonannotation-tooldatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learning
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.5 KiB
41 lines
1.5 KiB
from django.conf import settings
|
|
from rest_framework import generics, status
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
|
|
from ..models import Project, Role, RoleMapping
|
|
from ..permissions import IsInProjectReadOnlyOrAdmin
|
|
from ..serializers import ProjectPolymorphicSerializer, ProjectSerializer
|
|
|
|
|
|
class ProjectList(generics.ListCreateAPIView):
|
|
serializer_class = ProjectPolymorphicSerializer
|
|
pagination_class = None
|
|
permission_classes = [IsAuthenticated, ]
|
|
|
|
def get_queryset(self):
|
|
return self.request.user.projects
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(users=[self.request.user])
|
|
|
|
def delete(self, request, *args, **kwargs):
|
|
delete_ids = request.data['ids']
|
|
projects = Project.objects.filter(
|
|
role_mappings__user=self.request.user,
|
|
role_mappings__role__name=settings.ROLE_PROJECT_ADMIN,
|
|
pk__in=delete_ids
|
|
)
|
|
# Todo: I want to use bulk delete.
|
|
# But it causes the constraint error.
|
|
# See https://github.com/django-polymorphic/django-polymorphic/issues/229
|
|
for project in projects:
|
|
project.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class ProjectDetail(generics.RetrieveUpdateDestroyAPIView):
|
|
queryset = Project.objects.all()
|
|
serializer_class = ProjectSerializer
|
|
lookup_url_kwarg = 'project_id'
|
|
permission_classes = [IsAuthenticated & IsInProjectReadOnlyOrAdmin]
|