DocManagerBackend/docmanager_backend/accounts/views.py

from rest_framework import generics
from .serializers import CustomUserSerializer, CustomUserUpdateSerializer
from rest_framework.permissions import IsAuthenticated
from accounts.permissions import IsAdmin
from .models import CustomUser


class CustomUserDeleteView(generics.DestroyAPIView):
    """
    Used by admin to delete users. Accepts the user id as a URL parameter
    """

    http_method_names = ["delete"]
    serializer_class = CustomUserSerializer
    queryset = CustomUser.objects.all()
    permission_classes = [IsAuthenticated, IsAdmin]


class CustomUserUpdateView(generics.UpdateAPIView):
    """
    Used by staff to upload documents.
    """

    http_method_names = ["patch"]
    serializer_class = CustomUserUpdateSerializer
    queryset = CustomUser.objects.all()
    permission_classes = [IsAuthenticated, IsAdmin]