diff --git a/graphene_sqlalchemy/converter.py b/graphene_sqlalchemy/converter.py index 04061801..1b304d74 100644 --- a/graphene_sqlalchemy/converter.py +++ b/graphene_sqlalchemy/converter.py @@ -4,8 +4,8 @@ from sqlalchemy.dialects import postgresql from sqlalchemy.orm import interfaces, strategies -from graphene import (ID, Boolean, Dynamic, Enum, Field, Float, Int, List, - String) +from graphene import (ID, UUID, Boolean, Dynamic, Enum, Field, Float, Int, + List, String) from graphene.types.json import JSONString from .batching import get_batch_resolver @@ -260,3 +260,8 @@ def convert_json_to_string(type, column, registry=None): @convert_sqlalchemy_type.register(JSONType) def convert_json_type_to_string(type, column, registry=None): return JSONString + + +@convert_sqlalchemy_type.register(postgresql.UUID) +def convert_column_to_uuid(type, column, registry=None): + return UUID diff --git a/graphene_sqlalchemy/tests/test_converter.py b/graphene_sqlalchemy/tests/test_converter.py index 57c43058..d6f143c9 100644 --- a/graphene_sqlalchemy/tests/test_converter.py +++ b/graphene_sqlalchemy/tests/test_converter.py @@ -287,7 +287,7 @@ class Meta: def test_should_postgresql_uuid_convert(): - assert get_field(postgresql.UUID()).type == graphene.String + assert get_field(postgresql.UUID()).type == graphene.UUID def test_should_postgresql_enum_convert():