Wrap columns_in_query query in select statement (ClickHouse#222)

* Wrap columns_in_query query in select statement * formatting
Savid · Jan 17, 2024 · 30580d2 · 30580d2
1 parent 297aa80
commit 30580d2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dbt/adapters/clickhouse/httpclient.py b/dbt/adapters/clickhouse/httpclient.py
@@ -25,7 +25,7 @@ def command(self, sql, **kwargs):
 
     def columns_in_query(self, sql: str, **kwargs) -> List[ClickHouseColumn]:
         try:
-            query_result = self._client.query(f'{sql} LIMIT 0', **kwargs)
+            query_result = self._client.query(f"SELECT * FROM ({sql}) LIMIT 0", **kwargs)
             return [
                 ClickHouseColumn.create(name, ch_type.name)
                 for name, ch_type in zip(query_result.column_names, query_result.column_types)

diff --git a/dbt/adapters/clickhouse/nativeclient.py b/dbt/adapters/clickhouse/nativeclient.py
@@ -34,7 +34,9 @@ def command(self, sql, **kwargs):
 
     def columns_in_query(self, sql: str, **kwargs) -> List[ClickHouseColumn]:
         try:
-            _, columns = self._client.execute(f'{sql} LIMIT 0', with_column_types=True)
+            _, columns = self._client.execute(
+                f"SELECT * FROM ({sql}) LIMIT 0", with_column_types=True
+            )
             return [ClickHouseColumn.create(column[0], column[1]) for column in columns]
         except clickhouse_driver.errors.Error as ex:
             raise DbtDatabaseError(str(ex).strip()) from ex