support added for graphene==3.0b7

Author: arunsureshkumar

graphene_mongo/fields.py

@@ -325,21 +325,21 @@ def default_resolver(self, _root, info, required_fields=list(), **args):
                     args["pk__in"] = args["pk__in"][skip:]
                 iterables = self.get_queryset(self.model, info, required_fields, **args)
                 list_length = len(iterables)
-                # if isinstance(info, GraphQLResolveInfo):
-                #     if not info.context:
-                #         info.context = Context()
-                #     info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
+                if isinstance(info, GraphQLResolveInfo):
+                    if not info.context:
+                        info = info._replace(context=Context())
+                    info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
             elif _root is None or args:
                 count = self.get_queryset(self.model, info, required_fields, **args).count()
                 if count != 0:
                     skip, limit, reverse = find_skip_and_limit(first=first, after=after, last=last, before=before,
                                                                count=count)
                     iterables = self.get_queryset(self.model, info, required_fields, skip, limit, reverse, **args)
                     list_length = len(iterables)
-                    # if isinstance(info, GraphQLResolveInfo):
-                    #     if not info.context:
-                    #         info.context = Context()
-                    #     info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
+                    if isinstance(info, GraphQLResolveInfo):
+                        if not info.context:
+                            info = info._replace(context=Context())
+                        info.context.queryset = self.get_queryset(self.model, info, required_fields, **args)
 
         elif _root is not None:
             field_name = to_snake_case(info.field_name)
@@ -374,6 +374,9 @@ def default_resolver(self, _root, info, required_fields=list(), **args):
         return connection
 
     def chained_resolver(self, resolver, is_partial, root, info, **args):
+        for key, value in dict(args).items():
+            if value is None:
+                del args[key]
         required_fields = list()
         for field in self.required_fields:
             if field in self.model._fields_ordered:
@@ -388,10 +391,11 @@ def chained_resolver(self, resolver, is_partial, root, info, **args):
                 for arg_name, arg in args.copy().items():
                     if arg_name not in self.model._fields_ordered + tuple(self.filter_args.keys()):
                         args_copy.pop(arg_name)
-                # if isinstance(info, GraphQLResolveInfo):
-                #     if not info.context:
-                #         info.context = Context()
-                #     info.context.queryset = self.get_queryset(self.model, info, required_fields, **args_copy)
+                if isinstance(info, GraphQLResolveInfo):
+                    if not info.context:
+                        info = info._replace(context=Context())
+                    info.context.queryset = self.get_queryset(self.model, info, required_fields, **args_copy)
+
             # XXX: Filter nested args
             resolved = resolver(root, info, **args)
             if resolved is not None:

requirements.txt

@@ -3,7 +3,7 @@ coveralls==2.1.2; python_version > '3.5'
 flake8==3.7.9
 flake8-per-file-ignores==0.6
 future==0.18.2
-graphene>=2.1.8,<3
+graphene==3.0b7
 iso8601==0.1.13
 mongoengine==0.19.1; python_version <= '3.5'
 mongoengine==0.22.1; python_version > '3.5'