add query commenting back (#537)

(cherry picked from commit 5958da3)
dbt-labs · Jul 14, 2023 · abb0ff4 · abb0ff4
1 parent 26f0c99
commit abb0ff4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/.changes/unreleased/Fixes-20230713-190412.yaml b/.changes/unreleased/Fixes-20230713-190412.yaml
@@ -0,0 +1,6 @@
+kind: Fixes
+body: re-enable query commenting
+time: 2023-07-13T19:04:12.580073-04:00
+custom:
+  Author: dataders
+  Issue: "536"
diff --git a/dbt/adapters/redshift/connections.py b/dbt/adapters/redshift/connections.py
@@ -337,6 +337,7 @@ def exponential_backoff(attempt: int):
     def execute(
         self, sql: str, auto_begin: bool = False, fetch: bool = False
     ) -> Tuple[AdapterResponse, agate.Table]:
+        sql = self._add_query_comment(sql)
         _, cursor = self.add_query(sql, auto_begin)
         response = self.get_response(cursor)
         if fetch: