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

11171 fix graphql related models query

Browse Source
This commit is contained in:
Arthur 2022-12-13 08:34:12 -08:00
parent ad78f9e075
commit 152247b0e7
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
netbox/netbox/graphql/fields.py
View File

@ -1,3 +1,5 @@
from django.db.models.manager import Manager
from django.db.models.query import QuerySet
from functools import partial from functools import partial
import graphene import graphene
@ -36,6 +38,12 @@ class ObjectField(graphene.Field):
return partial(self.object_resolver, self._type) return partial(self.object_resolver, self._type)
def maybe_queryset(value):
if isinstance(value, Manager):
value = value.get_queryset()
return value
class ObjectListField(DjangoListField): class ObjectListField(DjangoListField):
""" """
Retrieve a list of objects, optionally filtered by one or more FilterSet filters. Retrieve a list of objects, optionally filtered by one or more FilterSet filters.
@ -55,7 +63,14 @@ class ObjectListField(DjangoListField):
@staticmethod @staticmethod
def list_resolver(django_object_type, resolver, default_manager, root, info, **args): def list_resolver(django_object_type, resolver, default_manager, root, info, **args):
# Get the QuerySet from the object type # Get the QuerySet from the object type
queryset = django_object_type.get_queryset(default_manager, info) # queryset = django_object_type.get_queryset(default_manager, info)
queryset = maybe_queryset(resolver(root, info, **args))
if queryset is None:
queryset = maybe_queryset(default_manager)
if isinstance(queryset, QuerySet):
# Pass queryset to the DjangoObjectType get_queryset method
queryset = maybe_queryset(django_object_type.get_queryset(queryset, info))
# Instantiate and apply the FilterSet, if defined # Instantiate and apply the FilterSet, if defined
filterset_class = django_object_type._meta.filterset_class filterset_class = django_object_type._meta.filterset_class