From 52a194ae83df20d8c200e7f94b4b8cb0823f2922 Mon Sep 17 00:00:00 2001 From: Paolo Quadri Date: Tue, 16 Jul 2024 14:53:02 +0200 Subject: [PATCH] fix: handle columns with quotes --- snowflake_utils/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/snowflake_utils/models.py b/snowflake_utils/models.py index f105d0d..cc3dacf 100644 --- 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)}" )