snowflake-data-validation 1.3.3__tar.gz → 1.3.4__tar.gz

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-data-validation
-Version: 1.3.3
+Version: 1.3.4
 Summary: Snowflake Data Validation
 Project-URL: Bug Tracker, https://github.com/snowflake-eng/migrations-data-validation/issues
 Project-URL: Source code, https://github.com/snowflake-eng/migrations-data-validation/
@@ -186,7 +186,7 @@ This project is licensed under the Snowflake Conversion Software Terms. See the
 ## 🆘 Support
 
 - **Documentation**: [Full documentation](https://docs.snowflake.com/en/migrations/snowconvert-docs/data-validation-cli/CLI_QUICK_REFERENCE)
-- **Issues**: [GitHub Issues](https://github.com/snowflakedb/migrations-data-validation/issues)
+- **Issues**: [GitHub Issues](https://github.com/snowflake-eng/migrations-data-validation/issues)
 
 ---
 

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/README.md

@@ -110,7 +110,7 @@ This project is licensed under the Snowflake Conversion Software Terms. See the
 ## 🆘 Support
 
 - **Documentation**: [Full documentation](https://docs.snowflake.com/en/migrations/snowconvert-docs/data-validation-cli/CLI_QUICK_REFERENCE)
-- **Issues**: [GitHub Issues](https://github.com/snowflakedb/migrations-data-validation/issues)
+- **Issues**: [GitHub Issues](https://github.com/snowflake-eng/migrations-data-validation/issues)
 
 ---
 

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/__version__.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.3.3"
+__version__ = "1.3.4"

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/public/__init__.py

@@ -42,6 +42,9 @@ from snowflake.snowflake_data_validation.public.data_validation_api import (
     build_validation_file_path,
     compare_md5_chunks,
     compare_md5_rows_chunk,
+    create_source_table_context_from_configuration,
+    create_table_context,
+    create_target_table_context_from_configuration,
     execute_query,
     execute_query_from_table_configuration,
     execute_statement,
@@ -155,6 +158,9 @@ __all__ = [
     "build_validation_file_path",
     "compare_md5_chunks",
     "compare_md5_rows_chunk",
+    "create_source_table_context_from_configuration",
+    "create_table_context",
+    "create_target_table_context_from_configuration",
     "execute_query",
     "execute_query_from_table_configuration",
     "execute_statement",

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/public/_internal.py

@@ -22,6 +22,8 @@ WARNING: Functions in this module are internal implementation details and
 may change without notice. Do not import from this module directly.
 """
 
+from __future__ import annotations
+
 import os
 
 from collections.abc import Callable

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/public/data_validation_api.py

@@ -13,7 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from collections.abc import Callable
+from pathlib import Path
 
 import pandas as pd
 
@@ -54,15 +57,23 @@ from snowflake.snowflake_data_validation.utils.constants import (
     ROW_COUNT_KEY,
     SCHEMA_NAME_KEY,
     TABLE_NAME_KEY,
+    ObjectType,
+    Origin,
     Platform,
 )
 from snowflake.snowflake_data_validation.utils.helpers.helper_database import (
     HelperDatabase,
 )
+from snowflake.snowflake_data_validation.utils.model.column_metadata import (
+    ColumnMetadata,
+)
 from snowflake.snowflake_data_validation.utils.model.table_column_metadata import (
     TableColumnMetadata,
 )
 from snowflake.snowflake_data_validation.utils.model.table_context import TableContext
+from snowflake.snowflake_data_validation.utils.model.templates_loader_manager import (
+    TemplatesLoaderManager,
+)
 from snowflake.snowflake_data_validation.validation.cell_data_validator import (
     CellDataValidator,
 )
@@ -1288,3 +1299,170 @@ def build_validation_file_path(
         filename = f"{table_context.fully_qualified_name}.csv"
 
     return os.path.join(base_directory, validation_type, filename)
+
+
+def create_table_context(
+    columns: list[ColumnMetadata],
+    database_name: str,
+    fully_qualified_name: str,
+    id: int,
+    internal_fully_qualified_name: str,
+    internal_table_name: str,
+    object_type: ObjectType,
+    origin: Origin,
+    platform: Platform,
+    queries_templates_directory_path: str,
+    row_count: int,
+    run_id: str,
+    run_start_time: str,
+    schema_name: str,
+    table_name: str,
+    templates_directory_path: str,
+    apply_metric_column_modifier: bool = True,
+    chunk_number: int = 1,
+    column_mappings: dict[str, str] = None,
+    column_selection_list: list[str] = None,
+    exclude_metrics: bool = False,
+    index_column_collection: list[str] = None,
+    is_case_sensitive: bool = False,
+    is_exclusion_mode: bool = False,
+    max_failed_rows_number: int = 100,
+    where_clause: str = "",
+) -> TableContext:
+    """Create a TableContext with the provided parameters."""
+    has_where_clause = where_clause != ""
+    is_temporary_table = object_type != ObjectType.TABLE
+    sql_generator = SQLQueriesTemplateGenerator(
+        jinja_templates_folder_path=Path(queries_templates_directory_path)
+    )
+    templates_loader_manager = TemplatesLoaderManager(
+        templates_directory_path=Path(templates_directory_path),
+        platform=platform,
+    )
+    if column_mappings is None:
+        column_mappings = {}
+
+    if column_selection_list is None:
+        column_selection_list = []
+
+    if index_column_collection is None:
+        index_column_collection = []
+
+    table_context = TableContext(
+        apply_metric_column_modifier=apply_metric_column_modifier,
+        chunk_number=chunk_number,
+        column_mappings=column_mappings,
+        column_selection_list=column_selection_list,
+        columns=columns,
+        database_name=database_name,
+        exclude_metrics=exclude_metrics,
+        fully_qualified_name=fully_qualified_name,
+        has_where_clause=has_where_clause,
+        id=id,
+        internal_fully_qualified_name=internal_fully_qualified_name,
+        internal_table_name=internal_table_name,
+        is_case_sensitive=is_case_sensitive,
+        is_exclusion_mode=is_exclusion_mode,
+        is_temporary_table=is_temporary_table,
+        max_failed_rows_number=max_failed_rows_number,
+        object_type=object_type,
+        origin=origin,
+        platform=platform,
+        row_count=row_count,
+        run_id=run_id,
+        run_start_time=run_start_time,
+        schema_name=schema_name,
+        sql_generator=sql_generator,
+        table_name=table_name,
+        templates_loader_manager=templates_loader_manager,
+        user_index_column_collection=index_column_collection,
+        where_clause=where_clause,
+    )
+
+    return table_context
+
+
+def create_source_table_context_from_configuration(
+    table_configuration: TableConfiguration,
+    columns: list[ColumnMetadata],
+    platform: Platform,
+    row_count: int,
+    queries_templates_directory_path: str,
+    templates_directory_path: str,
+    run_id: str,
+    run_start_time: str,
+) -> TableContext:
+    """Create a source TableContext from a given TableConfiguration."""
+    source_table_context = create_table_context(
+        apply_metric_column_modifier=table_configuration.apply_metric_column_modifier,
+        chunk_number=table_configuration.chunk_number,
+        column_mappings=table_configuration.column_mappings,
+        column_selection_list=table_configuration.column_selection_list,
+        columns=columns,
+        database_name=table_configuration.source_database,
+        exclude_metrics=table_configuration.exclude_metrics,
+        fully_qualified_name=table_configuration.fully_qualified_name,
+        id=table_configuration.id,
+        index_column_collection=table_configuration.index_column_list,
+        internal_fully_qualified_name=table_configuration.internal_fully_qualified_name,
+        internal_table_name=table_configuration.internal_source_table,
+        is_case_sensitive=table_configuration.is_case_sensitive,
+        is_exclusion_mode=table_configuration.use_column_selection_as_exclude_list,
+        max_failed_rows_number=table_configuration.max_failed_rows_number,
+        object_type=table_configuration.object_type,
+        origin=Origin.SOURCE,
+        platform=platform,
+        queries_templates_directory_path=queries_templates_directory_path,
+        row_count=row_count,
+        run_id=run_id,
+        run_start_time=run_start_time,
+        schema_name=table_configuration.source_schema,
+        table_name=table_configuration.source_table,
+        templates_directory_path=templates_directory_path,
+        where_clause=table_configuration.where_clause,
+    )
+
+    return source_table_context
+
+
+def create_target_table_context_from_configuration(
+    table_configuration: TableConfiguration,
+    columns: list[ColumnMetadata],
+    platform: Platform,
+    row_count: int,
+    queries_templates_directory_path: str,
+    templates_directory_path: str,
+    run_id: str,
+    run_start_time: str,
+) -> TableContext:
+    """Create a target TableContext from a given TableConfiguration."""
+    target_table_context = create_table_context(
+        apply_metric_column_modifier=table_configuration.apply_metric_column_modifier,
+        chunk_number=table_configuration.chunk_number,
+        column_mappings=table_configuration.column_mappings,
+        column_selection_list=table_configuration.column_selection_list,
+        columns=columns,
+        database_name=table_configuration.target_database,
+        exclude_metrics=table_configuration.exclude_metrics,
+        fully_qualified_name=table_configuration.target_fully_qualified_name,
+        id=table_configuration.id,
+        index_column_collection=table_configuration.target_index_column_list,
+        internal_fully_qualified_name=table_configuration.internal_target_fully_qualified_name,
+        internal_table_name=table_configuration.internal_target_name,
+        is_case_sensitive=table_configuration.is_case_sensitive,
+        is_exclusion_mode=table_configuration.use_column_selection_as_exclude_list,
+        max_failed_rows_number=table_configuration.max_failed_rows_number,
+        object_type=table_configuration.object_type,
+        origin=Origin.TARGET,
+        platform=platform,
+        queries_templates_directory_path=queries_templates_directory_path,
+        row_count=row_count,
+        run_id=run_id,
+        run_start_time=run_start_time,
+        schema_name=table_configuration.target_schema,
+        table_name=table_configuration.target_name,
+        templates_directory_path=templates_directory_path,
+        where_clause=table_configuration.target_where_clause,
+    )
+
+    return target_table_context

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/public/model/query_execution_result.py

@@ -15,6 +15,8 @@
 
 """Query execution result data transfer object."""
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 
 import pandas as pd

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/redshift/redshift_cli.py

@@ -428,7 +428,7 @@ def redshift_run_validation_ipc(
             mode=NAME_CONNECTION_MODE,
             name=snow_connection_name,
         )
-    elif snow_account and snow_password:
+    elif snow_account and (snow_password or snow_private_key_file):
         snow_credential_object = build_snowflake_credentials(
             account=snow_account,
             username=snow_username,

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/sqlserver/connector/connector_sql_server.py

@@ -74,6 +74,15 @@ def import_azure_identity():
 NAMED_INSTANCE_PATTERN_DETAILED = r"^[a-zA-Z0-9.-]+\\[a-zA-Z0-9_]+$"
 DEFAULT_SQL_SERVER_PORT = 1433
 
+def _escape_odbc_value(value: str) -> str:
+    """Escape a value for safe inclusion in an ODBC connection string.
+
+    ODBC uses `;` as a key-value separator and `{`/`}` as quoting
+    characters.  Wrapping in braces and doubling `}` is always safe,
+    even when the value has no special characters.
+    """
+    return "{" + value.replace("}", "}}") + "}"
+
 # Azure SQL Database scope for token authentication
 AZURE_SQL_SCOPE = "https://database.windows.net/.default"
 
@@ -575,14 +584,17 @@ class ConnectorSqlServer(BaseConnector):
 
             else:
                 # SQL Server Authentication (username/password)
+                safe_user = _escape_odbc_value(user) if user else user
+                safe_password = _escape_odbc_value(password) if password else password
+
                 if is_named_instance and not is_custom_port:
                     self.string_connection = SQLSERVER_CONNECTION_STRING_NAMED.format(
                         driver=driver,
                         server=host,
                         port=port,
                         database=database,
-                        user=user,
-                        password=password,
+                        user=safe_user,
+                        password=safe_password,
                         trust_server_certificate=trust_server_certificate,
                         encrypt=encrypt,
                     )
@@ -592,8 +604,8 @@ class ConnectorSqlServer(BaseConnector):
                         server=host,
                         port=port,
                         database=database,
-                        user=user,
-                        password=password,
+                        user=safe_user,
+                        password=safe_password,
                         trust_server_certificate=trust_server_certificate,
                         encrypt=encrypt,
                     )

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/sqlserver/sqlserver_cli.py

@@ -534,7 +534,7 @@ def sqlserver_run_validation_ipc(
             mode=NAME_CONNECTION_MODE,
             name=snow_connection_name,
         )
-    elif snow_account and snow_password:
+    elif snow_account and (snow_password or snow_private_key_file):
         # Build Snowflake credentials
         snow_credential_object = build_snowflake_credentials(
             account=snow_account,

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/teradata/teradata_cli.py

@@ -709,7 +709,7 @@ def teradata_run_validation_ipc(
             mode=NAME_CONNECTION_MODE,
             name=snow_connection_name,
         )
-    elif snow_account and snow_password:
+    elif snow_account and (snow_password or snow_private_key_file):
         # Build Snowflake credentials target
         snow_credential_object = build_snowflake_credentials(
             account=snow_account,

{snowflake_data_validation-1.3.3 → snowflake_data_validation-1.3.4}/src/snowflake/snowflake_data_validation/validation/validation_report_buffer.py

@@ -13,6 +13,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import logging
 import os
 import queue

snowflake_data_validation-1.3.4/tests/unit/test_escape_odbc_value.py

@@ -0,0 +1,94 @@
+# Copyright 2026 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Unit tests for _escape_odbc_value in ConnectorSqlServer."""
+
+import pytest
+
+from snowflake.snowflake_data_validation.sqlserver.connector.connector_sql_server import (
+    SQLSERVER_CONNECTION_STRING,
+    _escape_odbc_value,
+)
+
+
+class TestEscapeOdbcValue:
+    """Tests for _escape_odbc_value ODBC connection string escaping."""
+
+    def test_plain_password(self):
+        assert _escape_odbc_value("SimplePass1!") == "{SimplePass1!}"
+
+    def test_semicolon(self):
+        assert _escape_odbc_value("Semi;Colon123@") == "{Semi;Colon123@}"
+
+    def test_opening_brace_at_start(self):
+        assert _escape_odbc_value("{StartBrace1!") == "{{StartBrace1!}"
+
+    def test_closing_brace_doubled(self):
+        assert _escape_odbc_value("Pass}word1!") == "{Pass}}word1!}"
+
+    def test_both_braces(self):
+        assert _escape_odbc_value("a{b}c") == "{a{b}}c}"
+
+    def test_semicolon_and_braces(self):
+        assert _escape_odbc_value("Semi;{Colon}123@") == "{Semi;{Colon}}123@}"
+
+    def test_multiple_semicolons(self):
+        assert _escape_odbc_value("a;b;c;d") == "{a;b;c;d}"
+
+    def test_multiple_closing_braces(self):
+        assert _escape_odbc_value("a}}b") == "{a}}}}b}"
+
+    def test_empty_string(self):
+        assert _escape_odbc_value("") == "{}"
+
+    @pytest.mark.parametrize(
+        "char",
+        ["=", " ", "\\", "'", '"', "%", "&", "@", "#", "]", "["],
+    )
+    def test_non_special_chars_safe(self, char):
+        pwd = f"Pass{char}word1!"
+        result = _escape_odbc_value(pwd)
+        assert result == "{" + pwd + "}"
+
+
+class TestEscapedPasswordInConnectionString:
+    """Verify escaped passwords produce valid ODBC connection strings."""
+
+    PARAMS = dict(
+        driver="ODBC Driver 18 for SQL Server",
+        server="127.0.0.1",
+        port=1433,
+        database="testdb",
+        user="sa",
+        trust_server_certificate="yes",
+        encrypt="yes",
+    )
+
+    def test_semicolon_not_split(self):
+        safe = _escape_odbc_value("Semi;Colon123@")
+        conn_str = SQLSERVER_CONNECTION_STRING.format(password=safe, **self.PARAMS)
+        assert "PWD={Semi;Colon123@};" in conn_str
+        assert "Encrypt=yes" in conn_str
+
+    def test_brace_at_start_not_misinterpreted(self):
+        safe = _escape_odbc_value("{StartBrace1!")
+        conn_str = SQLSERVER_CONNECTION_STRING.format(password=safe, **self.PARAMS)
+        assert "PWD={{StartBrace1!};" in conn_str
+
+    def test_plain_password_in_connection_string(self):
+        safe = _escape_odbc_value("NormalPass1!")
+        conn_str = SQLSERVER_CONNECTION_STRING.format(password=safe, **self.PARAMS)
+        assert "PWD={NormalPass1!};" in conn_str
+        assert "UID=sa;" in conn_str