21 lines
No EOL
745 B
Python
21 lines
No EOL
745 B
Python
from rest_framework import viewsets
|
|
from .models import Task, SubTask, AssignedTask
|
|
from .serializers import TaskSerializer, SubTaskSerializer, AssignedTaskSerializer
|
|
|
|
class TaskViewSet(viewsets.ModelViewSet):
|
|
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
|
|
|
|
class AssignedTaskViewSet(viewsets.ModelViewSet):
|
|
queryset = AssignedTask.objects.all()
|
|
serializer_class = AssignedTaskSerializer |