Delegate validation to identifier function in snowflake

Snowflake-Labs · Apr 3, 2024 · 7456bc3 · 7456bc3
1 parent ff59dc6
commit 7456bc3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/schemachange/cli.py b/schemachange/cli.py
@@ -267,10 +267,10 @@ class SnowflakeSchemachangeSession:
         + "'{script_description}','{script_name}','{script_type}','{checksum}',{execution_time},"
         + "'{status}','{user}',CURRENT_TIMESTAMP);"
     )
-    _q_set_sess_role = "USE ROLE {role};"
-    _q_set_sess_database = "USE DATABASE {database};"
-    _q_set_sess_schema = "USE SCHEMA {schema};"
-    _q_set_sess_warehouse = "USE WAREHOUSE {warehouse};"
+    _q_set_sess_role = "USE ROLE IDENTIFIER({role});"
+    _q_set_sess_database = "USE DATABASE IDENTIFIER({database});"
+    _q_set_sess_schema = "USE SCHEMA IDENTIFIER({schema});"
+    _q_set_sess_warehouse = "USE WAREHOUSE IDENTIFIER({warehouse});"
     # endregion Query Templates
 
     def __init__(self, config):