fix: always quote partition by

remoteoss · Oct 8, 2024 · 3dc6b03 · 3dc6b03
1 parent 29a5c75
commit 3dc6b03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/snowflake_utils/models/table.py b/snowflake_utils/models/table.py
@@ -285,9 +285,9 @@ def qualify(
         primary_keys: list[str],
         replication_keys: list[str] | None,
     ) -> None:
-        qualify_partition = ",".join(primary_keys)
+        qualify_partition = ",".join(f'"{c.upper()}"' for c in primary_keys)
         qualify_order = ",".join(
-            f"{c} desc" for c in (replication_keys or primary_keys)
+            f'"{c.upper()}" desc' for c in (replication_keys or primary_keys)
         )
         logging.debug(
             f"Adding QUALIFY to table {self.fqn} on PARTITION {qualify_partition} ORDERED BY {qualify_order}"