fix: Rescan all fields on registry register

Author: FloatingGhost

graphene_mongo/converter.py

@@ -89,7 +89,6 @@ def dynamic_type():
         _type = registry.get_type_for_model(model)
         if not _type:
             return None
-
         return Field(_type)
 
     return Dynamic(dynamic_type)

graphene_mongo/registry.py

@@ -11,6 +11,10 @@ def register(self, cls):
         assert cls._meta.registry == self, 'Registry for a Model have to match.'
         self._registry[cls._meta.model] = cls
 
+        # Rescan all fields
+        for model, cls in self._registry.items():
+            cls.rescan_fields()
+
     def get_type_for_model(self, model):
         return self._registry.get(model)
 

graphene_mongo/types.py

@@ -80,7 +80,6 @@ def __init_subclass_with_meta__(cls, model=None, registry=None, skip_registry=Fa
             model, registry, only_fields, exclude_fields
         )
         mongoengine_fields = yank_fields_from_attrs(converted_fields, _as=Field)
-
         if use_connection is None and interfaces:
             use_connection = any((issubclass(interface, Node) for interface in interfaces))
 
@@ -103,6 +102,9 @@ def __init_subclass_with_meta__(cls, model=None, registry=None, skip_registry=Fa
         _meta.fields = mongoengine_fields
         _meta.filter_fields = filter_fields
         _meta.connection = connection
+        # Save them for later
+        _meta.only_fields = only_fields
+        _meta.exclude_fields = exclude_fields
 
         super(MongoengineObjectType, cls).__init_subclass_with_meta__(
             _meta=_meta, interfaces=interfaces, **options
@@ -117,6 +119,16 @@ def __init_subclass_with_meta__(cls, model=None, registry=None, skip_registry=Fa
                 cls._meta.fields.update(mongoengine_fields)
                 registry.register(cls)
 
+    @classmethod
+    def rescan_fields(cls):
+        converted_fields, self_referenced = construct_fields(
+            cls._meta.model, cls._meta.registry, cls._meta.only_fields,
+            cls._meta.exclude_fields
+        )
+        mongoengine_fields = yank_fields_from_attrs(converted_fields, _as=Field)
+        cls._meta.fields.update(mongoengine_fields)
+
+
     # noqa
     @classmethod
     def is_type_of(cls, root, info):