瀏覽代碼

Enforce object-level permissions for RackElevationListView

Jeremy Stretch 6 年之前
父節點
當前提交
781334b615
共有 1 個文件被更改,包括 3 次插入4 次删除
  1. 3 4
      netbox/dcim/views.py

+ 3 - 4
netbox/dcim/views.py

@@ -329,16 +329,15 @@ class RackListView(ObjectListView):
     table = tables.RackDetailTable
 
 
-class RackElevationListView(PermissionRequiredMixin, View):
+class RackElevationListView(ObjectListView):
     """
     Display a set of rack elevations side-by-side.
     """
-    permission_required = 'dcim.view_rack'
+    queryset = Rack.objects.prefetch_related('role')
 
     def get(self, request):
 
-        racks = Rack.objects.prefetch_related('role')
-        racks = filters.RackFilterSet(request.GET, racks).qs
+        racks = filters.RackFilterSet(request.GET, self.queryset).qs
         total_count = racks.count()
 
         # Pagination