Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[AutoPR track2_azure-mgmt-hdinsightcontainers] HDInsight: onboard new apiversion 2023-11-01-preview #5713

Closed
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions sdk/hdinsight/azure-mgmt-hdinsightcontainers/_meta.json
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
{
"commit": "ec882de9043fe225777bbf5fd28ed5259b0949db",
"commit": "e847bf1bd278acab5a8efd37f6be3d6a1e943b0a",
"repository_url": "https://github.com/Azure/azure-rest-api-specs",
"autorest": "3.9.7",
"use": [
"@autorest/[email protected]",
"@autorest/[email protected]"
],
"autorest_command": "autorest specification/hdinsight/resource-manager/Microsoft.HDInsight/HDInsightOnAks/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.9.7 --version-tolerant=False",
"autorest_command": "autorest specification/hdinsight/resource-manager/Microsoft.HDInsight/HDInsightOnAks/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.9.7 --version-tolerant=False",
"readme": "specification/hdinsight/resource-manager/Microsoft.HDInsight/HDInsightOnAks/readme.md"
}
Original file line number Diff line number Diff line change
Expand Up @@ -29,14 +29,14 @@ class HDInsightContainersMgmtClientConfiguration(Configuration): # pylint: disa
:type credential: ~azure.core.credentials.TokenCredential
:param subscription_id: The ID of the target subscription. The value must be an UUID. Required.
:type subscription_id: str
:keyword api_version: Api Version. Default value is "2023-06-01-preview". Note that overriding
:keyword api_version: Api Version. Default value is "2023-11-01-preview". Note that overriding
this default value may result in unsupported behavior.
:paramtype api_version: str
"""

def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None:
super(HDInsightContainersMgmtClientConfiguration, self).__init__(**kwargs)
api_version: str = kwargs.pop("api_version", "2023-06-01-preview")
api_version: str = kwargs.pop("api_version", "2023-11-01-preview")

if credential is None:
raise ValueError("Parameter 'credential' must not be None.")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,9 @@
from .operations import (
AvailableClusterPoolVersionsOperations,
AvailableClusterVersionsOperations,
ClusterAvailableUpgradesOperations,
ClusterJobsOperations,
ClusterPoolAvailableUpgradesOperations,
ClusterPoolsOperations,
ClustersOperations,
LocationsOperations,
Expand All @@ -31,12 +33,18 @@


class HDInsightContainersMgmtClient: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes
"""HDInsight On Aks Management Client.
"""HDInsight Containers Management Client.

:ivar cluster_pools: ClusterPoolsOperations operations
:vartype cluster_pools: azure.mgmt.hdinsightcontainers.operations.ClusterPoolsOperations
:ivar cluster_pool_available_upgrades: ClusterPoolAvailableUpgradesOperations operations
:vartype cluster_pool_available_upgrades:
azure.mgmt.hdinsightcontainers.operations.ClusterPoolAvailableUpgradesOperations
:ivar clusters: ClustersOperations operations
:vartype clusters: azure.mgmt.hdinsightcontainers.operations.ClustersOperations
:ivar cluster_available_upgrades: ClusterAvailableUpgradesOperations operations
:vartype cluster_available_upgrades:
azure.mgmt.hdinsightcontainers.operations.ClusterAvailableUpgradesOperations
:ivar cluster_jobs: ClusterJobsOperations operations
:vartype cluster_jobs: azure.mgmt.hdinsightcontainers.operations.ClusterJobsOperations
:ivar locations: LocationsOperations operations
Expand All @@ -55,7 +63,7 @@ class HDInsightContainersMgmtClient: # pylint: disable=client-accepts-api-versi
:type subscription_id: str
:param base_url: Service URL. Default value is "https://management.azure.com".
:type base_url: str
:keyword api_version: Api Version. Default value is "2023-06-01-preview". Note that overriding
:keyword api_version: Api Version. Default value is "2023-11-01-preview". Note that overriding
this default value may result in unsupported behavior.
:paramtype api_version: str
:keyword int polling_interval: Default waiting time between two polls for LRO operations if no
Expand All @@ -79,7 +87,13 @@ def __init__(
self._deserialize = Deserializer(client_models)
self._serialize.client_side_validation = False
self.cluster_pools = ClusterPoolsOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_pool_available_upgrades = ClusterPoolAvailableUpgradesOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.clusters = ClustersOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_available_upgrades = ClusterAvailableUpgradesOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.cluster_jobs = ClusterJobsOperations(self._client, self._config, self._serialize, self._deserialize)
self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize)
self.operations = Operations(self._client, self._config, self._serialize, self._deserialize)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,14 +29,14 @@ class HDInsightContainersMgmtClientConfiguration(Configuration): # pylint: disa
:type credential: ~azure.core.credentials_async.AsyncTokenCredential
:param subscription_id: The ID of the target subscription. The value must be an UUID. Required.
:type subscription_id: str
:keyword api_version: Api Version. Default value is "2023-06-01-preview". Note that overriding
:keyword api_version: Api Version. Default value is "2023-11-01-preview". Note that overriding
this default value may result in unsupported behavior.
:paramtype api_version: str
"""

def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None:
super(HDInsightContainersMgmtClientConfiguration, self).__init__(**kwargs)
api_version: str = kwargs.pop("api_version", "2023-06-01-preview")
api_version: str = kwargs.pop("api_version", "2023-11-01-preview")

if credential is None:
raise ValueError("Parameter 'credential' must not be None.")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,9 @@
from .operations import (
AvailableClusterPoolVersionsOperations,
AvailableClusterVersionsOperations,
ClusterAvailableUpgradesOperations,
ClusterJobsOperations,
ClusterPoolAvailableUpgradesOperations,
ClusterPoolsOperations,
ClustersOperations,
LocationsOperations,
Expand All @@ -31,12 +33,18 @@


class HDInsightContainersMgmtClient: # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes
"""HDInsight On Aks Management Client.
"""HDInsight Containers Management Client.

:ivar cluster_pools: ClusterPoolsOperations operations
:vartype cluster_pools: azure.mgmt.hdinsightcontainers.aio.operations.ClusterPoolsOperations
:ivar cluster_pool_available_upgrades: ClusterPoolAvailableUpgradesOperations operations
:vartype cluster_pool_available_upgrades:
azure.mgmt.hdinsightcontainers.aio.operations.ClusterPoolAvailableUpgradesOperations
:ivar clusters: ClustersOperations operations
:vartype clusters: azure.mgmt.hdinsightcontainers.aio.operations.ClustersOperations
:ivar cluster_available_upgrades: ClusterAvailableUpgradesOperations operations
:vartype cluster_available_upgrades:
azure.mgmt.hdinsightcontainers.aio.operations.ClusterAvailableUpgradesOperations
:ivar cluster_jobs: ClusterJobsOperations operations
:vartype cluster_jobs: azure.mgmt.hdinsightcontainers.aio.operations.ClusterJobsOperations
:ivar locations: LocationsOperations operations
Expand All @@ -55,7 +63,7 @@ class HDInsightContainersMgmtClient: # pylint: disable=client-accepts-api-versi
:type subscription_id: str
:param base_url: Service URL. Default value is "https://management.azure.com".
:type base_url: str
:keyword api_version: Api Version. Default value is "2023-06-01-preview". Note that overriding
:keyword api_version: Api Version. Default value is "2023-11-01-preview". Note that overriding
this default value may result in unsupported behavior.
:paramtype api_version: str
:keyword int polling_interval: Default waiting time between two polls for LRO operations if no
Expand All @@ -79,7 +87,13 @@ def __init__(
self._deserialize = Deserializer(client_models)
self._serialize.client_side_validation = False
self.cluster_pools = ClusterPoolsOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_pool_available_upgrades = ClusterPoolAvailableUpgradesOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.clusters = ClustersOperations(self._client, self._config, self._serialize, self._deserialize)
self.cluster_available_upgrades = ClusterAvailableUpgradesOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.cluster_jobs = ClusterJobsOperations(self._client, self._config, self._serialize, self._deserialize)
self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize)
self.operations = Operations(self._client, self._config, self._serialize, self._deserialize)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,9 @@
# --------------------------------------------------------------------------

from ._cluster_pools_operations import ClusterPoolsOperations
from ._cluster_pool_available_upgrades_operations import ClusterPoolAvailableUpgradesOperations
from ._clusters_operations import ClustersOperations
from ._cluster_available_upgrades_operations import ClusterAvailableUpgradesOperations
from ._cluster_jobs_operations import ClusterJobsOperations
from ._locations_operations import LocationsOperations
from ._operations import Operations
Expand All @@ -20,7 +22,9 @@

__all__ = [
"ClusterPoolsOperations",
"ClusterPoolAvailableUpgradesOperations",
"ClustersOperations",
"ClusterAvailableUpgradesOperations",
"ClusterJobsOperations",
"LocationsOperations",
"Operations",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,150 @@
# pylint: disable=too-many-lines
# 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, AsyncIterable, Callable, Dict, Optional, TypeVar
import urllib.parse

from azure.core.async_paging import AsyncItemPaged, AsyncList
from azure.core.exceptions import (
ClientAuthenticationError,
HttpResponseError,
ResourceExistsError,
ResourceNotFoundError,
ResourceNotModifiedError,
map_error,
)
from azure.core.pipeline import PipelineResponse
from azure.core.pipeline.transport import AsyncHttpResponse
from azure.core.rest import HttpRequest
from azure.core.tracing.decorator import distributed_trace
from azure.core.utils import case_insensitive_dict
from azure.mgmt.core.exceptions import ARMErrorFormat

from ... import models as _models
from ..._vendor import _convert_request
from ...operations._cluster_available_upgrades_operations import build_list_request

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


class ClusterAvailableUpgradesOperations:
"""
.. warning::
**DO NOT** instantiate this class directly.

Instead, you should access the following operations through
:class:`~azure.mgmt.hdinsightcontainers.aio.HDInsightContainersMgmtClient`'s
:attr:`cluster_available_upgrades` attribute.
"""

models = _models

def __init__(self, *args, **kwargs) -> None:
input_args = list(args)
self._client = input_args.pop(0) if input_args else kwargs.pop("client")
self._config = input_args.pop(0) if input_args else kwargs.pop("config")
self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer")
self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer")

@distributed_trace
def list(
self, resource_group_name: str, cluster_pool_name: str, cluster_name: str, **kwargs: Any
) -> AsyncIterable["_models.ClusterAvailableUpgrade"]:
"""List a cluster available upgrade.

:param resource_group_name: The name of the resource group. The name is case insensitive.
Required.
:type resource_group_name: str
:param cluster_pool_name: The name of the cluster pool. Required.
:type cluster_pool_name: str
:param cluster_name: The name of the HDInsight cluster. Required.
:type cluster_name: str
:keyword callable cls: A custom type or function that will be passed the direct response
:return: An iterator like instance of either ClusterAvailableUpgrade or the result of
cls(response)
:rtype:
~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.hdinsightcontainers.models.ClusterAvailableUpgrade]
:raises ~azure.core.exceptions.HttpResponseError:
"""
_headers = kwargs.pop("headers", {}) or {}
_params = case_insensitive_dict(kwargs.pop("params", {}) or {})

api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version))
cls: ClsType[_models.ClusterAvailableUpgradeList] = kwargs.pop("cls", None)

error_map = {
401: ClientAuthenticationError,
404: ResourceNotFoundError,
409: ResourceExistsError,
304: ResourceNotModifiedError,
}
error_map.update(kwargs.pop("error_map", {}) or {})

def prepare_request(next_link=None):
if not next_link:

request = build_list_request(
resource_group_name=resource_group_name,
cluster_pool_name=cluster_pool_name,
cluster_name=cluster_name,
subscription_id=self._config.subscription_id,
api_version=api_version,
template_url=self.list.metadata["url"],
headers=_headers,
params=_params,
)
request = _convert_request(request)
request.url = self._client.format_url(request.url)

else:
# make call to next link with the client's api-version
_parsed_next_link = urllib.parse.urlparse(next_link)
_next_request_params = case_insensitive_dict(
{
key: [urllib.parse.quote(v) for v in value]
for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()
}
)
_next_request_params["api-version"] = self._config.api_version
request = HttpRequest(
"GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
)
request = _convert_request(request)
request.url = self._client.format_url(request.url)
request.method = "GET"
return request

async def extract_data(pipeline_response):
deserialized = self._deserialize("ClusterAvailableUpgradeList", pipeline_response)
list_of_elem = deserialized.value
if cls:
list_of_elem = cls(list_of_elem) # type: ignore
return deserialized.next_link or None, AsyncList(list_of_elem)

async def get_next(next_link=None):
request = prepare_request(next_link)

_stream = False
pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access
request, stream=_stream, **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, pipeline_response)
raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)

return pipeline_response

return AsyncItemPaged(get_next, extract_data)

list.metadata = {
"url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.HDInsight/clusterpools/{clusterPoolName}/clusters/{clusterName}/availableUpgrades"
}
Original file line number Diff line number Diff line change
Expand Up @@ -299,7 +299,12 @@ def get_long_running_output(pipeline_response):

@distributed_trace
def list(
self, resource_group_name: str, cluster_pool_name: str, cluster_name: str, **kwargs: Any
self,
resource_group_name: str,
cluster_pool_name: str,
cluster_name: str,
filter: Optional[str] = None,
**kwargs: Any
) -> AsyncIterable["_models.ClusterJob"]:
"""Get jobs of HDInsight on AKS cluster.

Expand All @@ -310,6 +315,9 @@ def list(
:type cluster_pool_name: str
:param cluster_name: The name of the HDInsight cluster. Required.
:type cluster_name: str
:param filter: The system query option to filter job returned in the response. Allowed value is
'jobName eq {jobName}' or 'jarName eq {jarName}'. Default value is None.
:type filter: str
:keyword callable cls: A custom type or function that will be passed the direct response
:return: An iterator like instance of either ClusterJob or the result of cls(response)
:rtype:
Expand Down Expand Up @@ -338,6 +346,7 @@ def prepare_request(next_link=None):
cluster_pool_name=cluster_pool_name,
cluster_name=cluster_name,
subscription_id=self._config.subscription_id,
filter=filter,
api_version=api_version,
template_url=self.list.metadata["url"],
headers=_headers,
Expand Down
Loading
Loading