125 lines
No EOL
5 KiB
Python
125 lines
No EOL
5 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_task_by_id
|
|
from django.core.cache import cache
|
|
from .choices import TaskStatus
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from user_app.models import User
|
|
from rest_framework.exceptions import APIException
|
|
|
|
class TaskViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAuthenticated]
|
|
queryset = Task.objects.all()
|
|
serializer_class = TaskSerializer
|
|
|
|
def retrieve(self, request, pk=None):
|
|
task = get_cached_task_by_id(pk)
|
|
if not task:
|
|
return Response({'error': 'Task not found.'}, status=status.HTTP_404_NOT_FOUND)
|
|
serializer = self.get_serializer(task)
|
|
return Response(serializer.data)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
if not serializer.is_valid():
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
task = serializer.save()
|
|
self._assign_users_to_task(request.data.get('assigned', []), task)
|
|
|
|
subtasks = [
|
|
{"title": title, "status": status}
|
|
for title, status in zip(request.data.get("subtasks_title", []), request.data.get("subtasks_status", []))
|
|
]
|
|
|
|
if subtasks:
|
|
self._create_subtasks(subtasks, task)
|
|
|
|
cache.delete(f"task_{task.id}")
|
|
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
|
|
def destroy(self, request, pk=None):
|
|
try:
|
|
task = Task.objects.get(pk=pk)
|
|
task.delete()
|
|
cache.delete(f"task_{pk}")
|
|
return Response({'message': 'Task deleted successfully.'}, status=status.HTTP_204_NO_CONTENT)
|
|
except Task.DoesNotExist:
|
|
return Response({'error': 'Task not found.'}, status=status.HTTP_404_NOT_FOUND)
|
|
|
|
def _create_subtasks(self, subtask_data, parent_task):
|
|
if subtask_data:
|
|
subtasks = [
|
|
SubTask(title=sub['title'], status=sub['status'], task=parent_task)
|
|
for sub in subtask_data
|
|
]
|
|
SubTask.objects.bulk_create(subtasks)
|
|
|
|
def _assign_users_to_task(self, user_ids, task):
|
|
users = User.objects.filter(id__in=user_ids)
|
|
missing_users = set(user_ids) - set(users.values_list('id', flat=True))
|
|
|
|
if missing_users:
|
|
missing_users_list = list(map(str, missing_users))
|
|
raise APIException(f"Users not found: {', '.join(missing_users_list)}")
|
|
|
|
AssignedTask.objects.bulk_create(
|
|
[AssignedTask(user_id=user, task=task) for user in users]
|
|
)
|
|
|
|
@action(detail=True, methods=['patch'])
|
|
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}")
|
|
task.status = new_status
|
|
task.save()
|
|
return Response({'status': 'Status updated.'})
|
|
|
|
@action(detail=True, methods=['patch'], url_path='update_subtask')
|
|
def update_subtask(self, request, pk=None):
|
|
task = self.get_object()
|
|
subtask_id = request.data.get("subtask_id")
|
|
subtask_title = request.data.get("subtask_title")
|
|
subtask_status = request.data.get("subtask_status")
|
|
|
|
if not subtask_id:
|
|
return Response({"error": "Subtask ID is required."}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
subtask = SubTask.objects.filter(task=task, id=subtask_id).first()
|
|
if not subtask:
|
|
return Response({"error": "Subtask not found."}, status=status.HTTP_404_NOT_FOUND)
|
|
|
|
if subtask_status not in [True, False]:
|
|
return Response({'error': 'Invalid subtask status. It must be true or false.'}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if subtask_title:
|
|
subtask.title = subtask_title
|
|
if subtask_status not in [None]:
|
|
subtask.status = subtask_status
|
|
|
|
subtask.save()
|
|
return Response(SubTaskSerializer(subtask).data, status=status.HTTP_200_OK)
|
|
|
|
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 |