Added Facility Delete Endpoint

ohcnetwork · May 13, 2021 · 872b08a · 872b08a
1 parent 8debd3c
commit 872b08a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/care/facility/api/viewsets/facility.py b/care/facility/api/viewsets/facility.py
@@ -70,6 +70,7 @@ class FacilityViewSet(
     mixins.ListModelMixin,
     mixins.RetrieveModelMixin,
     mixins.UpdateModelMixin,
+    mixins.DestroyModelMixin,
     viewsets.GenericViewSet,
 ):
     """Viewset for facility CRUD operations."""
@@ -96,6 +97,11 @@ def get_serializer_class(self):
         else:
             return FacilitySerializer
 
+    def destroy(self, request, *args, **kwargs):
+        if request.user.is_superuser or request.user.user_type >= User.TYPE_VALUE_MAP["DistrictLabAdmin"]:
+            return super().destroy(request, *args, **kwargs)
+        return Response({"permission": "denied"}, status=status.HTTP_403_FORBIDDEN)
+
     def list(self, request, *args, **kwargs):
         """
         Facility List