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.
88 lines
3.1 KiB
88 lines
3.1 KiB
from django.shortcuts import get_object_or_404
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from pydantic import ValidationError
|
|
from rest_framework import filters, generics, status
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.views import APIView, Response
|
|
|
|
from examples.assignment.strategies import StrategyName
|
|
from examples.assignment.usecase import bulk_assign
|
|
from examples.assignment.workload import WorkloadAllocation
|
|
from examples.models import Assignment
|
|
from examples.serializers import AssignmentSerializer
|
|
from projects.models import Project
|
|
from projects.permissions import IsProjectAdmin, IsProjectStaffAndReadOnly
|
|
|
|
|
|
class AssignmentList(generics.ListCreateAPIView):
|
|
serializer_class = AssignmentSerializer
|
|
permission_classes = [IsAuthenticated & (IsProjectAdmin | IsProjectStaffAndReadOnly)]
|
|
filter_backends = (DjangoFilterBackend, filters.OrderingFilter)
|
|
ordering_fields = ("created_at", "updated_at")
|
|
model = Assignment
|
|
|
|
@property
|
|
def project(self):
|
|
return get_object_or_404(Project, pk=self.kwargs["project_id"])
|
|
|
|
def get_queryset(self):
|
|
queryset = self.model.objects.filter(project=self.project, assignee=self.request.user)
|
|
return queryset
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(project=self.project)
|
|
|
|
|
|
class AssignmentDetail(generics.RetrieveUpdateDestroyAPIView):
|
|
queryset = Assignment.objects.all()
|
|
serializer_class = AssignmentSerializer
|
|
lookup_url_kwarg = "assignment_id"
|
|
permission_classes = [IsAuthenticated & (IsProjectAdmin | IsProjectStaffAndReadOnly)]
|
|
|
|
|
|
class ResetAssignment(APIView):
|
|
permission_classes = [IsAuthenticated & IsProjectAdmin]
|
|
|
|
@property
|
|
def project(self):
|
|
return get_object_or_404(Project, pk=self.kwargs["project_id"])
|
|
|
|
def delete(self, *args, **kwargs):
|
|
Assignment.objects.filter(project=self.project).delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class BulkAssignment(APIView):
|
|
serializer_class = AssignmentSerializer
|
|
permission_classes = [IsAuthenticated & IsProjectAdmin]
|
|
|
|
def post(self, *args, **kwargs):
|
|
try:
|
|
strategy_name = StrategyName[self.request.data["strategy_name"]]
|
|
except KeyError:
|
|
return Response(
|
|
{"detail": "Invalid strategy name"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
try:
|
|
workload_allocation = WorkloadAllocation(workloads=self.request.data["workloads"])
|
|
except ValidationError as e:
|
|
return Response(
|
|
{"detail": e.errors()},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
try:
|
|
bulk_assign(
|
|
project_id=self.kwargs["project_id"],
|
|
strategy_name=strategy_name,
|
|
member_ids=workload_allocation.member_ids,
|
|
weights=workload_allocation.weights,
|
|
)
|
|
except ValueError as e:
|
|
return Response(
|
|
{"detail": str(e)},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
return Response(status=status.HTTP_201_CREATED)
|