dominodatalab-data 5.8.0.dev4__tar.gz → 5.8.0.dev6__tar.gz

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dominodatalab-data
-Version: 5.8.0.dev4
+Version: 5.8.0.dev6
 Summary: Domino Data API for interacting with Domino Data features
 Home-page: https://github.com/dominodatalab/domino-data
 License: Apache Software License 2.0

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/datasource_api_client/models/datasource_dto_auth_type.py

@@ -11,6 +11,7 @@ class DatasourceDtoAuthType(str, Enum):
     GCPBASIC = "GCPBasic"
     OAUTH = "OAuth"
     CLIENTIDSECRET = "ClientIdSecret"
+    PERSONALTOKEN = "PersonalToken"
 
     def __str__(self) -> str:
         return str(self.value)

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/datasource_api_client/models/datasource_dto_data_source_type.py

@@ -6,6 +6,7 @@ class DatasourceDtoDataSourceType(str, Enum):
     AZUREBLOBSTORAGECONFIG = "AzureBlobStorageConfig"
     BIGQUERYCONFIG = "BigQueryConfig"
     CLICKHOUSECONFIG = "ClickHouseConfig"
+    DATABRICKSCONFIG = "DatabricksConfig"
     DB2CONFIG = "DB2Config"
     DRUIDCONFIG = "DruidConfig"
     GCSCONFIG = "GCSConfig"

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/domino_data/configuration_gen.py

@@ -1,6 +1,6 @@
 """Code generated by gen.py; DO NOT EDIT.
 This file was generated by robots at
-2023-08-27 22:11:53.655834"""
+2023-09-14 16:09:29.448849"""
 from typing import Any, Dict, Optional, Union
 
 from enum import Enum
@@ -67,12 +67,14 @@ class ConfigElem(Enum):
     DATABASE = "database"
     EXTRAPROPERTIES = "extraProperties"
     HOST = "host"
+    HTTPPATH = "httpPath"
     NETWORKINGPROXY = "networkingProxy"
     PORT = "port"
     PROJECT = "project"
     REGION = "region"
     ROLE = "role"
     SCHEMA = "schema"
+    QUERYTIMEOUT = "queryTimeout"
     WAREHOUSE = "warehouse"
     CATALOGCODE = "catalogCode"
 
@@ -90,6 +92,7 @@ class CredElem(Enum):
     USERNAME = "username"
     CLIENTID = "clientId"
     CLIENTSECRET = "clientSecret"
+    PERSONALACCESSTOKEN = "personalAccessToken"
 
 
 def _cred(elem: CredElem) -> Any:
@@ -145,6 +148,17 @@ class ClickHouseConfig(Config):
     password: Optional[str] = _cred(elem=CredElem.PASSWORD)
 
 
+@attr.s(auto_attribs=True)
+class DatabricksConfig(Config):
+    """DatabricksConfig datasource configuration."""
+
+    catalog: Optional[str] = _config(elem=ConfigElem.CATALOG)
+    schema: Optional[str] = _config(elem=ConfigElem.SCHEMA)
+    query_timeout: Optional[str] = _config(elem=ConfigElem.QUERYTIMEOUT)
+
+    personal_access_token: Optional[str] = _cred(elem=CredElem.PERSONALACCESSTOKEN)
+
+
 @attr.s(auto_attribs=True)
 class DB2Config(Config):
     """DB2Config datasource configuration."""
@@ -390,6 +404,7 @@ DatasourceConfig = Union[
     AzureBlobStorageConfig,
     BigQueryConfig,
     ClickHouseConfig,
+    DatabricksConfig,
     DB2Config,
     DruidConfig,
     GCSConfig,
@@ -423,6 +438,7 @@ DATASOURCES = {
     DatasourceDtoDataSourceType.AZUREBLOBSTORAGECONFIG: "ObjectStoreDatasource",
     DatasourceDtoDataSourceType.BIGQUERYCONFIG: "TabularDatasource",
     DatasourceDtoDataSourceType.CLICKHOUSECONFIG: "TabularDatasource",
+    DatasourceDtoDataSourceType.DATABRICKSCONFIG: "TabularDatasource",
     DatasourceDtoDataSourceType.DB2CONFIG: "TabularDatasource",
     DatasourceDtoDataSourceType.DRUIDCONFIG: "TabularDatasource",
     DatasourceDtoDataSourceType.GCSCONFIG: "ObjectStoreDatasource",

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/domino_data/data_sources.py

@@ -341,7 +341,10 @@ class Datasource:
 
         context = httpx.create_ssl_context()
 
-        if self.datasource_type == DatasourceDtoDataSourceType.ADLSCONFIG.value:
+        if self.datasource_type in (
+            DatasourceDtoDataSourceType.ADLSCONFIG.value,
+            DatasourceDtoDataSourceType.AZUREBLOBSTORAGECONFIG.value,
+        ):
             self._httpx = httpx.Client(headers=ADLS_HEADERS, verify=context)
         elif self.datasource_type == DatasourceDtoDataSourceType.GENERICS3CONFIG.value:
             self._httpx = httpx.Client(verify=False)  # nosec
@@ -351,7 +354,10 @@ class Datasource:
 
     def pool_manager(self) -> urllib3.PoolManager:
         """Urllib3 pool manager for range downloads."""
-        if self.datasource_type == DatasourceDtoDataSourceType.ADLSCONFIG.value:
+        if self.datasource_type in (
+            DatasourceDtoDataSourceType.ADLSCONFIG.value,
+            DatasourceDtoDataSourceType.AZUREBLOBSTORAGECONFIG.value,
+        ):
             return urllib3.PoolManager(headers=ADLS_HEADERS)
         elif self.datasource_type == DatasourceDtoDataSourceType.GENERICS3CONFIG.value:
             return urllib3.PoolManager(assert_hostname=False)

{dominodatalab_data-5.8.0.dev4 → dominodatalab_data-5.8.0.dev6}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "dominodatalab-data"
-version = "5.8.0dev4"
+version = "5.8.0dev6"
 description = "Domino Data API for interacting with Domino Data features"
 readme = "README.md"
 authors = [