Skip to content

Commit 0021d9d

Browse files
[Bug Fix] Default resolver for reference fields
1 parent 20255ab commit 0021d9d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

graphene_mongo/converter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def reference_resolver(root, *args, **kwargs):
248248
required = False
249249
if field.db_field is not None:
250250
required = field.required
251-
resolver_function = getattr(_union, "resolve_" + field.db_field, None)
251+
resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
252252
if resolver_function and callable(resolver_function):
253253
field_resolver = resolver_function
254254
return graphene.Field(_union, resolver=field_resolver if field_resolver else reference_resolver,
@@ -302,7 +302,7 @@ def dynamic_type():
302302
required = False
303303
if field.db_field is not None:
304304
required = field.required
305-
resolver_function = getattr(_type, "resolve_" + field.db_field, None)
305+
resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
306306
if resolver_function and callable(resolver_function):
307307
field_resolver = resolver_function
308308
if isinstance(field, mongoengine.ReferenceField):
@@ -341,7 +341,7 @@ def dynamic_type():
341341
required = False
342342
if field.db_field is not None:
343343
required = field.required
344-
resolver_function = getattr(_type, "resolve_" + field.db_field, None)
344+
resolver_function = getattr(registry.get_type_for_model(field.owner_document), "resolve_" + field.db_field, None)
345345
if resolver_function and callable(resolver_function):
346346
field_resolver = resolver_function
347347
return graphene.Field(

0 commit comments

Comments
 (0)