diff --git a/user_app/views.py b/user_app/views.py index 18133af..0f8fc7c 100644 --- a/user_app/views.py +++ b/user_app/views.py @@ -5,6 +5,7 @@ from .serializers import UserSerializer from .caching import get_cached_user_by_id from rest_framework.permissions import IsAuthenticated from rest_framework import status +from django.core.cache import cache class UserViewSet(viewsets.ModelViewSet): permission_classes = [IsAuthenticated] @@ -22,4 +23,10 @@ class UserViewSet(viewsets.ModelViewSet): return Response({'error': 'User not found.'}, status=status.HTTP_404_NOT_FOUND) serializer = self.get_serializer(user) - return Response(serializer.data) \ No newline at end of file + return Response(serializer.data) + + def destroy(self, request, *args, **kwargs): + user = self.get_object() + user.delete() + cache.delete(f"user_{user.id}") + return Response({'message': 'User deleted successfully.'}, status=status.HTTP_200_OK) \ No newline at end of file