Пакетное выполнение заданий в DRF

У меня есть мой Django сайт, где я могу иметь задачи, созданные и подзадачи под задачами у меня есть опция отметить завершенным, которая работает нормально, мне нужно, чтобы они были завершены пакетно, как выбор нескольких задач сразу и их завершение.

serializers.py:

class TaskCompleteSerializer(serializers.ModelSerializer):
     class Meta:
    model = Task
    fields = (
        'is_done',
    )

     def update(self, instance, validated_data):
         person = self.context['request'].user.person

         task_is_done = validated_data.get('is_done', False)

         if task_is_done:
            instance.subtasks.update(is_done=True)

         instance.is_done = task_is_done
         instance.done_by.set([person])
         instance.save()

         return instance

views.py:

class TaskUpdateAPIView(UpdateAPIView):
     permission_classes = " "
     serializer_class = TaskCompleteSerializer
     queryset = Task.objects.all()
     model = Task
     lookup_url_kwarg = 'task_id'

urls.py

path('<int:task_id>/complete/',views.TaskUpdateAPIView.as_view(), 
name='task_update'),

models.py

class Task(BaseModel):
    name = models.CharField(max_length=255)
    done_by = models.ManyToManyField(
    User,
    related_name='tasks_completed',
    blank=True,
    )
    is_done = models.BooleanField(default=False)

Думаю, вам нужно создать функцию API представления, чтобы легко реализовать это.

class TaskBulkUpdateSerializer(serializers.Serializer):
    task_ids = serializers.ListField(
        child = serializers.IntegerField())
    is_done = serializers.BooleanField()

В файле views.py нужно создать bulk_update_tasks функцию.

from .serializers import TaskBulkUpdateSerializer
from rest_framework.response import Response
from rest_framework import status
from rest_framework.permissions import IsAuthenticated
from .models import Task, Subtask

@api_view(['POST'])
@permission_classes([IsAuthenticated])
def bulk_update_tasks(request):
    serializer = TaskBulkUpdateSerializer(request.data)
    if serializer.is_valid():
        input_data = serializer.validated_data
        task_ids = serializers.get('task_ids')
        is_done = serializers.get('is_done')
        
        # check if all the tasks exist and all of them are owned by the user
        for task_id in task_ids:
            try:
                task = Task.objects.get(pk = task_id)                
                if task.done_by.id != request.user.id:
                    return Response(status = status.HTTP_400_BAD_REQUEST)
            except:
                return Response(status = status.HTTP_400_BAD_REQUEST)

        # if it's ok, then update
        Task.objects.filter(id__in = task_ids).update(is_done = is_done)

        # update subtasks
        if is_done:
            Subtask.objects.filter(subtask_of__id__in = task_ids).update(is_done = True)
        
        return Response(status = status.HTTP_200_OK)
    else:
        return Response(status = status.HTTP_400_BAD_REQUEST)

В urls.py необходимо добавить api.

path('complete/bulk', bulk_update_task, name='task_bulk_update')
Вернуться на верх