Use RestrictedQuerySet for ConfigRevision

This commit is contained in:
Jeremy Stretch 2023-06-22 10:31:49 -04:00
parent 8658bfc934
commit 85657918ed
2 changed files with 4 additions and 4 deletions

View File

@ -612,6 +612,8 @@ class ConfigRevision(models.Model):
verbose_name='Configuration data' verbose_name='Configuration data'
) )
objects = RestrictedQuerySet.as_manager()
class Meta: class Meta:
ordering = ['-created'] ordering = ['-created']

View File

@ -5,7 +5,6 @@ from django.urls.exceptions import NoReverseMatch
from netbox.registry import registry from netbox.registry import registry
from .permissions import resolve_permission from .permissions import resolve_permission
from .querysets import RestrictedQuerySet
__all__ = ( __all__ = (
'ContentTypePermissionRequiredMixin', 'ContentTypePermissionRequiredMixin',
@ -79,9 +78,8 @@ class ObjectPermissionRequiredMixin(AccessMixin):
if user.has_perms((permission_required, *self.additional_permissions)): if user.has_perms((permission_required, *self.additional_permissions)):
# Update the view's QuerySet to filter only the permitted objects # Update the view's QuerySet to filter only the permitted objects
if isinstance(self.queryset, RestrictedQuerySet): action = resolve_permission(permission_required)[1]
action = resolve_permission(permission_required)[1] self.queryset = self.queryset.restrict(user, action)
self.queryset = self.queryset.restrict(user, action)
return True return True