Add data type info into schemas of Trino tables (#6348)

The current get_schema method of Trino query runner is missing table columns' data types information, so these data types won't be seen in Queries editing GUI. After the modification, these info are back.

Co-authored-by: V <[email protected]>

redash/query_runner/trino.py

@@ -84,7 +84,7 @@ def type(cls):
 
     def get_schema(self, get_stats=False):
         query = """
-            SELECT table_schema, table_name, column_name
+            SELECT table_schema, table_name, column_name, data_type
             FROM information_schema.columns
             WHERE table_schema NOT IN ('pg_catalog', 'information_schema')
         """
@@ -101,7 +101,8 @@ 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"])
+            column = {"name": row["column_name"], "type": row["data_type"]}
+            schema[table_name]["columns"].append(column)
 
         return list(schema.values())