@@ -296,7 +296,8 @@ def get_queryset(
296
296
reference_obj = reference_fields [arg_name ].document_type (pk = arg )
297
297
hydrated_references [arg_name ] = reference_obj
298
298
elif arg_name in self .model ._fields_ordered and isinstance (
299
- getattr (self .model , arg_name ), mongoengine .fields .GenericReferenceField
299
+ getattr (self .model , arg_name ),
300
+ mongoengine .fields .GenericReferenceField ,
300
301
):
301
302
try :
302
303
reference_obj = get_document (
@@ -306,7 +307,8 @@ def get_queryset(
306
307
reference_obj = get_document (arg ["_cls" ])(pk = arg ["_ref" ].id )
307
308
hydrated_references [arg_name ] = reference_obj
308
309
elif "__near" in arg_name and isinstance (
309
- getattr (self .model , arg_name .split ("__" )[0 ]), mongoengine .fields .PointField
310
+ getattr (self .model , arg_name .split ("__" )[0 ]),
311
+ mongoengine .fields .PointField ,
310
312
):
311
313
location = args .pop (arg_name , None )
312
314
hydrated_references [arg_name ] = location ["coordinates" ]
@@ -383,7 +385,8 @@ def default_resolver(self, _root, info, required_fields=None, resolved=None, **a
383
385
elif field_name in _root ._fields_ordered and not (
384
386
isinstance (_root ._fields [field_name ].field , mongoengine .EmbeddedDocumentField )
385
387
or isinstance (
386
- _root ._fields [field_name ].field , mongoengine .GenericEmbeddedDocumentField
388
+ _root ._fields [field_name ].field ,
389
+ mongoengine .GenericEmbeddedDocumentField ,
387
390
)
388
391
):
389
392
if getattr (_root , field_name , []) is not None :
@@ -464,13 +467,16 @@ def default_resolver(self, _root, info, required_fields=None, resolved=None, **a
464
467
elif (
465
468
isinstance (getattr (self .model , key ), mongoengine .fields .ReferenceField )
466
469
or isinstance (
467
- getattr (self .model , key ), mongoengine .fields .GenericReferenceField
470
+ getattr (self .model , key ),
471
+ mongoengine .fields .GenericReferenceField ,
468
472
)
469
473
or isinstance (
470
- getattr (self .model , key ), mongoengine .fields .LazyReferenceField
474
+ getattr (self .model , key ),
475
+ mongoengine .fields .LazyReferenceField ,
471
476
)
472
477
or isinstance (
473
- getattr (self .model , key ), mongoengine .fields .CachedReferenceField
478
+ getattr (self .model , key ),
479
+ mongoengine .fields .CachedReferenceField ,
474
480
)
475
481
):
476
482
if not isinstance (args_copy [key ], ObjectId ):
@@ -603,7 +609,7 @@ def chained_resolver(self, resolver, is_partial, root, info, **args):
603
609
connection_fields = [
604
610
field
605
611
for field in self .fields
606
- if type (self .fields [field ]) == MongoengineConnectionField
612
+ if isinstance (self .fields [field ], MongoengineConnectionField )
607
613
]
608
614
609
615
def filter_connection (x ):
0 commit comments