NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-12-21 21:02:23 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Move restrict_queryset() function to RestrictedQuerySet

Browse Source
This commit is contained in:
Jeremy Stretch
2020-05-29 15:09:08 -04:00
parent aefeb711a7
commit 10ab884671
4 changed files with 34 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
netbox/utilities/api.py
View File

@@ -16,7 +16,6 @@ from rest_framework.serializers import Field, ModelSerializer, ValidationError
from rest_framework.viewsets import ModelViewSet as _ModelViewSet
from netbox.api import TokenPermissions
from utilities.permissions import restrict_queryset
from .utils import dict_to_filter_params, dynamic_import
@@ -339,8 +338,8 @@ class ModelViewSet(_ModelViewSet):
}
permission_required = TokenPermissions.perms_map[request.method][0] % kwargs
# Update the view's QuerySet to filter only the permitted objects
self.queryset = restrict_queryset(self.queryset, request.user, permission_required)
# Restrict the view's QuerySet to allow only the permitted objects
self.queryset = self.queryset.restrict(request.user, permission_required)
def dispatch(self, request, *args, **kwargs):
logger = logging.getLogger('netbox.api.views.ModelViewSet')