NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 01:06:11 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

12591 move check for restrictedqueryset

Browse Source
This commit is contained in:
Arthur 2023-06-16 08:17:25 -07:00
parent 03705a3e61
commit cd5f3cc318
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
netbox/utilities/views.py
View File

@ -79,8 +79,9 @@ class ObjectPermissionRequiredMixin(AccessMixin):
if user.has_perms((permission_required, *self.additional_permissions)):
# Update the view's QuerySet to filter only the permitted objects
action = resolve_permission(permission_required)[1]
self.queryset = self.queryset.restrict(user, action)
if isinstance(self.queryset, RestrictedQuerySet):
action = resolve_permission(permission_required)[1]
self.queryset = self.queryset.restrict(user, action)
return True
@ -94,7 +95,7 @@ class ObjectPermissionRequiredMixin(AccessMixin):
'a base queryset'.format(self.__class__.__name__)
)
if isinstance(self.queryset, RestrictedQuerySet) and not self.has_permission():
if not self.has_permission():
return self.handle_no_permission()
return super().dispatch(request, *args, **kwargs)