From 1d7b836eb6304ebb04fb460f3b04f683096d5a21 Mon Sep 17 00:00:00 2001 From: Chneemann Date: Sat, 22 Mar 2025 06:39:53 +0100 Subject: [PATCH] feat: implement filtering of tasks by status in the TaskViewSet class --- task_app/views.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/task_app/views.py b/task_app/views.py index f658782..5402687 100644 --- a/task_app/views.py +++ b/task_app/views.py @@ -3,9 +3,15 @@ from .models import Task, SubTask, AssignedTask from .serializers import TaskSerializer, SubTaskSerializer, AssignedTaskSerializer class TaskViewSet(viewsets.ModelViewSet): - queryset = Task.objects.all() + queryset = Task.objects.all() serializer_class = TaskSerializer + def get_queryset(self): + status = self.request.query_params.get('status', None) + if status: + return Task.objects.filter(status=status) + return Task.objects.all() + class SubTaskViewSet(viewsets.ModelViewSet): queryset = SubTask.objects.all() serializer_class = SubTaskSerializer