datamasque-python 1.0.0__tar.gz → 1.0.2__tar.gz

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/HISTORY.rst

@@ -2,6 +2,16 @@
 History
 =======
 
+1.0.2 (2026-05-14)
+------------------
+
+* Added ``DatabricksDeltaS3ConnectionConfig`` for Databricks Delta tables stored in S3.
+
+1.0.1 (2026-05-11)
+------------------
+
+* Added ``databricks_lakebase`` to ``DatabaseType`` enum.
+
 1.0.0 (2026-04-21)
 ------------------
 

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamasque-python
-Version: 1.0.0
+Version: 1.0.2
 Summary: Official Python client for the DataMasque data-masking API.
 Project-URL: Homepage, https://datamasque.com/
 Project-URL: Documentation, https://datamasque-python.readthedocs.io/

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/datamasque/client/__init__.py

@@ -30,6 +30,7 @@ from datamasque.client.models.connection import (
     ConnectionId,
     DatabaseConnectionConfig,
     DatabaseType,
+    DatabricksDeltaS3ConnectionConfig,
     DynamoConnectionConfig,
     FileConnectionConfig,
     MongoConnectionConfig,
@@ -128,6 +129,7 @@ __all__ = [
     "DataMasqueUserError",
     "DatabaseConnectionConfig",
     "DatabaseType",
+    "DatabricksDeltaS3ConnectionConfig",
     "DiscoveryMatch",
     "DynamoConnectionConfig",
     "FailedToStartError",

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/datamasque/client/models/connection.py

@@ -44,6 +44,7 @@ class DatabaseType(Enum):
     mssql_linked = "mssql_linked"
     snowflake = "snowflake"
     mongodb = "mongodb"
+    databricks_lakebase = "databricks_lakebase"
 
 
 class SnowflakeStageLocation(str, Enum):
@@ -390,10 +391,19 @@ class MountedShareConnectionConfig(FileConnectionConfig):
     type: Literal["mounted_share_connection"] = "mounted_share_connection"
 
 
+class DatabricksDeltaS3ConnectionConfig(FileConnectionConfig):
+    """Connection configuration for Databricks Delta tables stored in S3."""
+
+    type: Literal["databricks_delta_s3_connection"] = "databricks_delta_s3_connection"
+    bucket: str = ""
+    iam_role_arn: Optional[str] = None
+
+
 FILE_TYPE_MAP: dict[str, type[FileConnectionConfig]] = {
     "s3_connection": S3ConnectionConfig,
     "azure_blob_connection": AzureConnectionConfig,
     "mounted_share_connection": MountedShareConnectionConfig,
+    "databricks_delta_s3_connection": DatabricksDeltaS3ConnectionConfig,
 }
 
 DB_TYPE_MAP: dict[str, type[ConnectionConfig]] = {

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datamasque-python"
-version = "1.0.0"
+version = "1.0.2"
 description = "Official Python client for the DataMasque data-masking API."
 authors = [
     { name = "DataMasque Ltd" },

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.0.0
+current_version = 1.0.2
 commit = True
 tag = True
 

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/tests/test_connections.py

@@ -9,6 +9,7 @@ from datamasque.client.models.connection import (
     ConnectionId,
     DatabaseConnectionConfig,
     DatabaseType,
+    DatabricksDeltaS3ConnectionConfig,
     DynamoConnectionConfig,
     MongoConnectionConfig,
     MountedShareConnectionConfig,
@@ -695,6 +696,65 @@ def test_s3_connection_model_validate_no_iam_role():
     assert conn.iam_role_arn is None
 
 
+def test_databricks_delta_s3_connection_model_validate():
+    payload = {
+        "id": "11223344-5566-7788-99aa-bbccddeeff00",
+        "name": "delta_s3",
+        "mask_type": "file",
+        "type": "databricks_delta_s3_connection",
+        "base_directory": "delta/",
+        "is_file_mask_source": True,
+        "is_file_mask_destination": False,
+        "bucket": "my-delta-bucket",
+        "iam_role_arn": "arn:aws:iam::111122223333:role/delta-role",
+    }
+
+    conn = DatabricksDeltaS3ConnectionConfig.model_validate(payload)
+
+    assert isinstance(conn, DatabricksDeltaS3ConnectionConfig)
+    assert conn.id == "11223344-5566-7788-99aa-bbccddeeff00"
+    assert conn.name == "delta_s3"
+    assert conn.bucket == "my-delta-bucket"
+    assert conn.base_directory == "delta/"
+    assert conn.is_file_mask_source is True
+    assert conn.is_file_mask_destination is False
+    assert conn.iam_role_arn == "arn:aws:iam::111122223333:role/delta-role"
+
+
+def test_databricks_delta_s3_connection_model_validate_no_iam_role():
+    payload = {
+        "id": "id-delta",
+        "name": "delta_s3",
+        "mask_type": "file",
+        "type": "databricks_delta_s3_connection",
+        "base_directory": "",
+        "is_file_mask_source": True,
+        "is_file_mask_destination": False,
+        "bucket": "my-delta-bucket",
+    }
+
+    conn = DatabricksDeltaS3ConnectionConfig.model_validate(payload)
+    assert conn.iam_role_arn is None
+
+
+def test_validate_connection_dispatches_databricks_delta_s3():
+    payload = {
+        "id": "aabb-ccdd",
+        "name": "delta",
+        "mask_type": "file",
+        "type": "databricks_delta_s3_connection",
+        "base_directory": "",
+        "is_file_mask_source": False,
+        "is_file_mask_destination": True,
+        "bucket": "delta-bucket",
+    }
+
+    conn = validate_connection(payload)
+
+    assert isinstance(conn, DatabricksDeltaS3ConnectionConfig)
+    assert conn.bucket == "delta-bucket"
+
+
 def test_azure_connection_model_validate_blanks_encrypted_connection_string():
     payload = {
         "id": "490502e5-5bf6-4abb-b67b-c6091d40ecf0",
@@ -768,6 +828,27 @@ def test_database_connection_model_validate_keeps_schema_for_postgres():
     assert conn.db_schema == "public"
 
 
+def test_database_connection_model_validate_databricks_lakebase():
+    payload = {
+        "id": "abc-lakebase",
+        "name": "lakebase",
+        "mask_type": "database",
+        "db_type": "databricks_lakebase",
+        "host": "lakebase-host",
+        "port": 5432,
+        "database": "lakebasedb",
+        "user": "lakebase_user",
+        "schema": "public",
+        "is_read_only": False,
+    }
+
+    conn = DatabaseConnectionConfig.model_validate(payload)
+
+    assert isinstance(conn, DatabaseConnectionConfig)
+    assert conn.database_type is DatabaseType.databricks_lakebase
+    assert conn.db_schema == "public"
+
+
 def test_mssql_linked_connection_model_validate_includes_linked_server():
     payload = {
         "id": "48a7af45-f63f-4e05-bf9f-7b1cc3a0e89d",

{datamasque_python-1.0.0 → datamasque_python-1.0.2}/uv.lock

@@ -428,7 +428,7 @@ toml = [
 
 [[package]]
 name = "datamasque-python"
-version = "1.0.0"
+version = "1.0.2"
 source = { editable = "." }
 dependencies = [
     { name = "pydantic" },