diff --git a/HISTORY.rst b/HISTORY.rst index 6a2a49c..9ce0365 100644 --- a/HISTORY.rst +++ b/HISTORY.rst @@ -2,6 +2,11 @@ History ======= +1.0.1 (2026-05-11) +------------------ + +* Added ``databricks_lakebase`` to ``DatabaseType`` enum. + 1.0.0 (2026-04-21) ------------------ diff --git a/datamasque/client/models/connection.py b/datamasque/client/models/connection.py index 45f4780..a51d932 100644 --- a/datamasque/client/models/connection.py +++ b/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): diff --git a/pyproject.toml b/pyproject.toml index 3ae4919..24d07ec 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "datamasque-python" -version = "1.0.0" +version = "1.0.1" description = "Official Python client for the DataMasque data-masking API." authors = [ { name = "DataMasque Ltd" }, diff --git a/setup.cfg b/setup.cfg index da234e7..d0bf705 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,5 +1,5 @@ [bumpversion] -current_version = 1.0.0 +current_version = 1.0.1 commit = True tag = True diff --git a/tests/test_connections.py b/tests/test_connections.py index ff8dd30..d743fd6 100644 --- a/tests/test_connections.py +++ b/tests/test_connections.py @@ -768,6 +768,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", diff --git a/uv.lock b/uv.lock index 7ddfdac..ab96282 100644 --- a/uv.lock +++ b/uv.lock @@ -428,7 +428,7 @@ toml = [ [[package]] name = "datamasque-python" -version = "1.0.0" +version = "1.0.1" source = { editable = "." } dependencies = [ { name = "pydantic" },