1# coding=utf-8 2# -------------------------------------------------------------------------- 3# Copyright (c) Microsoft Corporation. All rights reserved. 4# Licensed under the MIT License. See License.txt in the project root for license information. 5# Code generated by Microsoft (R) AutoRest Code Generator. 6# Changes may cause incorrect behavior and will be lost if the code is regenerated. 7# -------------------------------------------------------------------------- 8from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar 9import warnings 10 11from azure.core.async_paging import AsyncItemPaged, AsyncList 12from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error 13from azure.core.pipeline import PipelineResponse 14from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest 15from azure.mgmt.core.exceptions import ARMErrorFormat 16 17from ... import models as _models 18 19T = TypeVar('T') 20ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] 21 22class LoadBalancerProbesOperations: 23 """LoadBalancerProbesOperations async operations. 24 25 You should not instantiate this class directly. Instead, you should create a Client instance that 26 instantiates it for you and attaches it as an attribute. 27 28 :ivar models: Alias to model classes used in this operation group. 29 :type models: ~azure.mgmt.network.v2020_03_01.models 30 :param client: Client for service requests. 31 :param config: Configuration of service client. 32 :param serializer: An object model serializer. 33 :param deserializer: An object model deserializer. 34 """ 35 36 models = _models 37 38 def __init__(self, client, config, serializer, deserializer) -> None: 39 self._client = client 40 self._serialize = serializer 41 self._deserialize = deserializer 42 self._config = config 43 44 def list( 45 self, 46 resource_group_name: str, 47 load_balancer_name: str, 48 **kwargs 49 ) -> AsyncIterable["_models.LoadBalancerProbeListResult"]: 50 """Gets all the load balancer probes. 51 52 :param resource_group_name: The name of the resource group. 53 :type resource_group_name: str 54 :param load_balancer_name: The name of the load balancer. 55 :type load_balancer_name: str 56 :keyword callable cls: A custom type or function that will be passed the direct response 57 :return: An iterator like instance of either LoadBalancerProbeListResult or the result of cls(response) 58 :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.network.v2020_03_01.models.LoadBalancerProbeListResult] 59 :raises: ~azure.core.exceptions.HttpResponseError 60 """ 61 cls = kwargs.pop('cls', None) # type: ClsType["_models.LoadBalancerProbeListResult"] 62 error_map = { 63 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError 64 } 65 error_map.update(kwargs.pop('error_map', {})) 66 api_version = "2020-03-01" 67 accept = "application/json" 68 69 def prepare_request(next_link=None): 70 # Construct headers 71 header_parameters = {} # type: Dict[str, Any] 72 header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') 73 74 if not next_link: 75 # Construct URL 76 url = self.list.metadata['url'] # type: ignore 77 path_format_arguments = { 78 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), 79 'loadBalancerName': self._serialize.url("load_balancer_name", load_balancer_name, 'str'), 80 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), 81 } 82 url = self._client.format_url(url, **path_format_arguments) 83 # Construct parameters 84 query_parameters = {} # type: Dict[str, Any] 85 query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') 86 87 request = self._client.get(url, query_parameters, header_parameters) 88 else: 89 url = next_link 90 query_parameters = {} # type: Dict[str, Any] 91 request = self._client.get(url, query_parameters, header_parameters) 92 return request 93 94 async def extract_data(pipeline_response): 95 deserialized = self._deserialize('LoadBalancerProbeListResult', pipeline_response) 96 list_of_elem = deserialized.value 97 if cls: 98 list_of_elem = cls(list_of_elem) 99 return deserialized.next_link or None, AsyncList(list_of_elem) 100 101 async def get_next(next_link=None): 102 request = prepare_request(next_link) 103 104 pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) 105 response = pipeline_response.http_response 106 107 if response.status_code not in [200]: 108 map_error(status_code=response.status_code, response=response, error_map=error_map) 109 raise HttpResponseError(response=response, error_format=ARMErrorFormat) 110 111 return pipeline_response 112 113 return AsyncItemPaged( 114 get_next, extract_data 115 ) 116 list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes'} # type: ignore 117 118 async def get( 119 self, 120 resource_group_name: str, 121 load_balancer_name: str, 122 probe_name: str, 123 **kwargs 124 ) -> "_models.Probe": 125 """Gets load balancer probe. 126 127 :param resource_group_name: The name of the resource group. 128 :type resource_group_name: str 129 :param load_balancer_name: The name of the load balancer. 130 :type load_balancer_name: str 131 :param probe_name: The name of the probe. 132 :type probe_name: str 133 :keyword callable cls: A custom type or function that will be passed the direct response 134 :return: Probe, or the result of cls(response) 135 :rtype: ~azure.mgmt.network.v2020_03_01.models.Probe 136 :raises: ~azure.core.exceptions.HttpResponseError 137 """ 138 cls = kwargs.pop('cls', None) # type: ClsType["_models.Probe"] 139 error_map = { 140 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError 141 } 142 error_map.update(kwargs.pop('error_map', {})) 143 api_version = "2020-03-01" 144 accept = "application/json" 145 146 # Construct URL 147 url = self.get.metadata['url'] # type: ignore 148 path_format_arguments = { 149 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), 150 'loadBalancerName': self._serialize.url("load_balancer_name", load_balancer_name, 'str'), 151 'probeName': self._serialize.url("probe_name", probe_name, 'str'), 152 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), 153 } 154 url = self._client.format_url(url, **path_format_arguments) 155 156 # Construct parameters 157 query_parameters = {} # type: Dict[str, Any] 158 query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') 159 160 # Construct headers 161 header_parameters = {} # type: Dict[str, Any] 162 header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') 163 164 request = self._client.get(url, query_parameters, header_parameters) 165 pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) 166 response = pipeline_response.http_response 167 168 if response.status_code not in [200]: 169 map_error(status_code=response.status_code, response=response, error_map=error_map) 170 raise HttpResponseError(response=response, error_format=ARMErrorFormat) 171 172 deserialized = self._deserialize('Probe', pipeline_response) 173 174 if cls: 175 return cls(pipeline_response, deserialized, {}) 176 177 return deserialized 178 get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'} # type: ignore 179