From c3249bb94aa11e34143babf591a0df765bcf0d68 Mon Sep 17 00:00:00 2001 From: Bugra Gedik Date: Tue, 19 Mar 2019 15:57:48 +0300 Subject: [PATCH] Remove the SAWarning: Did not recognize type 'varchar(100)' of column 'name' --- pyhive/sqlalchemy_presto.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pyhive/sqlalchemy_presto.py b/pyhive/sqlalchemy_presto.py index a85ebff1..88b577e0 100644 --- a/pyhive/sqlalchemy_presto.py +++ b/pyhive/sqlalchemy_presto.py @@ -144,8 +144,12 @@ def get_columns(self, connection, table_name, schema=None, **kw): rows = self._get_table_columns(connection, table_name, schema) result = [] for row in rows: + row_type = row.Type + index = row_type.find("(") + if index != -1: + row_type = row_type[:index] try: - coltype = _type_map[row.Type] + coltype = _type_map[row_type] except KeyError: util.warn("Did not recognize type '%s' of column '%s'" % (row.Type, row.Column)) coltype = types.NullType