CodeGen from PR 14756 in Azure/azure-rest-api-specs

Merge 1b2072e92426252b5a4650c5fba8465ce78a3e0c into 9b0e13a

sdk/databricks/azure-mgmt-databricks/MANIFEST.in

@@ -1,3 +1,4 @@
+include _meta.json
 recursive-include tests *.py *.yaml
 include *.md
 include azure/__init__.py

sdk/databricks/azure-mgmt-databricks/_meta.json

@@ -0,0 +1,8 @@
+{
+  "autorest": "V2",
+  "use": "@microsoft.azure/autorest.python@~4.0.71",
+  "commit": "dbef665acb107827eff4cb9564ef36e00adb65c0",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "autorest_command": "autorest specification/databricks/resource-manager/readme.md --keep-version-file --multiapi --no-async --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --use=@microsoft.azure/autorest.python@~4.0.71 --version=V2",
+  "readme": "specification/databricks/resource-manager/readme.md"
+}

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 AzureDatabricksManagementClientConfiguration
+from ._azure_databricks_management_client import AzureDatabricksManagementClient
+__all__ = ['AzureDatabricksManagementClient', 'AzureDatabricksManagementClientConfiguration']
 
-__all__ = ['DatabricksClient']
+from .version import VERSION
 
 __version__ = VERSION
 

sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/databricks_client.py → sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_azure_databricks_management_client.py

@@ -11,55 +11,32 @@
 
 from msrest.service_client import SDKClient
 from msrest import Serializer, Deserializer
-from msrestazure import AzureConfiguration
-from .version import VERSION
-from .operations.workspaces_operations import WorkspacesOperations
-from .operations.operations import Operations
-from . import models
-
-
-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)
-
-        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
+from ._configuration import AzureDatabricksManagementClientConfiguration
+from .operations import WorkspacesOperations
+from .operations import Operations
+from .operations import PrivateLinkResourcesOperations
+from .operations import PrivateEndpointConnectionsOperations
+from .operations import VNetPeeringOperations
+from . import models
 
 
-class DatabricksClient(SDKClient):
-    """ARM Databricks
+class AzureDatabricksManagementClient(SDKClient):
+    """The Microsoft Azure management APIs allow end users to operate on Azure Databricks Workspace resources.
 
     :ivar config: Configuration for client.
-    :vartype config: DatabricksClientConfiguration
+    :vartype config: AzureDatabricksManagementClientConfiguration
 
     :ivar workspaces: Workspaces operations
     :vartype workspaces: azure.mgmt.databricks.operations.WorkspacesOperations
     :ivar operations: Operations operations
     :vartype operations: azure.mgmt.databricks.operations.Operations
+    :ivar private_link_resources: PrivateLinkResources operations
+    :vartype private_link_resources: azure.mgmt.databricks.operations.PrivateLinkResourcesOperations
+    :ivar private_endpoint_connections: PrivateEndpointConnections operations
+    :vartype private_endpoint_connections: azure.mgmt.databricks.operations.PrivateEndpointConnectionsOperations
+    :ivar vnet_peering: VNetPeering operations
+    :vartype vnet_peering: azure.mgmt.databricks.operations.VNetPeeringOperations
 
     :param credentials: Credentials needed for the client to connect to Azure.
     :type credentials: :mod:`A msrestazure Credentials
@@ -72,15 +49,20 @@ class DatabricksClient(SDKClient):
     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)
+        self.config = AzureDatabricksManagementClientConfiguration(credentials, subscription_id, base_url)
+        super(AzureDatabricksManagementClient, 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.operations = Operations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.private_link_resources = PrivateLinkResourcesOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.vnet_peering = VNetPeeringOperations(
+            self._client, self.config, self._serialize, self._deserialize)

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 AzureDatabricksManagementClientConfiguration(AzureConfiguration):
+    """Configuration for AzureDatabricksManagementClient
+    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(AzureDatabricksManagementClientConfiguration, 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/models/__init__.py

@@ -10,48 +10,141 @@
 # --------------------------------------------------------------------------
 
 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 EncryptionEntitiesDefinition
+    from ._models_py3 import EncryptionV2
+    from ._models_py3 import EncryptionV2KeyVaultProperties
+    from ._models_py3 import ErrorDetail
+    from ._models_py3 import ErrorInfo
+    from ._models_py3 import ErrorResponse, ErrorResponseException
+    from ._models_py3 import GroupIdInformation
+    from ._models_py3 import GroupIdInformationProperties
+    from ._models_py3 import ManagedIdentityConfiguration
+    from ._models_py3 import Operation
+    from ._models_py3 import OperationDisplay
+    from ._models_py3 import PrivateEndpoint
+    from ._models_py3 import PrivateEndpointConnection
+    from ._models_py3 import PrivateEndpointConnectionProperties
+    from ._models_py3 import PrivateLinkServiceConnectionState
+    from ._models_py3 import Resource
+    from ._models_py3 import Sku
+    from ._models_py3 import SystemData
+    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 WorkspacePropertiesEncryption
+    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 EncryptionEntitiesDefinition
+    from ._models import EncryptionV2
+    from ._models import EncryptionV2KeyVaultProperties
+    from ._models import ErrorDetail
+    from ._models import ErrorInfo
+    from ._models import ErrorResponse, ErrorResponseException
+    from ._models import GroupIdInformation
+    from ._models import GroupIdInformationProperties
+    from ._models import ManagedIdentityConfiguration
+    from ._models import Operation
+    from ._models import OperationDisplay
+    from ._models import PrivateEndpoint
+    from ._models import PrivateEndpointConnection
+    from ._models import PrivateEndpointConnectionProperties
+    from ._models import PrivateLinkServiceConnectionState
+    from ._models import Resource
+    from ._models import Sku
+    from ._models import SystemData
+    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 WorkspacePropertiesEncryption
+    from ._models import WorkspaceProviderAuthorization
+    from ._models import WorkspaceUpdate
+from ._paged_models import GroupIdInformationPaged
+from ._paged_models import OperationPaged
+from ._paged_models import PrivateEndpointConnectionPaged
+from ._paged_models import VirtualNetworkPeeringPaged
+from ._paged_models import WorkspacePaged
+from ._azure_databricks_management_client_enums import (
+    CustomParameterType,
+    KeySource,
     ProvisioningState,
+    PrivateLinkServiceConnectionStatus,
+    PrivateEndpointConnectionProvisioningState,
+    PublicNetworkAccess,
+    RequiredNsgRules,
+    CreatedByType,
+    PeeringProvisioningState,
+    PeeringState,
 )
 
 __all__ = [
-    'WorkspaceProviderAuthorization',
-    'Sku',
-    'Workspace',
-    'TrackedResource',
-    'Resource',
-    'WorkspaceUpdate',
+    'AddressSpace',
+    'CreatedBy',
+    'Encryption',
+    'EncryptionEntitiesDefinition',
+    'EncryptionV2',
+    'EncryptionV2KeyVaultProperties',
     'ErrorDetail',
     'ErrorInfo',
     'ErrorResponse', 'ErrorResponseException',
-    'OperationDisplay',
+    'GroupIdInformation',
+    'GroupIdInformationProperties',
+    'ManagedIdentityConfiguration',
     'Operation',
+    'OperationDisplay',
+    'PrivateEndpoint',
+    'PrivateEndpointConnection',
+    'PrivateEndpointConnectionProperties',
+    'PrivateLinkServiceConnectionState',
+    'Resource',
+    'Sku',
+    'SystemData',
+    'TrackedResource',
+    'VirtualNetworkPeering',
+    'VirtualNetworkPeeringPropertiesFormatDatabricksVirtualNetwork',
+    'VirtualNetworkPeeringPropertiesFormatRemoteVirtualNetwork',
+    'Workspace',
+    'WorkspaceCustomBooleanParameter',
+    'WorkspaceCustomObjectParameter',
+    'WorkspaceCustomParameters',
+    'WorkspaceCustomStringParameter',
+    'WorkspaceEncryptionParameter',
+    'WorkspacePropertiesEncryption',
+    'WorkspaceProviderAuthorization',
+    'WorkspaceUpdate',
     'WorkspacePaged',
     'OperationPaged',
+    'GroupIdInformationPaged',
+    'PrivateEndpointConnectionPaged',
+    'VirtualNetworkPeeringPaged',
+    'CustomParameterType',
+    'KeySource',
     'ProvisioningState',
+    'PrivateLinkServiceConnectionStatus',
+    'PrivateEndpointConnectionProvisioningState',
+    'PublicNetworkAccess',
+    'RequiredNsgRules',
+    'CreatedByType',
+    'PeeringProvisioningState',
+    'PeeringState',
 ]