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
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