63 lines
No EOL
2.2 KiB
Python
63 lines
No EOL
2.2 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from .models import Task, SubTask, AssignedTask
|
|
from .serializers import TaskSerializer, SubTaskSerializer, AssignedTaskSerializer
|
|
from rest_framework.decorators import action
|
|
from .caching import get_cached_tasks, get_cached_task_by_id, get_cached_tasks_by_status
|
|
from django.core.cache import cache
|
|
from .choices import TaskStatus
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
class TaskViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
queryset = Task.objects.all()
|
|
serializer_class = TaskSerializer
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
status_param = request.query_params.get('status', None)
|
|
if status_param:
|
|
tasks = get_cached_tasks_by_status(status_param)
|
|
else:
|
|
tasks = get_cached_tasks()
|
|
serializer = self.get_serializer(tasks, many=True)
|
|
return Response(serializer.data)
|
|
|
|
def retrieve(self, request, pk=None):
|
|
task = get_cached_task_by_id(pk)
|
|
serializer = self.get_serializer(task)
|
|
return Response(serializer.data)
|
|
|
|
@action(detail=True, methods=['put'])
|
|
def update_status(self, request, pk=None):
|
|
task = self.get_object()
|
|
new_status = request.data.get('status')
|
|
|
|
if not new_status:
|
|
return Response({'error': 'Status is required.'}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
valid_statuses = [choice[0] for choice in TaskStatus.choices]
|
|
if new_status not in valid_statuses:
|
|
return Response({'error': f'Invalid status. Valid statuses: {valid_statuses}'}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
cache.delete(f"task_{task.id}")
|
|
cache.delete(f"tasks_by_status_{task.status}")
|
|
|
|
task.status = new_status
|
|
task.save()
|
|
return Response({'status': 'Status updated.'})
|
|
|
|
|
|
class SubTaskViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
queryset = SubTask.objects.all()
|
|
serializer_class = SubTaskSerializer
|
|
|
|
|
|
class AssignedTaskViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
queryset = AssignedTask.objects.all()
|
|
serializer_class = AssignedTaskSerializer |