table-validator 0.1.0__py3-none-any.whl

table_validator/__init__.py

@@ -0,0 +1,46 @@
+"""table_validator: validates data migrations between Azure and Databricks Delta Lake."""
+
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("table-validator")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+from table_validator.config.manager import (
+    ConfigNotFoundError,
+    default_config,
+    load_config,
+    require_config,
+    save_config,
+)
+from table_validator.config.schema import ValidatorConfig
+from table_validator.connectors.azure_connector import AzureConnector, AzureSqlConnector
+from table_validator.connectors.databricks_connector import DatabricksConnector
+from table_validator.models import CatalogValidationRequest, CatalogValidationResponse
+from table_validator.validators.blob_discovery import BlobCatalogValidator
+from table_validator.validators.catalog_validator import CatalogValidator
+from table_validator.validators.row_validator import AzureCsvValidator, AzureSqlValidator
+
+__all__ = [
+    "__version__",
+    # Validators
+    "CatalogValidator",
+    "AzureCsvValidator",
+    "AzureSqlValidator",
+    "BlobCatalogValidator",
+    # Connectors
+    "DatabricksConnector",
+    "AzureConnector",
+    "AzureSqlConnector",
+    # Config
+    "ValidatorConfig",
+    "ConfigNotFoundError",
+    "default_config",
+    "load_config",
+    "require_config",
+    "save_config",
+    # Core request/response models
+    "CatalogValidationRequest",
+    "CatalogValidationResponse",
+]

table_validator/auth/__init__.py

@@ -0,0 +1 @@
+"""Auth package: credential abstractions for Azure and Databricks."""

table_validator/auth/azure_auth.py

@@ -0,0 +1,52 @@
+"""Azure auth abstraction.
+
+Phase 1 auth: credentials are entered manually via the CLI wizard and
+stored in ~/.table_validator/.env. get_azure_credential() is the single
+place that reads them - every connector must call it instead of reading
+os.environ directly, so only this function needs to change when a later
+phase adds Azure CLI / Service Principal auth.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+from dotenv import dotenv_values
+
+from table_validator.config.schema import ValidatorConfig
+
+ENV_PATH = Path.home() / ".table_validator" / ".env"
+
+
+@dataclass
+class AzureCredential:
+    """Resolved Azure credentials for the connectors this tool uses today.
+
+    storage_account_key authenticates AzureConnector (Blob Storage);
+    sql_username/sql_password authenticate AzureSqlConnector (Azure SQL
+    Database). Both are optional here since a given validation run may
+    only need one of the two.
+    """
+
+    storage_account_key: Optional[str] = None
+    sql_username: Optional[str] = None
+    sql_password: Optional[str] = None
+
+
+def get_azure_credential(config: ValidatorConfig, env_path: Path = ENV_PATH) -> AzureCredential:
+    """
+    Resolve Azure credentials for the given config.
+
+    Phase 1: reads AZURE_STORAGE_KEY / AZURE_SQL_USERNAME / AZURE_SQL_PASSWORD
+    from ~/.table_validator/.env. A later phase can swap this body for
+    Azure CLI / Service Principal auth without changing any caller.
+    """
+    values = dotenv_values(env_path) if env_path.exists() else {}
+
+    return AzureCredential(
+        storage_account_key=values.get("AZURE_STORAGE_KEY") or None,
+        sql_username=values.get("AZURE_SQL_USERNAME") or None,
+        sql_password=values.get("AZURE_SQL_PASSWORD") or None,
+    )

table_validator/auth/databricks_auth.py

@@ -0,0 +1,31 @@
+"""Databricks auth abstraction.
+
+Phase 1 auth: the personal access token (PAT) is entered manually via the
+CLI wizard and stored in ~/.table_validator/.env. get_databricks_token() is
+the single place that reads it - every connector must call it instead of
+reading os.environ directly, so only this function needs to change when a
+later phase adds Databricks CLI / OAuth auth.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Optional
+
+from dotenv import dotenv_values
+
+from table_validator.config.schema import ValidatorConfig
+
+ENV_PATH = Path.home() / ".table_validator" / ".env"
+
+
+def get_databricks_token(config: ValidatorConfig, env_path: Path = ENV_PATH) -> Optional[str]:
+    """
+    Resolve the Databricks personal access token for the given config.
+
+    Phase 1: reads DATABRICKS_TOKEN from ~/.table_validator/.env. A later
+    phase can swap this body for Databricks CLI / OAuth auth without
+    changing any caller.
+    """
+    values = dotenv_values(env_path) if env_path.exists() else {}
+    return values.get("DATABRICKS_TOKEN") or None

table_validator/cli/__init__.py

@@ -0,0 +1 @@
+"""CLI package: Typer app and interactive configuration wizard."""