Enforce object-level permissions for RackElevationListView

This commit is contained in:
Jeremy Stretch 2020-05-22 11:51:04 -04:00
parent 5282ae2250
commit 781334b615

View File

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