mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Transition ObjectEditView to use ObjectPermissionRequiredMixin

Browse Source
This commit is contained in:
Jeremy Stretch
2020-05-21 13:59:19 -04:00
parent 993ee8c900
commit 406b076b95
15 changed files with 99 additions and 283 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
netbox/utilities/views.py
View File

@@ -278,7 +278,7 @@ class ObjectListView(ObjectPermissionRequiredMixin, View):
return {}
class ObjectEditView(GetReturnURLMixin, View):
class ObjectEditView(GetReturnURLMixin, ObjectPermissionRequiredMixin, View):
"""
Create or edit a single object.
@@ -290,6 +290,12 @@ class ObjectEditView(GetReturnURLMixin, View):
model_form = None
template_name = 'utilities/obj_edit.html'
def get_required_permission(self):
# Determine required permission based on whether we are editing an existing object
if self.obj.pk is None:
return get_permission_for_model(self.queryset.model, 'add')
return get_permission_for_model(self.queryset.model, 'change')
def get_object(self, kwargs):
# Look up an existing object by slug or PK, if provided.
if 'slug' in kwargs: