Skip to content

Commit

Permalink
CodeGen from PR 15498 in Azure/azure-rest-api-specs
Browse files Browse the repository at this point in the history
Added OutboundNetworkDependenciesEndpoints API (Azure#15498)

* Adds VNet peering to 2021-04-01-preview api-version.

* Added OutboundNetworkDependencies.

* Fixed Prettier issue

* Fixed lint error

* Edited sample output

* Added reference to preview/vnetpeering.json in readme file

* Edited description for outbound network dependency endpoints.

* Corrected return type for outbound network dependency endpoints.

* Corrected return type for outbound network dependency endpoints.

* Correcting schema

* Correcting schema

Co-authored-by: nipuneth <[email protected]>
Co-authored-by: Sharanjeet Kaur <[email protected]>
  • Loading branch information
3 people committed Aug 18, 2021
1 parent 728b28b commit 971485d
Show file tree
Hide file tree
Showing 16 changed files with 425 additions and 15 deletions.
8 changes: 4 additions & 4 deletions sdk/databricks/azure-mgmt-databricks/_meta.json
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
{
"autorest": "3.4.2",
"autorest": "3.4.5",
"use": [
"@autorest/[email protected].1",
"@autorest/[email protected].4",
"@autorest/[email protected]"
],
"commit": "fa3ba1acdd45ddad8950133befc5b0a6f1ee5163",
"commit": "a3c9363637ad7d30407cd6dd26d280cbb166cbf9",
"repository_url": "https://github.com/Azure/azure-rest-api-specs",
"autorest_command": "autorest specification/databricks/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/[email protected].1 --use=@autorest/[email protected] --version=3.4.2",
"autorest_command": "autorest specification/databricks/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/[email protected].4 --use=@autorest/[email protected] --version=3.4.5",
"readme": "specification/databricks/resource-manager/readme.md"
}
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
from .operations import Operations
from .operations import PrivateLinkResourcesOperations
from .operations import PrivateEndpointConnectionsOperations
from .operations import OutboundNetworkDependenciesEndpointsOperations
from .operations import VNetPeeringOperations
from . import models

Expand All @@ -38,6 +39,8 @@ class AzureDatabricksManagementClient(object):
:vartype private_link_resources: azure_databricks_management_client.operations.PrivateLinkResourcesOperations
:ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations
:vartype private_endpoint_connections: azure_databricks_management_client.operations.PrivateEndpointConnectionsOperations
:ivar outbound_network_dependencies_endpoints: OutboundNetworkDependenciesEndpointsOperations operations
:vartype outbound_network_dependencies_endpoints: azure_databricks_management_client.operations.OutboundNetworkDependenciesEndpointsOperations
:ivar vnet_peering: VNetPeeringOperations operations
:vartype vnet_peering: azure_databricks_management_client.operations.VNetPeeringOperations
:param credential: Credential needed for the client to connect to Azure.
Expand Down Expand Up @@ -74,6 +77,8 @@ def __init__(
self._client, self._config, self._serialize, self._deserialize)
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.outbound_network_dependencies_endpoints = OutboundNetworkDependenciesEndpointsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vnet_peering = VNetPeeringOperations(
self._client, self._config, self._serialize, self._deserialize)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ def __init__(

self.credential = credential
self.subscription_id = subscription_id
self.api_version = "2021-04-01-preview"
self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
kwargs.setdefault('sdk_moniker', 'mgmt-databricks/{}'.format(VERSION))
self._configure(**kwargs)
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{
"chosen_version": "",
"total_api_version_list": ["2018-04-01", "2021-04-01-preview"],
"chosen_version": "2021-04-01-preview",
"total_api_version_list": ["2021-04-01-preview"],
"client": {
"name": "AzureDatabricksManagementClient",
"filename": "_azure_databricks_management_client",
Expand Down Expand Up @@ -102,6 +102,7 @@
"operations": "Operations",
"private_link_resources": "PrivateLinkResourcesOperations",
"private_endpoint_connections": "PrivateEndpointConnectionsOperations",
"outbound_network_dependencies_endpoints": "OutboundNetworkDependenciesEndpointsOperations",
"vnet_peering": "VNetPeeringOperations"
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,4 @@
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
# --------------------------------------------------------------------------

VERSION = "1.1.0b1"
VERSION = "1.0.0b1"
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
from .operations import Operations
from .operations import PrivateLinkResourcesOperations
from .operations import PrivateEndpointConnectionsOperations
from .operations import OutboundNetworkDependenciesEndpointsOperations
from .operations import VNetPeeringOperations
from .. import models

Expand All @@ -36,6 +37,8 @@ class AzureDatabricksManagementClient(object):
:vartype private_link_resources: azure_databricks_management_client.aio.operations.PrivateLinkResourcesOperations
:ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations
:vartype private_endpoint_connections: azure_databricks_management_client.aio.operations.PrivateEndpointConnectionsOperations
:ivar outbound_network_dependencies_endpoints: OutboundNetworkDependenciesEndpointsOperations operations
:vartype outbound_network_dependencies_endpoints: azure_databricks_management_client.aio.operations.OutboundNetworkDependenciesEndpointsOperations
:ivar vnet_peering: VNetPeeringOperations operations
:vartype vnet_peering: azure_databricks_management_client.aio.operations.VNetPeeringOperations
:param credential: Credential needed for the client to connect to Azure.
Expand Down Expand Up @@ -71,6 +74,8 @@ def __init__(
self._client, self._config, self._serialize, self._deserialize)
self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.outbound_network_dependencies_endpoints = OutboundNetworkDependenciesEndpointsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vnet_peering = VNetPeeringOperations(
self._client, self._config, self._serialize, self._deserialize)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ def __init__(

self.credential = credential
self.subscription_id = subscription_id
self.api_version = "2021-04-01-preview"
self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
kwargs.setdefault('sdk_moniker', 'mgmt-databricks/{}'.format(VERSION))
self._configure(**kwargs)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,14 @@
from ._operations import Operations
from ._private_link_resources_operations import PrivateLinkResourcesOperations
from ._private_endpoint_connections_operations import PrivateEndpointConnectionsOperations
from ._outbound_network_dependencies_endpoints_operations import OutboundNetworkDependenciesEndpointsOperations
from ._vnet_peering_operations import VNetPeeringOperations

__all__ = [
'WorkspacesOperations',
'Operations',
'PrivateLinkResourcesOperations',
'PrivateEndpointConnectionsOperations',
'OutboundNetworkDependenciesEndpointsOperations',
'VNetPeeringOperations',
]
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
# 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 typing import Any, Callable, Dict, Generic, List, Optional, TypeVar
import warnings

from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
from azure.core.pipeline import PipelineResponse
from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest
from azure.mgmt.core.exceptions import ARMErrorFormat

from ... import models as _models

T = TypeVar('T')
ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]

class OutboundNetworkDependenciesEndpointsOperations:
"""OutboundNetworkDependenciesEndpointsOperations async operations.
You should not instantiate this class directly. Instead, you should create a Client instance that
instantiates it for you and attaches it as an attribute.
:ivar models: Alias to model classes used in this operation group.
:type models: ~azure_databricks_management_client.models
:param client: Client for service requests.
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
"""

models = _models

def __init__(self, client, config, serializer, deserializer) -> None:
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self._config = config

async def list(
self,
resource_group_name: str,
workspace_name: str,
**kwargs: Any
) -> List["_models.OutboundEnvironmentEndpoint"]:
"""Gets a list of egress endpoints (network endpoints of all outbound dependencies) in the specified Workspace.
Gets the list of endpoints that VNET Injected Workspace's Cluster call Azure Databricks Control
Plane. You must configure outbound access to these endpoints. For more information, see
https://docs.microsoft.com/en-us/azure/databricks/administration-guide/cloud-configurations/azure/udr.
:param resource_group_name: The name of the resource group. The name is case insensitive.
:type resource_group_name: str
:param workspace_name: The name of the workspace.
:type workspace_name: str
:keyword callable cls: A custom type or function that will be passed the direct response
:return: list of OutboundEnvironmentEndpoint, or the result of cls(response)
:rtype: list[~azure_databricks_management_client.models.OutboundEnvironmentEndpoint]
:raises: ~azure.core.exceptions.HttpResponseError
"""
cls = kwargs.pop('cls', None) # type: ClsType[List["_models.OutboundEnvironmentEndpoint"]]
error_map = {
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2021-04-01-preview"
accept = "application/json"

# Construct URL
url = self.list.metadata['url'] # type: ignore
path_format_arguments = {
'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'),
'workspaceName': self._serialize.url("workspace_name", workspace_name, 'str', max_length=64, min_length=3),
'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
}
url = self._client.format_url(url, **path_format_arguments)

# Construct parameters
query_parameters = {} # type: Dict[str, Any]
query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')

# Construct headers
header_parameters = {} # type: Dict[str, Any]
header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')

request = self._client.get(url, query_parameters, header_parameters)
pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs)
response = pipeline_response.http_response

if response.status_code not in [200]:
map_error(status_code=response.status_code, response=response, error_map=error_map)
error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, response)
raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)

deserialized = self._deserialize('[OutboundEnvironmentEndpoint]', pipeline_response)

if cls:
return cls(pipeline_response, deserialized, {})

return deserialized
list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Databricks/workspaces/{workspaceName}/outboundNetworkDependenciesEndpoints'} # type: ignore
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ async def get(
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2018-04-01"
api_version = "2021-04-01-preview"
accept = "application/json"

# Construct URL
Expand Down Expand Up @@ -120,7 +120,7 @@ async def _delete_initial(
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2018-04-01"
api_version = "2021-04-01-preview"
accept = "application/json"

# Construct URL
Expand Down Expand Up @@ -237,7 +237,7 @@ async def _create_or_update_initial(
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2018-04-01"
api_version = "2021-04-01-preview"
content_type = kwargs.pop("content_type", "application/json")
accept = "application/json"

Expand Down Expand Up @@ -383,7 +383,7 @@ def list_by_workspace(
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2018-04-01"
api_version = "2021-04-01-preview"
accept = "application/json"

def prepare_request(next_link=None):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@
from ._models_py3 import EncryptionEntitiesDefinition
from ._models_py3 import EncryptionV2
from ._models_py3 import EncryptionV2KeyVaultProperties
from ._models_py3 import EndpointDependency
from ._models_py3 import EndpointDetail
from ._models_py3 import ErrorDetail
from ._models_py3 import ErrorInfo
from ._models_py3 import ErrorResponse
Expand All @@ -22,6 +24,7 @@
from ._models_py3 import Operation
from ._models_py3 import OperationDisplay
from ._models_py3 import OperationListResult
from ._models_py3 import OutboundEnvironmentEndpoint
from ._models_py3 import PrivateEndpoint
from ._models_py3 import PrivateEndpointConnection
from ._models_py3 import PrivateEndpointConnectionProperties
Expand Down Expand Up @@ -53,6 +56,8 @@
from ._models import EncryptionEntitiesDefinition # type: ignore
from ._models import EncryptionV2 # type: ignore
from ._models import EncryptionV2KeyVaultProperties # type: ignore
from ._models import EndpointDependency # type: ignore
from ._models import EndpointDetail # type: ignore
from ._models import ErrorDetail # type: ignore
from ._models import ErrorInfo # type: ignore
from ._models import ErrorResponse # type: ignore
Expand All @@ -62,6 +67,7 @@
from ._models import Operation # type: ignore
from ._models import OperationDisplay # type: ignore
from ._models import OperationListResult # type: ignore
from ._models import OutboundEnvironmentEndpoint # type: ignore
from ._models import PrivateEndpoint # type: ignore
from ._models import PrivateEndpointConnection # type: ignore
from ._models import PrivateEndpointConnectionProperties # type: ignore
Expand Down Expand Up @@ -108,6 +114,8 @@
'EncryptionEntitiesDefinition',
'EncryptionV2',
'EncryptionV2KeyVaultProperties',
'EndpointDependency',
'EndpointDetail',
'ErrorDetail',
'ErrorInfo',
'ErrorResponse',
Expand All @@ -117,6 +125,7 @@
'Operation',
'OperationDisplay',
'OperationListResult',
'OutboundEnvironmentEndpoint',
'PrivateEndpoint',
'PrivateEndpointConnection',
'PrivateEndpointConnectionProperties',
Expand Down
Loading

0 comments on commit 971485d

Please sign in to comment.