graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-13 17:52:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

first attempt at adding get_queryset

Browse Source
This commit is contained in:
Jason Kraus 2018-10-04 14:19:08 -07:00
parent f76f38ef30
commit eb938da353
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
graphene_django/fields.py
View File

@ -135,7 +135,8 @@ class DjangoConnectionField(ConnectionField):
args["last"] = min(last, max_limit)
iterable = resolver(root, info, **args)
on_resolve = partial(cls.resolve_connection, connection, default_manager, args)
queryset = connection._meta.node.get_queryset(info)
on_resolve = partial(cls.resolve_connection, connection, queryset, args)
if Promise.is_thenable(iterable):
return Promise.resolve(iterable).then(on_resolve)

6
graphene_django/types.py
View File

@ -126,10 +126,14 @@ class DjangoObjectType(ObjectType):
model = root._meta.model._meta.concrete_model
return model == cls._meta.model
@classmethod
def get_queryset(cls, info):
return cls._meta.model.objects
@classmethod
def get_node(cls, info, id):
try:
return cls._meta.model.objects.get(pk=id)
return cls.get_queryset(info).get(pk=id)
except cls._meta.model.DoesNotExist:
return None