From 64839d3e5cd0e9c7539659df0fe771464a0c10b7 Mon Sep 17 00:00:00 2001 From: Nicolas Delaby Date: Mon, 8 Apr 2019 13:54:48 +0200 Subject: [PATCH] Map pg's uuid to graphene's uuid --- graphene_sqlalchemy/converter.py | 9 +++++++-- graphene_sqlalchemy/tests/test_converter.py | 2 +- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/graphene_sqlalchemy/converter.py b/graphene_sqlalchemy/converter.py index 2bf32f5e..c2dbdc6c 100644 --- a/graphene_sqlalchemy/converter.py +++ b/graphene_sqlalchemy/converter.py @@ -3,8 +3,8 @@ 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, UUID, Boolean, Dynamic, Enum, Field, Float, Int, + List, String) from graphene.types.json import JSONString from .enums import enum_for_sa_enum @@ -191,3 +191,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 f255350d..219da7e5 100644 --- a/graphene_sqlalchemy/tests/test_converter.py +++ b/graphene_sqlalchemy/tests/test_converter.py @@ -252,7 +252,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():