add column type info to query runners (re #152, #23)

mozilla · Jul 25, 2018 · 973a71d · 973a71d
1 parent f33416c
commit 973a71d
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 7 deletions.
diff --git a/redash/query_runner/athena.py b/redash/query_runner/athena.py
@@ -156,7 +156,7 @@ def get_schema(self, get_stats=False):
 
         schema = {}
         query = """
-        SELECT table_schema, table_name, column_name
+        SELECT table_schema, table_name, column_name, data_type as column_type
         FROM information_schema.columns
         WHERE table_schema NOT IN ('information_schema')
         """
@@ -170,7 +170,7 @@ def get_schema(self, get_stats=False):
             table_name = '{0}.{1}'.format(row['table_schema'], row['table_name'])
             if table_name not in schema:
                 schema[table_name] = {'name': table_name, 'columns': []}
-            schema[table_name]['columns'].append(row['column_name'])
+            schema[table_name]['columns'].append(row['column_name'] + ' (' + row['column_type'] + ')')
 
         return schema.values()
 

diff --git a/redash/query_runner/mysql.py b/redash/query_runner/mysql.py
@@ -115,7 +115,8 @@ def _get_tables(self, schema):
         query = """
         SELECT col.table_schema,
                col.table_name,
-               col.column_name
+               col.column_name,
+               col.column_type
         FROM `information_schema`.`columns` col
         WHERE col.table_schema NOT IN ('information_schema', 'performance_schema', 'mysql');
         """
@@ -136,7 +137,7 @@ def _get_tables(self, schema):
             if table_name not in schema:
                 schema[table_name] = {'name': table_name, 'columns': []}
 
-            schema[table_name]['columns'].append(row['column_name'])
+            schema[table_name]['columns'].append(row['column_name'] + ' (' + row['column_type'] + ')')
 
         return schema.values()
 

diff --git a/redash/query_runner/pg.py b/redash/query_runner/pg.py
@@ -117,7 +117,7 @@ def _get_definitions(self, schema, query):
             if table_name not in schema:
                 schema[table_name] = {'name': table_name, 'columns': []}
 
-            schema[table_name]['columns'].append(row['column_name'])
+            schema[table_name]['columns'].append(row['column_name'] + ' (' + row['column_type'] + ')')
 
     def _get_tables(self, schema):
         '''
@@ -137,6 +137,7 @@ def _get_tables(self, schema):
         query = """
         SELECT s.nspname as table_schema,
                c.relname as table_name,
+               t.typname as column_type,
                a.attname as column_name
         FROM pg_class c
         JOIN pg_namespace s
@@ -146,6 +147,9 @@ def _get_tables(self, schema):
         ON a.attrelid = c.oid
         AND a.attnum > 0
         AND NOT a.attisdropped
+        JOIN pg_type t
+        ON pg_type t
+        ON c.reltype = t.oid
         WHERE c.relkind IN ('r', 'v', 'm', 'f', 'p')
         """
 

diff --git a/redash/query_runner/presto.py b/redash/query_runner/presto.py
@@ -84,7 +84,7 @@ def __init__(self, configuration):
     def get_schema(self, get_stats=False):
         schema = {}
         query = """
-        SELECT table_schema, table_name, column_name
+        SELECT table_schema, table_name, column_name, data_type as column_type
         FROM information_schema.columns
         WHERE table_schema NOT IN ('pg_catalog', 'information_schema')
         """
@@ -102,7 +102,7 @@ def get_schema(self, get_stats=False):
             if table_name not in schema:
                 schema[table_name] = {'name': table_name, 'columns': []}
 
-            schema[table_name]['columns'].append(row['column_name'])
+            schema[table_name]['columns'].append(row['column_name'] + ' (' + row['column_type'] + ')')
 
         return schema.values()