Use singledispatch to convert Django field to GraphQL field

Author: dittos

graphql/contrib/django/__init__.py

@@ -1,7 +1,43 @@
 from graphql.api import Schema
+from singledispatch import singledispatch
 from django.db import models
 
 
+@singledispatch
+def convert_django_field(field, schema):
+    raise NotImplemented
+
+
+@convert_django_field.register(models.CharField)
+def _(field, schema):
+    return schema.Field(schema.String)
+
+
+@convert_django_field.register(models.AutoField)
+def _(field, schema):
+    return schema.Field(schema.ID)
+
+
+@convert_django_field.register(models.IntegerField)
+def _(field, schema):
+    return schema.Field(schema.Int)
+
+
+@convert_django_field.register(models.BigIntegerField)
+def _(field, schema):
+    raise NotImplemented
+
+
+@convert_django_field.register(models.BooleanField)
+def _(field, schema):
+    return schema.Field(schema.Boolean)
+
+
+@convert_django_field.register(models.FloatField)
+def _(field, schema):
+    return schema.Field(schema.Float)
+
+
 class DjangoSchema(Schema):
     def __init__(self):
         super(DjangoSchema, self).__init__()
@@ -14,7 +50,6 @@ def _define_object(self, dct):
                 if field.is_relation:
                     pass  # TODO
                 else:
-                    if isinstance(field, models.CharField):
-                        dct[field.name] = self.Field(self.String)
+                    dct[field.name] = convert_django_field(field, self)
                     # ... TODO
         return super(DjangoSchema, self)._define_object(dct)

setup.py

@@ -49,7 +49,10 @@ def run_tests(self):
     install_requires=[],
     tests_require=['pytest>=2.7.2'],
     extras_require={
-        'django': ['Django>=1.8.0,<1.9'],
+        'django': [
+            'Django>=1.8.0,<1.9',
+            'singledispatch>=3.4.0.3',
+        ],
     },
 
     cmdclass={'test': PyTest},

tests/contrib_django/test_django.py

@@ -17,6 +17,9 @@
 
 class Human(models.Model):
     name = models.CharField()
+    i = models.IntegerField()
+    si = models.SmallIntegerField()
+    b = models.BooleanField()
 
     class Meta:
         app_label = 'graphql'
@@ -34,14 +37,14 @@ class HumanType(gql.ObjectType):
     class QueryRoot(gql.QueryRoot):
         @gql.Field(gql.List(gql.NonNull(HumanType)))
         def humans(self, *args, **kwargs):
-            return [Human(name='hi')]
+            return [Human(id=1, name='hi', i=123, si=1, b=True)]
 
     result = gql.execute('''{
-        humans { name }
+        humans { id, name, i, si, b }
     }''')
     assert not result.errors
     assert result.data == {
         'humans': [
-            {'name': 'hi'}
+            {'id': '1', 'name': 'hi', 'i': 123, 'si': 1, 'b': True}
         ]
     }