From 3d38b39c09358e6b700327ee2b39ccf44896b4a6 Mon Sep 17 00:00:00 2001
From: Padraig O'Sullivan <osullivan.padraig@gmail.com>
Date: Thu, 31 Oct 2024 15:38:55 -0400
Subject: [PATCH] Update trino__get_columns_in_relation to use
 information_schema.columns

---
 dbt/include/trino/macros/adapters.sql | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/dbt/include/trino/macros/adapters.sql b/dbt/include/trino/macros/adapters.sql
index 98bc4161..612cfc06 100644
--- a/dbt/include/trino/macros/adapters.sql
+++ b/dbt/include/trino/macros/adapters.sql
@@ -5,7 +5,12 @@
 
 {% macro trino__get_columns_in_relation(relation) -%}
   {%- set sql -%}
-    describe {{ relation }}
+    select column_name, data_type
+    from {{ relation.information_schema() }}.columns
+    where
+      table_catalog = '{{ relation.database | lower }}'
+      and table_schema = '{{ relation.schema | lower }}'
+      and table_name = '{{ relation.identifier  | lower}}'
   {%- endset -%}
   {%- set result = run_query(sql) -%}
 
@@ -20,7 +25,7 @@
 
   {% set columns = [] %}
   {% for row in result %}
-    {% do columns.append(api.Column.from_description(row['Column'].lower(), row['Type'])) %}
+    {% do columns.append(api.Column.from_description(row['column_name'].lower(), row['data_type'])) %}
   {% endfor %}
   {% do return(columns) %}
 {% endmacro %}