NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-23 04:22:01 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Enforce object-level permissions for CableTraceView

Browse Source
This commit is contained in:
Jeremy Stretch 2020-05-22 12:05:34 -04:00
parent 1bce148be2
commit 581dc4e070
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/dcim/views.py
View File

@ -1857,15 +1857,21 @@ class CableView(ObjectView):
})
class CableTraceView(PermissionRequiredMixin, View):
class CableTraceView(ObjectPermissionRequiredMixin, View):
"""
Trace a cable path beginning from the given termination.
"""
permission_required = 'dcim.view_cable'
def get(self, request, model, pk):
def dispatch(self, request, *args, **kwargs):
model = kwargs.pop('model')
self.queryset = model.objects.all()
obj = get_object_or_404(model, pk=pk)
return super().dispatch(request, *args, **kwargs)
def get(self, request, pk):
obj = get_object_or_404(self.queryset, pk=pk)
path, split_ends = obj.trace()
total_length = sum(
[entry[1]._abs_length for entry in path if entry[1] and entry[1]._abs_length]