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