Allow postgres client to read tables with UUID, JSON, JSONB types.

costrouc · Sep 26, 2019 · b514069 · b514069
1 parent be78e26
commit b514069
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/ibis/expr/datatypes.py b/ibis/expr/datatypes.py
@@ -129,6 +129,9 @@ def _literal_value_hash_key(self, value) -> int:
 
 
 class Any(DataType):
+    scalar = ir.AnyScalar
+    column = ir.AnyColumn
+
     __slots__ = ()
 
 

diff --git a/ibis/sql/alchemy.py b/ibis/sql/alchemy.py
@@ -136,6 +136,21 @@ def sa_double(_, satype, nullable=True):
     return dt.Double(nullable=nullable)
 
 
+@dt.dtype.register(PostgreSQLDialect, sa.dialects.postgresql.UUID)
+def sa_uuid(_, satype, nullable=True):
+    return dt.Any(nullable=nullable)
+
+
+@dt.dtype.register(PostgreSQLDialect, sa.dialects.postgresql.JSON)
+def sa_json(_, satype, nullable=True):
+    return dt.Any(nullable=nullable)
+
+
+@dt.dtype.register(PostgreSQLDialect, sa.dialects.postgresql.JSONB)
+def sa_jsonb(_, satype, nullable=True):
+    return dt.Any(nullable=nullable)
+
+
 if geospatial_supported:
 
     @dt.dtype.register(SQLAlchemyDialect, (ga.Geometry, ga.types._GISType))