Skip to content

Commit 3f4eee1

Browse files
authoredAug 19, 2024
fix: remove leading whitespaces on post-hook operations (#705)
1 parent fbe3bc4 commit 3f4eee1

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed
 

‎dbt/adapters/athena/query_headers.py

+1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ def add(self, sql: str) -> str:
2525

2626
# alter or vacuum statements don't seem to support properly query comments
2727
# let's just exclude them
28+
sql = sql.lstrip()
2829
if any(sql.lower().startswith(keyword) for keyword in ["alter", "drop", "optimize", "vacuum", "msck"]):
2930
return sql
3031

‎tests/unit/test_query_headers.py

+12
Original file line numberDiff line numberDiff line change
@@ -74,3 +74,15 @@ def test_no_query_comment_on_msck(self):
7474
self.query_header.comment.append = True
7575
sql = "MSCK REPAIR TABLE"
7676
assert self.query_header.add(sql) == sql
77+
78+
def test_no_query_comment_on_vacuum_with_leading_whitespaces(self):
79+
self.query_header.comment.query_comment = "executed by dbt"
80+
self.query_header.comment.append = True
81+
sql = " VACUUM table"
82+
assert self.query_header.add(sql) == "VACUUM table"
83+
84+
def test_no_query_comment_on_vacuum_with_lowercase(self):
85+
self.query_header.comment.query_comment = "executed by dbt"
86+
self.query_header.comment.append = True
87+
sql = "vacuum table"
88+
assert self.query_header.add(sql) == sql

0 commit comments

Comments
 (0)
Please sign in to comment.