removed camel_to_snake custom function and used to_snake_case from graphene.utils.str_converters

re-indented documentation.

Author: Arun S Kumar

graphene_mongo/converter.py

@@ -3,10 +3,10 @@
 import uuid
 
 from graphene.types.json import JSONString
+from graphene.utils.str_converters import to_snake_case
 from mongoengine.base import get_document
-
 from . import advanced_types
-from .utils import import_single_dispatch, get_field_description, get_query_fields, camel_to_snake
+from .utils import import_single_dispatch, get_field_description, get_query_fields
 
 singledispatch = import_single_dispatch()
 
@@ -172,11 +172,11 @@ def convert_field_to_dynamic(field, registry=None):
 
     def reference_resolver(root, *args, **kwargs):
         document = getattr(root, field.name or field.db_name)
-        only_fields = [camel_to_snake(i) for i in get_query_fields(args[0]).keys()]
+        only_fields = [to_snake_case(i) for i in get_query_fields(args[0]).keys()]
         return field.document_type.objects().no_dereference().only(*only_fields).get(pk=document.id)
 
     def cached_reference_resolver(root, *args, **kwargs):
-        only_fields = [camel_to_snake(i) for i in get_query_fields(args[0]).keys()]
+        only_fields = [to_snake_case(i) for i in get_query_fields(args[0]).keys()]
         return field.document_type.objects().no_dereference().only(*only_fields).get(
             pk=getattr(root, field.name or field.db_name))
 
@@ -202,7 +202,7 @@ def convert_lazy_field_to_dynamic(field, registry=None):
 
     def lazy_resolver(root, *args, **kwargs):
         document = getattr(root, field.name or field.db_name)
-        only_fields = [camel_to_snake(i) for i in get_query_fields(args[0]).keys()]
+        only_fields = [to_snake_case(i) for i in get_query_fields(args[0]).keys()]
         return document.document_type.objects().no_dereference().only(*only_fields).get(pk=document.pk)
 
     def dynamic_type():

graphene_mongo/fields.py

@@ -5,6 +5,7 @@
 
 import graphene
 import mongoengine
+from graphene.utils.str_converters import to_snake_case
 from promise import Promise
 from graphql_relay import from_global_id
 from graphene.relay import ConnectionField
@@ -21,7 +22,7 @@
 )
 from .converter import convert_mongoengine_field, MongoEngineConversionError
 from .registry import get_global_registry
-from .utils import get_model_reference_fields, get_node_from_global_id, get_query_fields, camel_to_snake
+from .utils import get_model_reference_fields, get_node_from_global_id, get_query_fields
 
 
 class MongoengineConnectionField(ConnectionField):
@@ -242,8 +243,8 @@ def default_resolver(self, _root, info, only_fields=list(), **args):
     def chained_resolver(self, resolver, is_partial, root, info, **args):
         only_fields = list()
         for field in get_query_fields(info):
-            if camel_to_snake(field) in self.model._fields_ordered:
-                only_fields.append(camel_to_snake(field))
+            if to_snake_case(field) in self.model._fields_ordered:
+                only_fields.append(to_snake_case(field))
         if not bool(args) or not is_partial:
             # XXX: Filter nested args
             resolved = resolver(root, info, **args)

graphene_mongo/utils.py

@@ -113,12 +113,15 @@ def collect_query_fields(node, fragments):
 
     Returns:
         A dict mapping each field found, along with their sub fields.
-
-        {'name': {},
-         'image': {'id': {},
-                                   'name': {},
-                                   'description': {}},
-         'slug': {}}
+        {
+            'name': {},
+            'image': {
+                        'id': {},
+                        'name': {},
+                        'description': {}
+                    },
+            'slug': {}
+        }
     """
 
     field = {}
@@ -156,7 +159,3 @@ def get_query_fields(info):
     if "edges" in query:
         return query["edges"]["node"].keys()
     return query
-
-
-def camel_to_snake(field):
-    return ''.join(['_' + c.lower() if c.isupper() else c for c in field]).lstrip('_')