[ReleasePR azure-mgmt-databricks] [Hub Generated] Public private branch 'databricks-privatelink'

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/__init__.py

@@ -9,10 +9,11 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-from .databricks_client import DatabricksClient
-from .version import VERSION
+from ._configuration import DatabricksClientConfiguration
+from ._databricks_client import DatabricksClient
+__all__ = ['DatabricksClient', 'DatabricksClientConfiguration']
 
-__all__ = ['DatabricksClient']
+from .version import VERSION
 
 __version__ = VERSION
 

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_configuration.py

@@ -0,0 +1,48 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+from msrestazure import AzureConfiguration
+
+from .version import VERSION
+
+
+class DatabricksClientConfiguration(AzureConfiguration):
+    """Configuration for DatabricksClient
+    Note that all parameters used to create this instance are saved as instance
+    attributes.
+
+    :param credentials: Credentials needed for the client to connect to Azure.
+    :type credentials: :mod:`A msrestazure Credentials
+     object<msrestazure.azure_active_directory>`
+    :param subscription_id: The ID of the target subscription.
+    :type subscription_id: str
+    :param str base_url: Service URL
+    """
+
+    def __init__(
+            self, credentials, subscription_id, base_url=None):
+
+        if credentials is None:
+            raise ValueError("Parameter 'credentials' must not be None.")
+        if subscription_id is None:
+            raise ValueError("Parameter 'subscription_id' must not be None.")
+        if not base_url:
+            base_url = 'https://management.azure.com'
+
+        super(DatabricksClientConfiguration, self).__init__(base_url)
+
+        # Starting Autorest.Python 4.0.64, make connection pool activated by default
+        self.keep_alive = True
+
+        self.add_user_agent('azure-mgmt-databricks/{}'.format(VERSION))
+        self.add_user_agent('Azure-SDK-For-Python')
+
+        self.credentials = credentials
+        self.subscription_id = subscription_id

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_databricks_client.py

@@ -0,0 +1,59 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+from msrest.service_client import SDKClient
+from msrest import Serializer, Deserializer
+
+from ._configuration import DatabricksClientConfiguration
+from .operations import WorkspacesOperations
+from .operations import VNetPeeringOperations
+from .operations import Operations
+from . import models
+
+
+class DatabricksClient(SDKClient):
+    """ARM Databricks
+
+    :ivar config: Configuration for client.
+    :vartype config: DatabricksClientConfiguration
+
+    :ivar workspaces: Workspaces operations
+    :vartype workspaces: azure.mgmt.databricks.operations.WorkspacesOperations
+    :ivar vnet_peering: VNetPeering operations
+    :vartype vnet_peering: azure.mgmt.databricks.operations.VNetPeeringOperations
+    :ivar operations: Operations operations
+    :vartype operations: azure.mgmt.databricks.operations.Operations
+
+    :param credentials: Credentials needed for the client to connect to Azure.
+    :type credentials: :mod:`A msrestazure Credentials
+     object<msrestazure.azure_active_directory>`
+    :param subscription_id: The ID of the target subscription.
+    :type subscription_id: str
+    :param str base_url: Service URL
+    """
+
+    def __init__(
+            self, credentials, subscription_id, base_url=None):
+
+        self.config = DatabricksClientConfiguration(credentials, subscription_id, base_url)
+        super(DatabricksClient, self).__init__(self.config.credentials, self.config)
+
+        client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
+        self.api_version = '2018-04-01'
+        self._serialize = Serializer(client_models)
+        self._deserialize = Deserializer(client_models)
+
+        self.workspaces = WorkspacesOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.vnet_peering = VNetPeeringOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.operations = Operations(
+            self._client, self.config, self._serialize, self._deserialize)

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/__init__.py

@@ -10,48 +10,94 @@
 # --------------------------------------------------------------------------
 
 try:
-    from .workspace_provider_authorization_py3 import WorkspaceProviderAuthorization
-    from .sku_py3 import Sku
-    from .workspace_py3 import Workspace
-    from .tracked_resource_py3 import TrackedResource
-    from .resource_py3 import Resource
-    from .workspace_update_py3 import WorkspaceUpdate
-    from .error_detail_py3 import ErrorDetail
-    from .error_info_py3 import ErrorInfo
-    from .error_response_py3 import ErrorResponse, ErrorResponseException
-    from .operation_display_py3 import OperationDisplay
-    from .operation_py3 import Operation
+    from ._models_py3 import AddressSpace
+    from ._models_py3 import CreatedBy
+    from ._models_py3 import Encryption
+    from ._models_py3 import ErrorDetail
+    from ._models_py3 import ErrorInfo
+    from ._models_py3 import ErrorResponse, ErrorResponseException
+    from ._models_py3 import ManagedIdentityConfiguration
+    from ._models_py3 import Operation
+    from ._models_py3 import OperationDisplay
+    from ._models_py3 import Resource
+    from ._models_py3 import Sku
+    from ._models_py3 import TrackedResource
+    from ._models_py3 import VirtualNetworkPeering
+    from ._models_py3 import VirtualNetworkPeeringPropertiesFormatDatabricksVirtualNetwork
+    from ._models_py3 import VirtualNetworkPeeringPropertiesFormatRemoteVirtualNetwork
+    from ._models_py3 import Workspace
+    from ._models_py3 import WorkspaceCustomBooleanParameter
+    from ._models_py3 import WorkspaceCustomObjectParameter
+    from ._models_py3 import WorkspaceCustomParameters
+    from ._models_py3 import WorkspaceCustomStringParameter
+    from ._models_py3 import WorkspaceEncryptionParameter
+    from ._models_py3 import WorkspaceProviderAuthorization
+    from ._models_py3 import WorkspaceUpdate
 except (SyntaxError, ImportError):
-    from .workspace_provider_authorization import WorkspaceProviderAuthorization
-    from .sku import Sku
-    from .workspace import Workspace
-    from .tracked_resource import TrackedResource
-    from .resource import Resource
-    from .workspace_update import WorkspaceUpdate
-    from .error_detail import ErrorDetail
-    from .error_info import ErrorInfo
-    from .error_response import ErrorResponse, ErrorResponseException
-    from .operation_display import OperationDisplay
-    from .operation import Operation
-from .workspace_paged import WorkspacePaged
-from .operation_paged import OperationPaged
-from .databricks_client_enums import (
+    from ._models import AddressSpace
+    from ._models import CreatedBy
+    from ._models import Encryption
+    from ._models import ErrorDetail
+    from ._models import ErrorInfo
+    from ._models import ErrorResponse, ErrorResponseException
+    from ._models import ManagedIdentityConfiguration
+    from ._models import Operation
+    from ._models import OperationDisplay
+    from ._models import Resource
+    from ._models import Sku
+    from ._models import TrackedResource
+    from ._models import VirtualNetworkPeering
+    from ._models import VirtualNetworkPeeringPropertiesFormatDatabricksVirtualNetwork
+    from ._models import VirtualNetworkPeeringPropertiesFormatRemoteVirtualNetwork
+    from ._models import Workspace
+    from ._models import WorkspaceCustomBooleanParameter
+    from ._models import WorkspaceCustomObjectParameter
+    from ._models import WorkspaceCustomParameters
+    from ._models import WorkspaceCustomStringParameter
+    from ._models import WorkspaceEncryptionParameter
+    from ._models import WorkspaceProviderAuthorization
+    from ._models import WorkspaceUpdate
+from ._paged_models import OperationPaged
+from ._paged_models import VirtualNetworkPeeringPaged
+from ._paged_models import WorkspacePaged
+from ._databricks_client_enums import (
+    CustomParameterType,
+    KeySource,
     ProvisioningState,
+    PeeringProvisioningState,
+    PeeringState,
 )
 
 __all__ = [
-    'WorkspaceProviderAuthorization',
-    'Sku',
-    'Workspace',
-    'TrackedResource',
-    'Resource',
-    'WorkspaceUpdate',
+    'AddressSpace',
+    'CreatedBy',
+    'Encryption',
     'ErrorDetail',
     'ErrorInfo',
     'ErrorResponse', 'ErrorResponseException',
-    'OperationDisplay',
+    'ManagedIdentityConfiguration',
     'Operation',
+    'OperationDisplay',
+    'Resource',
+    'Sku',
+    'TrackedResource',
+    'VirtualNetworkPeering',
+    'VirtualNetworkPeeringPropertiesFormatDatabricksVirtualNetwork',
+    'VirtualNetworkPeeringPropertiesFormatRemoteVirtualNetwork',
+    'Workspace',
+    'WorkspaceCustomBooleanParameter',
+    'WorkspaceCustomObjectParameter',
+    'WorkspaceCustomParameters',
+    'WorkspaceCustomStringParameter',
+    'WorkspaceEncryptionParameter',
+    'WorkspaceProviderAuthorization',
+    'WorkspaceUpdate',
     'WorkspacePaged',
+    'VirtualNetworkPeeringPaged',
     'OperationPaged',
+    'CustomParameterType',
+    'KeySource',
     'ProvisioningState',
+    'PeeringProvisioningState',
+    'PeeringState',
 ]

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/_databricks_client_enums.py

@@ -0,0 +1,55 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+from enum import Enum
+
+
+class CustomParameterType(str, Enum):
+
+    bool_enum = "Bool"
+    object_enum = "Object"
+    string = "String"
+
+
+class KeySource(str, Enum):
+
+    default = "Default"
+    microsoft_keyvault = "Microsoft.Keyvault"
+
+
+class ProvisioningState(str, Enum):
+
+    accepted = "Accepted"
+    running = "Running"
+    ready = "Ready"
+    creating = "Creating"
+    created = "Created"
+    deleting = "Deleting"
+    deleted = "Deleted"
+    canceled = "Canceled"
+    failed = "Failed"
+    succeeded = "Succeeded"
+    updating = "Updating"
+
+
+class PeeringProvisioningState(str, Enum):
+
+    succeeded = "Succeeded"
+    updating = "Updating"
+    deleting = "Deleting"
+    failed = "Failed"
+
+
+class PeeringState(str, Enum):
+
+    initiated = "Initiated"
+    connected = "Connected"
+    disconnected = "Disconnected"