Add support for interfaces and input

Author: BenoitToulet

.gitignore

@@ -10,4 +10,5 @@ htmlcov/
 *.pyc
 *.swo
 *.swp
-
+venv/
+.vscode/

graphene_mongo/__init__.py

@@ -1,7 +1,13 @@
 from .fields import MongoengineConnectionField
 
-from .types import MongoengineObjectType
+from .types import MongoengineObjectType, MongoengineInputType, MongoengineInterfaceType
 
 __version__ = "0.1.1"
 
-__all__ = ["__version__", "MongoengineObjectType", "MongoengineConnectionField"]
+__all__ = [
+    "__version__",
+    "MongoengineObjectType",
+    "MongoengineInputType",
+    "MongoengineInterfaceType",
+    "MongoengineConnectionField"
+    ]

graphene_mongo/fields.py

@@ -40,12 +40,12 @@ def __init__(self, type, *args, **kwargs):
 
     @property
     def type(self):
-        from .types import MongoengineObjectType
+        from .types import GrapheneMongoengineObjectTypes
 
         _type = super(ConnectionField, self).type
         assert issubclass(
-            _type, MongoengineObjectType
-        ), "MongoengineConnectionField only accepts MongoengineObjectType types"
+            _type, GrapheneMongoengineObjectTypes
+        ), "MongoengineConnectionField only accepts Mongoengine object types"
         assert _type._meta.connection, "The type {} doesn't have a connection".format(
             _type.__name__
         )

graphene_mongo/registry.py

@@ -3,11 +3,12 @@ def __init__(self):
         self._registry = {}
 
     def register(self, cls):
-        from .types import MongoengineObjectType
+        from .types import GrapheneMongoengineObjectTypes
 
         assert issubclass(
-            cls, MongoengineObjectType
-        ), 'Only MongoengineObjectTypes can be registered, received "{}"'.format(
+            cls,
+            GrapheneMongoengineObjectTypes
+            ), 'Only Mongoengine object types can be registered, received "{}"'.format(
             cls.__name__
         )
         assert cls._meta.registry == self, "Registry for a Model have to match."
@@ -22,6 +23,14 @@ def get_type_for_model(self, model):
 
 
 registry = None
+inputs_registry = None
+
+
+def get_inputs_registry():
+    global inputs_registry
+    if not inputs_registry:
+        inputs_registry = Registry()
+    return inputs_registry
 
 
 def get_global_registry():
@@ -31,6 +40,11 @@ def get_global_registry():
     return registry
 
 
+def reset_inputs_registry():
+    global inputs_registry
+    inputs_registry = None
+
+
 def reset_global_registry():
     global registry
     registry = None

graphene_mongo/tests/test_types.py

@@ -157,7 +157,7 @@ class A(TypeSubclassWithBadOptions):
             class Meta:
                 model = Article
 
-    assert "instance of MongoengineObjectTypeOptions" in str(einfo.value)
+    assert "instance of MongoengineGenericObjectTypeOptions" in str(einfo.value)
 
     class TypeSubclass(MongoengineObjectType):
         class Meta: