fix: handle columns with quotes

remoteoss · Jul 16, 2024 · 52a194a · 52a194a
1 parent 9f58dc5
commit 52a194a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/snowflake_utils/models.py b/snowflake_utils/models.py
@@ -412,12 +412,12 @@ def _set_tag(
         self, cursor: SnowflakeCursor, column: str, tag_name: str, tag_value: str
     ) -> None:
         cursor.execute(
-            f"ALTER TABLE {self.fqn} MODIFY COLUMN {column} SET TAG {governance_settings.fqn(tag_name)} = '{tag_value}'"
+            f"ALTER TABLE {self.fqn} MODIFY COLUMN "{column.upper()}" SET TAG {governance_settings.fqn(tag_name)} = '{tag_value}'"
         )
 
     def _unset_tag(self, cursor: SnowflakeCursor, column: str, tag: str):
         cursor.execute(
-            f"ALTER TABLE {self.fqn} MODIFY COLUMN {column} UNSET TAG {governance_settings.fqn(tag)}"
+            f"ALTER TABLE {self.fqn} MODIFY COLUMN "{column.upper()}" UNSET TAG {governance_settings.fqn(tag)}"
         )