Fix evaluation of RestrictedQuerySets

This commit is contained in:
Jeremy Stretch
2020-06-29 11:35:13 -04:00
parent 10e6b6ca66
commit eb45ad600e
2 changed files with 11 additions and 6 deletions

View File

@@ -1,5 +1,5 @@
from django.conf import settings
from django.db.models import Count
from django.db.models import Count, Prefetch
from django.shortcuts import get_object_or_404
from django_pglocks import advisory_lock
from drf_yasg.utils import swagger_auto_schema
@@ -270,6 +270,9 @@ class VLANViewSet(CustomFieldModelViewSet):
#
class ServiceViewSet(ModelViewSet):
queryset = Service.objects.prefetch_related('device').prefetch_related('tags')
queryset = Service.objects.prefetch_related(
Prefetch('ipaddresses', queryset=IPAddress.objects.unrestricted()),
'device', 'virtual_machine', 'tags'
)
serializer_class = serializers.ServiceSerializer
filterset_class = filters.ServiceFilterSet