Map pg's uuid to graphene's uuid

Author: ticosax

graphene_sqlalchemy/converter.py

@@ -4,7 +4,7 @@
 from sqlalchemy.orm import interfaces
 
 from graphene import (ID, Boolean, Dynamic, Enum, Field, Float, Int, List,
-                      String)
+                      String, UUID)
 from graphene.types.json import JSONString
 
 from .fields import createConnectionField
@@ -196,3 +196,10 @@ def convert_json_type_to_string(type, column, registry=None):
     return JSONString(
         description=get_column_doc(column), required=not (is_column_nullable(column))
     )
+
+
+@convert_sqlalchemy_type.register(postgresql.UUID)
+def convert_column_to_uuid(type, column, registry=None):
+    return UUID(
+        description=get_column_doc(column), required=not (is_column_nullable(column))
+    )

graphene_sqlalchemy/tests/test_converter.py

@@ -264,7 +264,7 @@ class Meta:
 
 
 def test_should_postgresql_uuid_convert():
-    assert_column_conversion(postgresql.UUID(), graphene.String)
+    assert_column_conversion(postgresql.UUID(), graphene.UUID)
 
 
 def test_should_postgresql_enum_convert():