Add geoalchemy2 Geometry Column support

graphql-python · Aug 15, 2020 · 0b6007f · 0b6007f
1 parent 4b7ceed
commit 0b6007f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/graphene_sqlalchemy/converter.py b/graphene_sqlalchemy/converter.py
@@ -27,6 +27,11 @@ def __getattr__(self, name):
 except ImportError:
     sqlalchemy_utils = DummyImport()
 
+try:
+    from geoalchemy2.elements import WKBElement
+except ImportError:
+    WKBElement = object
+
 
 is_selectin_available = getattr(strategies, 'SelectInLoader', None)
 
@@ -194,6 +199,7 @@ def convert_sqlalchemy_type(type, column, registry=None):
 @convert_sqlalchemy_type.register(sqlalchemy_utils.EmailType)
 @convert_sqlalchemy_type.register(sqlalchemy_utils.URLType)
 @convert_sqlalchemy_type.register(sqlalchemy_utils.IPAddressType)
+@convert_sqlalchemy_type.register(WKBElement)
 def convert_column_to_string(type, column, registry=None):
     return String