Merge pull request #53 from DalgoT4D/minor-bug-fix

quote column name requires warehouse type
DalgoT4D · Feb 12, 2024 · decc577 · decc577
2 parents d584094 + 3616787
commit decc577
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dbt_automation/utils/bigquery.py b/dbt_automation/utils/bigquery.py
@@ -77,7 +77,7 @@ def get_table_data(
         # order
         if order_by:
             query += f"""
-            ORDER BY {quote_columnname(order_by)} {"ASC" if order == 1 else "DESC"}
+            ORDER BY {quote_columnname(order_by, "bigquery")} {"ASC" if order == 1 else "DESC"}
             """
 
         # offset, limit

diff --git a/dbt_automation/utils/postgres.py b/dbt_automation/utils/postgres.py
@@ -107,7 +107,7 @@ def get_table_data(
         # order
         if order_by:
             query += f"""
-            ORDER BY {quote_columnname(order_by)} {"ASC" if order == 1 else "DESC"}
+            ORDER BY {quote_columnname(order_by, "postgres")} {"ASC" if order == 1 else "DESC"}
             """
 
         # offset, limit