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