fix column order mismatch using explicit copy into selects

MeltanoLabs · Sep 12, 2023 · fafa15a · fafa15a
1 parent 3ab7f03
commit fafa15a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/target_snowflake/connector.py b/target_snowflake/connector.py
@@ -349,10 +349,11 @@ def _get_merge_from_stage_statement(
     def _get_copy_statement(self, full_table_name, schema, sync_id, file_format):
         """Get Snowflake COPY statement."""
         formatter = SnowflakeIdentifierPreparer(SnowflakeDialect())
+        formatted_properties = ", ".join([formatter.format_collation(col) for col in schema["properties"].keys()])
         column_selections = self._get_column_selections(schema, formatter)
         return (
             text(
-                f"copy into {full_table_name} from "
+                f"copy into {full_table_name} ({formatted_properties}) from "
                 + f"(select {', '.join(column_selections)} from "
                 + f"'@~/target-snowflake/{sync_id}')"
                 + f"file_format = (format_name='{file_format}')"