fix(api): corrct device permissions

!16
nofusscomputing · Jun 1, 2024 · 95dc979 · 95dc979
1 parent 33b1a6c
commit 95dc979
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/app/api/serializers/itam/device.py b/app/api/serializers/itam/device.py
@@ -28,5 +28,6 @@ class Meta:
             'inventorydate',
             'is_global',
             'organization',
+            'slug',
         ]
 
diff --git a/app/api/views/itam/device/detail.py b/app/api/views/itam/device/detail.py
@@ -3,11 +3,16 @@
 from rest_framework import generics
 
 from itam.models.device import Device
-from api.serializers.itam.device import DeviceSerializer
 
+from api.serializers.itam.device import DeviceSerializer
+from api.views.mixin import OrganizationPermissionAPI
 
 class List(generics.ListCreateAPIView):
-    permission_required = 'itam.view_device'
+
+    permission_classes = [
+        OrganizationPermissionAPI
+    ]
+
     queryset = Device.objects.all()
     serializer_class = DeviceSerializer
 
@@ -16,8 +21,13 @@ def get_view_name(self):
 
 
 class Detail(generics.RetrieveUpdateDestroyAPIView):
-    permission_required = 'itam.view_device'
+
+    permission_classes = [
+        OrganizationPermissionAPI
+    ]
+
     queryset = Device.objects.all()
+
     serializer_class = DeviceSerializer
 
     def get_view_name(self):