-
-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1592 from doccano/enhancement/refactorAnnotationAPI
[Enhancement] Refactor Annotation API
- Loading branch information
Showing
10 changed files
with
304 additions
and
23 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,57 @@ | ||
from django.core.exceptions import ValidationError | ||
from django.shortcuts import get_object_or_404 | ||
from rest_framework import generics, status | ||
from rest_framework.permissions import IsAuthenticated | ||
from rest_framework.response import Response | ||
|
||
from ...models import Project | ||
from ...permissions import IsInProjectOrAdmin, IsOwnAnnotation | ||
|
||
|
||
class BaseListAPI(generics.ListCreateAPIView): | ||
annotation_class = None | ||
pagination_class = None | ||
permission_classes = [IsAuthenticated & IsInProjectOrAdmin] | ||
swagger_schema = None | ||
|
||
@property | ||
def project(self): | ||
return get_object_or_404(Project, pk=self.kwargs['project_id']) | ||
|
||
def get_queryset(self): | ||
queryset = self.annotation_class.objects.filter(example=self.kwargs['example_id']) | ||
if not self.project.collaborative_annotation: | ||
queryset = queryset.filter(user=self.request.user) | ||
return queryset | ||
|
||
def create(self, request, *args, **kwargs): | ||
request.data['example'] = self.kwargs['example_id'] | ||
try: | ||
response = super().create(request, args, kwargs) | ||
except ValidationError as err: | ||
response = Response({'detail': err.messages}, status=status.HTTP_400_BAD_REQUEST) | ||
return response | ||
|
||
def perform_create(self, serializer): | ||
serializer.save(example_id=self.kwargs['example_id'], user=self.request.user) | ||
|
||
def delete(self, request, *args, **kwargs): | ||
queryset = self.get_queryset() | ||
queryset.all().delete() | ||
return Response(status=status.HTTP_204_NO_CONTENT) | ||
|
||
|
||
class BaseDetailAPI(generics.RetrieveUpdateDestroyAPIView): | ||
lookup_url_kwarg = 'annotation_id' | ||
swagger_schema = None | ||
|
||
@property | ||
def project(self): | ||
return get_object_or_404(Project, pk=self.kwargs['project_id']) | ||
|
||
def get_permissions(self): | ||
if self.project.collaborative_annotation: | ||
self.permission_classes = [IsAuthenticated & IsInProjectOrAdmin] | ||
else: | ||
self.permission_classes = [IsAuthenticated & IsInProjectOrAdmin & IsOwnAnnotation] | ||
return super().get_permissions() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,18 @@ | ||
from ...models import Category | ||
from ...serializers import CategorySerializer | ||
from .base import BaseDetailAPI, BaseListAPI | ||
|
||
|
||
class CategoryListAPI(BaseListAPI): | ||
annotation_class = Category | ||
serializer_class = CategorySerializer | ||
|
||
def create(self, request, *args, **kwargs): | ||
if self.project.single_class_classification: | ||
self.get_queryset().delete() | ||
return super().create(request, args, kwargs) | ||
|
||
|
||
class CategoryDetailAPI(BaseDetailAPI): | ||
queryset = Category.objects.all() | ||
serializer_class = CategorySerializer |
Oops, something went wrong.