graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-02-22 14:30:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Clean up wrapped list resolver

Browse Source
This commit is contained in:
Josh Warwick 2023-05-16 19:45:02 +01:00
parent 84ba7d775a
commit 848536ee3a
1 changed files with 13 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
graphene_django/fields.py
View File

@ -49,10 +49,14 @@ class DjangoListField(Field):
def get_manager(self): def get_manager(self):
return self.model._default_manager return self.model._default_manager
@staticmethod @classmethod
def list_resolver( def list_resolver(
django_object_type, resolver, default_manager, root, info, **args cls, django_object_type, resolver, default_manager, root, info, **args
): ):
if is_running_async():
if is_sync_function(resolver):
resolver = sync_to_async(resolver)
iterable = resolver(root, info, **args) iterable = resolver(root, info, **args)
if info.is_awaitable(iterable): if info.is_awaitable(iterable):
@ -82,7 +86,7 @@ class DjangoListField(Field):
# Pass queryset to the DjangoObjectType get_queryset method # Pass queryset to the DjangoObjectType get_queryset method
queryset = maybe_queryset(django_object_type.get_queryset(queryset, info)) queryset = maybe_queryset(django_object_type.get_queryset(queryset, info))
return queryset return list(queryset)
def wrap_resolve(self, parent_resolver): def wrap_resolve(self, parent_resolver):
resolver = super().wrap_resolve(parent_resolver) resolver = super().wrap_resolve(parent_resolver)
@ -90,28 +94,13 @@ class DjangoListField(Field):
if isinstance(_type, NonNull): if isinstance(_type, NonNull):
_type = _type.of_type _type = _type.of_type
django_object_type = _type.of_type.of_type django_object_type = _type.of_type.of_type
if not is_running_async():
return partial( return partial(
self.list_resolver, django_object_type, resolver, self.get_manager() self.list_resolver,
)
else:
if is_sync_function(resolver):
async_resolver = sync_to_async(resolver)
## This is needed because our middleware can't detect the resolver as async when we returns partial[couroutine]
async def wrapped_resolver(root, info, **args):
return await self.list_resolver(
django_object_type, django_object_type,
async_resolver, resolver,
self.get_manager(), self.get_manager(),
root,
info,
**args
) )
return wrapped_resolver
class DjangoConnectionField(ConnectionField): class DjangoConnectionField(ConnectionField):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):