@@ -113,7 +113,7 @@ def get_reference_objects(*args, **kwargs):
113
113
document_field = convert_mongoengine_field (document_field , registry )
114
114
document_field_type = document_field .get_type ().type ._meta .name
115
115
required_fields = [to_snake_case (i ) for i in
116
- get_query_fields (args [0 ][3 ][0 ])[document_field_type ].keys ()]
116
+ get_query_fields (args [0 ][3 ][0 ])[document_field_type ].keys ()]
117
117
return document .objects ().no_dereference ().only (* required_fields ).filter (pk__in = args [0 ][1 ])
118
118
else :
119
119
return []
@@ -238,7 +238,7 @@ def reference_resolver(root, *args, **kwargs):
238
238
if document :
239
239
_type = registry .get_type_for_model (field .document_type )
240
240
required_fields = _type ._meta .required_fields .split ("," ) if isinstance (_type ._meta .required_fields ,
241
- str ) else list ()
241
+ str ) else list ()
242
242
return field .document_type .objects ().no_dereference ().only (
243
243
* ((list (set (required_fields + [to_snake_case (i ) for i in get_query_fields (args [0 ]).keys ()]))))).get (
244
244
pk = document .id )
@@ -248,7 +248,7 @@ def cached_reference_resolver(root, *args, **kwargs):
248
248
if field :
249
249
_type = registry .get_type_for_model (field .document_type )
250
250
required_fields = _type ._meta .required_fields .split ("," ) if isinstance (_type ._meta .required_fields ,
251
- str ) else list ()
251
+ str ) else list ()
252
252
return field .document_type .objects ().no_dereference ().only (
253
253
* (list (set (required_fields + [to_snake_case (i ) for i in get_query_fields (args [0 ]).keys ()]))
254
254
)).get (
@@ -280,7 +280,7 @@ def lazy_resolver(root, *args, **kwargs):
280
280
if document :
281
281
_type = registry .get_type_for_model (document .document_type )
282
282
required_fields = _type ._meta .required_fields .split ("," ) if isinstance (_type ._meta .required_fields ,
283
- str ) else list ()
283
+ str ) else list ()
284
284
return document .document_type .objects ().no_dereference ().only (
285
285
* (list (set ((required_fields + [to_snake_case (i ) for i in get_query_fields (args [0 ]).keys ()]))))).get (
286
286
pk = document .pk )
0 commit comments