Skip to content

Commit

Permalink
CodeGen from PR 12677 in Azure/azure-rest-api-specs
Browse files Browse the repository at this point in the history
Merge 30d369607e0f5a3e773a5024e5ae6d2fc8c805bf into a57e89f
  • Loading branch information
SDKAuto committed Jan 28, 2021
1 parent 82e5afc commit 49c2b42
Show file tree
Hide file tree
Showing 9 changed files with 491 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -144,9 +144,9 @@
"virtual_hubs": "VirtualHubsOperations",
"hub_virtual_network_connections": "HubVirtualNetworkConnectionsOperations",
"vpn_gateways": "VpnGatewaysOperations",
"vpn_link_connections": "VpnLinkConnectionsOperations",
"vpn_connections": "VpnConnectionsOperations",
"vpn_site_link_connections": "VpnSiteLinkConnectionsOperations",
"vpn_link_connections": "VpnLinkConnectionsOperations",
"nat_rules": "NatRulesOperations",
"p2_svpn_gateways": "P2SVpnGatewaysOperations",
"vpn_server_configurations_associated_with_virtual_wan": "VpnServerConfigurationsAssociatedWithVirtualWanOperations",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,9 @@
from .operations import VirtualHubsOperations
from .operations import HubVirtualNetworkConnectionsOperations
from .operations import VpnGatewaysOperations
from .operations import VpnLinkConnectionsOperations
from .operations import VpnConnectionsOperations
from .operations import VpnSiteLinkConnectionsOperations
from .operations import VpnLinkConnectionsOperations
from .operations import NatRulesOperations
from .operations import P2SVpnGatewaysOperations
from .operations import VpnServerConfigurationsAssociatedWithVirtualWanOperations
Expand Down Expand Up @@ -315,12 +315,12 @@ class NetworkManagementClient(NetworkManagementClientOperationsMixin):
:vartype hub_virtual_network_connections: azure.mgmt.network.v2020_08_01.operations.HubVirtualNetworkConnectionsOperations
:ivar vpn_gateways: VpnGatewaysOperations operations
:vartype vpn_gateways: azure.mgmt.network.v2020_08_01.operations.VpnGatewaysOperations
:ivar vpn_link_connections: VpnLinkConnectionsOperations operations
:vartype vpn_link_connections: azure.mgmt.network.v2020_08_01.operations.VpnLinkConnectionsOperations
:ivar vpn_connections: VpnConnectionsOperations operations
:vartype vpn_connections: azure.mgmt.network.v2020_08_01.operations.VpnConnectionsOperations
:ivar vpn_site_link_connections: VpnSiteLinkConnectionsOperations operations
:vartype vpn_site_link_connections: azure.mgmt.network.v2020_08_01.operations.VpnSiteLinkConnectionsOperations
:ivar vpn_link_connections: VpnLinkConnectionsOperations operations
:vartype vpn_link_connections: azure.mgmt.network.v2020_08_01.operations.VpnLinkConnectionsOperations
:ivar nat_rules: NatRulesOperations operations
:vartype nat_rules: azure.mgmt.network.v2020_08_01.operations.NatRulesOperations
:ivar p2_svpn_gateways: P2SVpnGatewaysOperations operations
Expand Down Expand Up @@ -552,12 +552,12 @@ def __init__(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_gateways = VpnGatewaysOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_link_connections = VpnLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_connections = VpnConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_site_link_connections = VpnSiteLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_link_connections = VpnLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.nat_rules = NatRulesOperations(
self._client, self._config, self._serialize, self._deserialize)
self.p2_svpn_gateways = P2SVpnGatewaysOperations(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,9 +109,9 @@
from .operations import VirtualHubsOperations
from .operations import HubVirtualNetworkConnectionsOperations
from .operations import VpnGatewaysOperations
from .operations import VpnLinkConnectionsOperations
from .operations import VpnConnectionsOperations
from .operations import VpnSiteLinkConnectionsOperations
from .operations import VpnLinkConnectionsOperations
from .operations import NatRulesOperations
from .operations import P2SVpnGatewaysOperations
from .operations import VpnServerConfigurationsAssociatedWithVirtualWanOperations
Expand Down Expand Up @@ -313,12 +313,12 @@ class NetworkManagementClient(NetworkManagementClientOperationsMixin):
:vartype hub_virtual_network_connections: azure.mgmt.network.v2020_08_01.aio.operations.HubVirtualNetworkConnectionsOperations
:ivar vpn_gateways: VpnGatewaysOperations operations
:vartype vpn_gateways: azure.mgmt.network.v2020_08_01.aio.operations.VpnGatewaysOperations
:ivar vpn_link_connections: VpnLinkConnectionsOperations operations
:vartype vpn_link_connections: azure.mgmt.network.v2020_08_01.aio.operations.VpnLinkConnectionsOperations
:ivar vpn_connections: VpnConnectionsOperations operations
:vartype vpn_connections: azure.mgmt.network.v2020_08_01.aio.operations.VpnConnectionsOperations
:ivar vpn_site_link_connections: VpnSiteLinkConnectionsOperations operations
:vartype vpn_site_link_connections: azure.mgmt.network.v2020_08_01.aio.operations.VpnSiteLinkConnectionsOperations
:ivar vpn_link_connections: VpnLinkConnectionsOperations operations
:vartype vpn_link_connections: azure.mgmt.network.v2020_08_01.aio.operations.VpnLinkConnectionsOperations
:ivar nat_rules: NatRulesOperations operations
:vartype nat_rules: azure.mgmt.network.v2020_08_01.aio.operations.NatRulesOperations
:ivar p2_svpn_gateways: P2SVpnGatewaysOperations operations
Expand Down Expand Up @@ -549,12 +549,12 @@ def __init__(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_gateways = VpnGatewaysOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_link_connections = VpnLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_connections = VpnConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_site_link_connections = VpnSiteLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.vpn_link_connections = VpnLinkConnectionsOperations(
self._client, self._config, self._serialize, self._deserialize)
self.nat_rules = NatRulesOperations(
self._client, self._config, self._serialize, self._deserialize)
self.p2_svpn_gateways = P2SVpnGatewaysOperations(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -99,9 +99,9 @@
from ._virtual_hubs_operations import VirtualHubsOperations
from ._hub_virtual_network_connections_operations import HubVirtualNetworkConnectionsOperations
from ._vpn_gateways_operations import VpnGatewaysOperations
from ._vpn_link_connections_operations import VpnLinkConnectionsOperations
from ._vpn_connections_operations import VpnConnectionsOperations
from ._vpn_site_link_connections_operations import VpnSiteLinkConnectionsOperations
from ._vpn_link_connections_operations import VpnLinkConnectionsOperations
from ._nat_rules_operations import NatRulesOperations
from ._p2_svpn_gateways_operations import P2SVpnGatewaysOperations
from ._vpn_server_configurations_associated_with_virtual_wan_operations import VpnServerConfigurationsAssociatedWithVirtualWanOperations
Expand Down Expand Up @@ -208,9 +208,9 @@
'VirtualHubsOperations',
'HubVirtualNetworkConnectionsOperations',
'VpnGatewaysOperations',
'VpnLinkConnectionsOperations',
'VpnConnectionsOperations',
'VpnSiteLinkConnectionsOperations',
'VpnLinkConnectionsOperations',
'NatRulesOperations',
'P2SVpnGatewaysOperations',
'VpnServerConfigurationsAssociatedWithVirtualWanOperations',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1239,3 +1239,113 @@ def get_long_running_output(pipeline_response):
else:
return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
begin_get_ike_sas.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/connections/{virtualNetworkGatewayConnectionName}/getikesas'} # type: ignore

async def _reset_connection_initial(
self,
resource_group_name: str,
virtual_network_gateway_connection_name: str,
**kwargs
) -> None:
cls = kwargs.pop('cls', None) # type: ClsType[None]
error_map = {
401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
}
error_map.update(kwargs.pop('error_map', {}))
api_version = "2020-08-01"
accept = "application/json"

# Construct URL
url = self._reset_connection_initial.metadata['url'] # type: ignore
path_format_arguments = {
'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
'virtualNetworkGatewayConnectionName': self._serialize.url("virtual_network_gateway_connection_name", virtual_network_gateway_connection_name, 'str'),
'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.post(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 [202]:
map_error(status_code=response.status_code, response=response, error_map=error_map)
error = self._deserialize(_models.ErrorResponse, response)
raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)

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

_reset_connection_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/connections/{virtualNetworkGatewayConnectionName}/resetconnection'} # type: ignore

async def begin_reset_connection(
self,
resource_group_name: str,
virtual_network_gateway_connection_name: str,
**kwargs
) -> AsyncLROPoller[None]:
"""Resets the virtual network gateway connection specified.
:param resource_group_name: The name of the resource group.
:type resource_group_name: str
:param virtual_network_gateway_connection_name: The name of the virtual network gateway
Connection.
:type virtual_network_gateway_connection_name: str
:keyword callable cls: A custom type or function that will be passed the direct response
:keyword str continuation_token: A continuation token to restart a poller from a saved state.
:keyword polling: True for ARMPolling, False for no polling, or a
polling object for personal polling strategy
:paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
:keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
:return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
:rtype: ~azure.core.polling.AsyncLROPoller[None]
:raises ~azure.core.exceptions.HttpResponseError:
"""
polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod]
cls = kwargs.pop('cls', None) # type: ClsType[None]
lro_delay = kwargs.pop(
'polling_interval',
self._config.polling_interval
)
cont_token = kwargs.pop('continuation_token', None) # type: Optional[str]
if cont_token is None:
raw_result = await self._reset_connection_initial(
resource_group_name=resource_group_name,
virtual_network_gateway_connection_name=virtual_network_gateway_connection_name,
cls=lambda x,y,z: x,
**kwargs
)

kwargs.pop('error_map', None)
kwargs.pop('content_type', None)

def get_long_running_output(pipeline_response):
if cls:
return cls(pipeline_response, None, {})

path_format_arguments = {
'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
'virtualNetworkGatewayConnectionName': self._serialize.url("virtual_network_gateway_connection_name", virtual_network_gateway_connection_name, 'str'),
'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
}

if polling is True: polling_method = AsyncARMPolling(lro_delay, lro_options={'final-state-via': 'location'}, path_format_arguments=path_format_arguments, **kwargs)
elif polling is False: polling_method = AsyncNoPolling()
else: polling_method = polling
if cont_token:
return AsyncLROPoller.from_continuation_token(
polling_method=polling_method,
continuation_token=cont_token,
client=self._client,
deserialization_callback=get_long_running_output
)
else:
return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
begin_reset_connection.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/connections/{virtualNetworkGatewayConnectionName}/resetconnection'} # type: ignore
Loading

0 comments on commit 49c2b42

Please sign in to comment.