Map pg's uuid to graphene's uuid

Author: ticosax

graphene_sqlalchemy/converter.py

@@ -3,7 +3,7 @@
 from sqlalchemy.dialects import postgresql
 from sqlalchemy.orm import interfaces
 
-from graphene import ID, Boolean, Dynamic, Enum, Field, Float, Int, List, String
+from graphene import ID, Boolean, Dynamic, Enum, Field, Float, Int, List, String, UUID
 from graphene.types.json import JSONString
 
 from .fields import createConnectionField
@@ -195,3 +195,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

@@ -266,7 +266,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():