duckguard 2.0.0__py3-none-any.whl → 2.3.0__py3-none-any.whl

duckguard/connectors/__init__.py

@@ -1,8 +1,8 @@
 """Connectors for various data sources."""
 
-from duckguard.connectors.base import Connector, ConnectionConfig
-from duckguard.connectors.files import FileConnector, S3Connector, GCSConnector, AzureConnector
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.connectors.factory import connect, register_connector
+from duckguard.connectors.files import AzureConnector, FileConnector, GCSConnector, S3Connector
 
 # Database connectors (imported lazily to avoid import errors)
 __all__ = [

duckguard/connectors/bigquery.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/databricks.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 from urllib.parse import parse_qs, urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/factory.py

@@ -4,12 +4,11 @@ from __future__ import annotations
 
 from typing import Any
 
-from duckguard.connectors.base import Connector, ConnectionConfig
-from duckguard.connectors.files import FileConnector, S3Connector, GCSConnector, AzureConnector
+from duckguard.connectors.base import ConnectionConfig, Connector
+from duckguard.connectors.files import AzureConnector, FileConnector, GCSConnector, S3Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 
-
 # Registry of available connectors
 _CONNECTORS: list[type[Connector]] = [
     S3Connector,

duckguard/connectors/files.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import os
 from pathlib import Path
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/kafka.py

@@ -6,7 +6,7 @@ import json
 from typing import Any
 from urllib.parse import parse_qs, urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 
@@ -320,7 +320,7 @@ class KafkaStreamValidator:
             "messages_failed": 0,
         }
 
-    def add_rule(self, rule: callable) -> "KafkaStreamValidator":
+    def add_rule(self, rule: callable) -> KafkaStreamValidator:
         """Add a validation rule."""
         self.rules.append(rule)
         return self

duckguard/connectors/mongodb.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/mysql.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/oracle.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/postgres.py

@@ -2,10 +2,9 @@
 
 from __future__ import annotations
 
-import re
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/redshift.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from urllib.parse import urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/snowflake.py

@@ -2,11 +2,10 @@
 
 from __future__ import annotations
 
-import re
 from typing import Any
 from urllib.parse import parse_qs, urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/sqlite.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import os
 from pathlib import Path
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 

duckguard/connectors/sqlserver.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 from urllib.parse import parse_qs, urlparse
 
-from duckguard.connectors.base import Connector, ConnectionConfig
+from duckguard.connectors.base import ConnectionConfig, Connector
 from duckguard.core.dataset import Dataset
 from duckguard.core.engine import DuckGuardEngine
 
@@ -55,20 +55,17 @@ class SQLServerConnector(Connector):
             Dataset object
         """
         # Try pyodbc first, then pymssql
-        try:
-            import pyodbc
+        import importlib.util
 
+        if importlib.util.find_spec("pyodbc") is not None:
             driver_module = "pyodbc"
-        except ImportError:
-            try:
-                import pymssql
-
-                driver_module = "pymssql"
-            except ImportError:
-                raise ImportError(
-                    "SQL Server support requires pyodbc or pymssql. "
-                    "Install with: pip install duckguard[sqlserver]"
-                )
+        elif importlib.util.find_spec("pymssql") is not None:
+            driver_module = "pymssql"
+        else:
+            raise ImportError(
+                "SQL Server support requires pyodbc or pymssql. "
+                "Install with: pip install duckguard[sqlserver]"
+            )
 
         if not config.table:
             raise ValueError("Table name is required for SQL Server connections")

duckguard/contracts/__init__.py

@@ -14,17 +14,17 @@ Example:
         print(f"Contract violations: {result.violations}")
 """
 
+from duckguard.contracts.diff import SchemaDiff, diff_contracts
+from duckguard.contracts.generator import generate_contract
+from duckguard.contracts.loader import contract_to_yaml, load_contract, load_contract_from_string
 from duckguard.contracts.schema import (
+    ContractMetadata,
     DataContract,
-    SchemaField,
     FieldType,
     QualitySLA,
-    ContractMetadata,
+    SchemaField,
 )
-from duckguard.contracts.loader import load_contract, load_contract_from_string, contract_to_yaml
-from duckguard.contracts.validator import validate_contract, ContractValidationResult
-from duckguard.contracts.generator import generate_contract
-from duckguard.contracts.diff import diff_contracts, SchemaDiff
+from duckguard.contracts.validator import ContractValidationResult, validate_contract
 
 __all__ = [
     # Schema

duckguard/contracts/diff.py

@@ -9,7 +9,7 @@ from dataclasses import dataclass, field
 from enum import Enum
 from typing import Any
 
-from duckguard.contracts.schema import DataContract, SchemaField, FieldType
+from duckguard.contracts.schema import DataContract, FieldType, SchemaField
 
 
 class ChangeType(Enum):

duckguard/contracts/generator.py

@@ -7,19 +7,18 @@ from __future__ import annotations
 
 from datetime import datetime
 from pathlib import Path
-from typing import Any
 
-from duckguard.core.dataset import Dataset
 from duckguard.connectors import connect
+from duckguard.contracts.loader import contract_to_yaml
 from duckguard.contracts.schema import (
+    ContractMetadata,
     DataContract,
-    SchemaField,
-    FieldType,
     FieldConstraint,
+    FieldType,
     QualitySLA,
-    ContractMetadata,
+    SchemaField,
 )
-from duckguard.contracts.loader import contract_to_yaml
+from duckguard.core.dataset import Dataset
 from duckguard.semantic import SemanticAnalyzer, SemanticType
 
 

duckguard/contracts/loader.py

@@ -47,12 +47,12 @@ from typing import Any
 import yaml
 
 from duckguard.contracts.schema import (
+    ContractMetadata,
     DataContract,
-    SchemaField,
-    FieldType,
     FieldConstraint,
+    FieldType,
     QualitySLA,
-    ContractMetadata,
+    SchemaField,
 )
 
 
@@ -82,7 +82,7 @@ def load_contract(path: str | Path) -> DataContract:
     if not path.exists():
         raise FileNotFoundError(f"Contract file not found: {path}")
 
-    with open(path, "r", encoding="utf-8") as f:
+    with open(path, encoding="utf-8") as f:
         content = f.read()
 
     return load_contract_from_string(content, source_file=str(path))

duckguard/contracts/validator.py

@@ -6,14 +6,13 @@ Validates datasets against data contracts to ensure compliance.
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from datetime import datetime, timedelta
+from datetime import datetime
 from enum import Enum
 from typing import Any
-import re
 
-from duckguard.core.dataset import Dataset
 from duckguard.connectors import connect
-from duckguard.contracts.schema import DataContract, SchemaField, FieldType
+from duckguard.contracts.schema import DataContract, SchemaField
+from duckguard.core.dataset import Dataset
 
 
 class ViolationType(Enum):

duckguard/core/__init__.py

@@ -1,8 +1,8 @@
 """Core module containing the engine, dataset, and column classes."""
 
-from duckguard.core.engine import DuckGuardEngine
-from duckguard.core.dataset import Dataset
 from duckguard.core.column import Column
-from duckguard.core.result import ValidationResult, CheckResult
+from duckguard.core.dataset import Dataset
+from duckguard.core.engine import DuckGuardEngine
+from duckguard.core.result import CheckResult, ValidationResult
 
 __all__ = ["DuckGuardEngine", "Dataset", "Column", "ValidationResult", "CheckResult"]