add type conversion sqlalchemy_utils JSONType

Author: sjhewitt

graphene_sqlalchemy/converter.py

@@ -11,14 +11,17 @@
 from .fields import SQLAlchemyConnectionField
 
 try:
-    from sqlalchemy_utils import ChoiceType, ScalarListType
+    from sqlalchemy_utils import ChoiceType, JSONType, ScalarListType
 except ImportError:
     class ChoiceType(object):
         pass
 
     class ScalarListType(object):
         pass
 
+    class JSONType(object):
+        pass
+
 
 def convert_sqlalchemy_relationship(relationship, registry):
     direction = relationship.direction
@@ -133,3 +136,8 @@ def convert_postgres_array_to_list(type, column, registry=None):
 @convert_sqlalchemy_type.register(postgresql.JSONB)
 def convert_json_to_string(type, column, registry=None):
     return JSONString(description=column.doc, required=not(column.nullable))
+
+
+@convert_sqlalchemy_type.register(JSONType)
+def convert_json_type_to_string(type, column, registry=None):
+    return JSONString(description=column.doc, required=not(column.nullable))

graphene_sqlalchemy/tests/test_converter.py

@@ -3,7 +3,7 @@
 from sqlalchemy.dialects import postgresql
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import composite
-from sqlalchemy_utils import ChoiceType, ScalarListType
+from sqlalchemy_utils import ChoiceType, JSONType, ScalarListType
 
 import graphene
 from graphene.relay import Node
@@ -127,6 +127,10 @@ def test_should_scalar_list_convert_list():
     assert_column_conversion(ScalarListType(), graphene.List)
 
 
+def test_should_jsontype_convert_jsonstring():
+    assert_column_conversion(JSONType(), JSONString)
+
+
 def test_should_manytomany_convert_connectionorlist():
     registry = Registry()
     dynamic_field = convert_sqlalchemy_relationship(Reporter.pets.property, registry)