add column type info to postgres query_runner

mozilla#152 mozilla#23
alison985 · Jul 28, 2017 · e39fc2e · e39fc2e
1 parent 7e8c1c7
commit e39fc2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/redash/query_runner/pg.py b/redash/query_runner/pg.py
@@ -123,11 +123,11 @@ 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):
         query = """
-        SELECT table_schema, table_name, column_name
+        SELECT table_schema, table_name, column_name, udt_name as column_type
         FROM information_schema.columns
         WHERE table_schema NOT IN ('pg_catalog', 'information_schema');
         """