diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/CHANGELOG.md b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/CHANGELOG.md index 676b5c00ad09..743aaad9eadf 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/CHANGELOG.md +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/CHANGELOG.md @@ -1,5 +1,27 @@ # Release History +## 1.0.0b2 (2026-03-18) + +### Features Added + + - Client `KubernetesConfigurationExtensionsMgmtClient` added parameter `cloud_setting` in method `__init__` + - Client `KubernetesConfigurationExtensionsMgmtClient` added method `send_request` + - Model `ProxyResource` added property `system_data` + - Model `Resource` added property `system_data` + - Added model `PatchExtensionProperties` + - Added enum `ResourceIdentityType` + +### Breaking Changes + + - This version introduces new hybrid models which have dual dictionary and model nature. Please follow https://aka.ms/azsdk/python/migrate/hybrid-models for migration. + - For the method breakings, please refer to https://aka.ms/azsdk/python/migrate/operations for migration. + - Model `PatchExtension` moved instance variable `auto_upgrade_minor_version`, `release_train`, `version`, `configuration_settings` and `configuration_protected_settings` under property `properties` + - Method `ExtensionsOperations.begin_delete` changed its parameter `force_delete` from `positional_or_keyword` to `keyword_only` + +### Other Changes + + - Deleted model `ExtensionsList` which actually was not used by SDK users + ## 1.0.0b1 (2025-05-19) ### Other Changes diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/MANIFEST.in b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/MANIFEST.in index abbf85688758..9a47241b23b5 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/MANIFEST.in +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/MANIFEST.in @@ -1,9 +1,8 @@ -include _meta.json -recursive-include tests *.py *.json -recursive-include samples *.py *.md include *.md +include LICENSE +include azure/mgmt/kubernetesconfiguration/extensions/py.typed +recursive-include tests *.py +recursive-include samples *.py *.md include azure/__init__.py include azure/mgmt/__init__.py include azure/mgmt/kubernetesconfiguration/__init__.py -include LICENSE -include azure/mgmt/kubernetesconfiguration/extensions/py.typed diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_meta.json b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_meta.json deleted file mode 100644 index 86d4feb44b6e..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_meta.json +++ /dev/null @@ -1,11 +0,0 @@ -{ - "commit": "5b9c12325aaeccb614ccce96e2baad116497679b", - "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest": "3.10.2", - "use": [ - "@autorest/python@6.34.1", - "@autorest/modelerfour@4.27.0" - ], - "autorest_command": "autorest specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.34.1 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", - "readme": "specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/readme.md" -} \ No newline at end of file diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_metadata.json b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_metadata.json new file mode 100644 index 000000000000..54e37c469b61 --- /dev/null +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/_metadata.json @@ -0,0 +1,10 @@ +{ + "apiVersion": "2024-11-01", + "apiVersions": { + "Microsoft.KubernetesConfiguration": "2024-11-01" + }, + "commit": "bf2f971b4319da9f5624da7f3c13d487be477897", + "repository_url": "https://github.com/Azure/azure-rest-api-specs", + "typespec_src": "specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions", + "emitterVersion": "0.61.0" +} \ No newline at end of file diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/apiview-properties.json b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/apiview-properties.json new file mode 100644 index 000000000000..2d98ac1eda8e --- /dev/null +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/apiview-properties.json @@ -0,0 +1,40 @@ +{ + "CrossLanguagePackageId": "Microsoft.KubernetesConfiguration", + "CrossLanguageDefinitionId": { + "azure.mgmt.kubernetesconfiguration.extensions.models.ErrorAdditionalInfo": "Azure.ResourceManager.CommonTypes.ErrorAdditionalInfo", + "azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail": "Azure.ResourceManager.CommonTypes.ErrorDetail", + "azure.mgmt.kubernetesconfiguration.extensions.models.ErrorResponse": "Azure.ResourceManager.CommonTypes.ErrorResponse", + "azure.mgmt.kubernetesconfiguration.extensions.models.Resource": "Azure.ResourceManager.CommonTypes.Resource", + "azure.mgmt.kubernetesconfiguration.extensions.models.ProxyResource": "Azure.ResourceManager.CommonTypes.ProxyResource", + "azure.mgmt.kubernetesconfiguration.extensions.models.Extension": "Microsoft.KubernetesConfiguration.Extension", + "azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionProperties": "Microsoft.KubernetesConfiguration.ExtensionProperties", + "azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionPropertiesAksAssignedIdentity": "Microsoft.KubernetesConfiguration.ExtensionPropertiesAksAssignedIdentity", + "azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionStatus": "Microsoft.KubernetesConfiguration.ExtensionStatus", + "azure.mgmt.kubernetesconfiguration.extensions.models.Identity": "Azure.ResourceManager.CommonTypes.Identity", + "azure.mgmt.kubernetesconfiguration.extensions.models.OperationStatusResult": "Microsoft.KubernetesConfiguration.OperationStatusResult", + "azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension": "Microsoft.KubernetesConfiguration.PatchExtension", + "azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtensionProperties": "Microsoft.KubernetesConfiguration.PatchExtensionProperties", + "azure.mgmt.kubernetesconfiguration.extensions.models.Plan": "Azure.ResourceManager.CommonTypes.Plan", + "azure.mgmt.kubernetesconfiguration.extensions.models.Scope": "Microsoft.KubernetesConfiguration.Scope", + "azure.mgmt.kubernetesconfiguration.extensions.models.ScopeCluster": "Microsoft.KubernetesConfiguration.ScopeCluster", + "azure.mgmt.kubernetesconfiguration.extensions.models.ScopeNamespace": "Microsoft.KubernetesConfiguration.ScopeNamespace", + "azure.mgmt.kubernetesconfiguration.extensions.models.SystemData": "Azure.ResourceManager.CommonTypes.SystemData", + "azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType": "Azure.ResourceManager.CommonTypes.createdByType", + "azure.mgmt.kubernetesconfiguration.extensions.models.ProvisioningState": "Microsoft.KubernetesConfiguration.ProvisioningState", + "azure.mgmt.kubernetesconfiguration.extensions.models.LevelType": "Microsoft.KubernetesConfiguration.LevelType", + "azure.mgmt.kubernetesconfiguration.extensions.models.AKSIdentityType": "Microsoft.KubernetesConfiguration.AKSIdentityType", + "azure.mgmt.kubernetesconfiguration.extensions.models.ResourceIdentityType": "Azure.ResourceManager.CommonTypes.ResourceIdentityType", + "azure.mgmt.kubernetesconfiguration.extensions.operations.ExtensionsOperations.get": "Microsoft.KubernetesConfiguration.Extensions.get", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.ExtensionsOperations.get": "Microsoft.KubernetesConfiguration.Extensions.get", + "azure.mgmt.kubernetesconfiguration.extensions.operations.ExtensionsOperations.begin_create": "Microsoft.KubernetesConfiguration.Extensions.create", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.ExtensionsOperations.begin_create": "Microsoft.KubernetesConfiguration.Extensions.create", + "azure.mgmt.kubernetesconfiguration.extensions.operations.ExtensionsOperations.begin_update": "Microsoft.KubernetesConfiguration.Extensions.update", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.ExtensionsOperations.begin_update": "Microsoft.KubernetesConfiguration.Extensions.update", + "azure.mgmt.kubernetesconfiguration.extensions.operations.ExtensionsOperations.begin_delete": "Microsoft.KubernetesConfiguration.Extensions.delete", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.ExtensionsOperations.begin_delete": "Microsoft.KubernetesConfiguration.Extensions.delete", + "azure.mgmt.kubernetesconfiguration.extensions.operations.ExtensionsOperations.list": "Microsoft.KubernetesConfiguration.Extensions.list", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.ExtensionsOperations.list": "Microsoft.KubernetesConfiguration.Extensions.list", + "azure.mgmt.kubernetesconfiguration.extensions.operations.OperationStatusOperations.get": "Microsoft.KubernetesConfiguration.Extensions.operationStatusGet", + "azure.mgmt.kubernetesconfiguration.extensions.aio.operations.OperationStatusOperations.get": "Microsoft.KubernetesConfiguration.Extensions.operationStatusGet" + } +} \ No newline at end of file diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/__init__.py index 8db66d3d0f0f..d55ccad1f573 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/__init__.py @@ -1 +1 @@ -__path__ = __import__("pkgutil").extend_path(__path__, __name__) +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/__init__.py index 8db66d3d0f0f..d55ccad1f573 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/__init__.py @@ -1 +1 @@ -__path__ = __import__("pkgutil").extend_path(__path__, __name__) +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/__init__.py index 8db66d3d0f0f..d55ccad1f573 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/__init__.py @@ -1 +1 @@ -__path__ = __import__("pkgutil").extend_path(__path__, __name__) +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/__init__.py index 5242c767a60a..3d27daceaf2d 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/__init__.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._kubernetes_configuration_extensions_mgmt_client import KubernetesConfigurationExtensionsMgmtClient # type: ignore +from ._client import KubernetesConfigurationExtensionsMgmtClient # type: ignore from ._version import VERSION __version__ = VERSION diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_kubernetes_configuration_extensions_mgmt_client.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_client.py similarity index 71% rename from sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_kubernetes_configuration_extensions_mgmt_client.py rename to sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_client.py index db71802891fa..cd5c2aac93a5 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_kubernetes_configuration_extensions_mgmt_client.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_client.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -17,17 +17,17 @@ from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from . import models as _models from ._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration from ._utils.serialization import Deserializer, Serializer from .operations import ExtensionsOperations, OperationStatusOperations if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials import TokenCredential class KubernetesConfigurationExtensionsMgmtClient: # pylint: disable=name-too-long - """KubernetesConfiguration Extensions Client. + """Use these APIs to create extension resources through ARM, for Kubernetes Clusters. :ivar extensions: ExtensionsOperations operations :vartype extensions: @@ -35,29 +35,45 @@ class KubernetesConfigurationExtensionsMgmtClient: # pylint: disable=name-too-l :ivar operation_status: OperationStatusOperations operations :vartype operation_status: azure.mgmt.kubernetesconfiguration.extensions.operations.OperationStatusOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: The ID of the target subscription. Required. + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is None. + :param base_url: Service host. Default value is None. :type base_url: str - :keyword api_version: Api Version. Default value is "2024-11-01". Note that overriding this - default value may result in unsupported behavior. + :keyword cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is + None. + :paramtype cloud_setting: ~azure.core.AzureClouds + :keyword api_version: The API version to use for this operation. Known values are "2024-11-01" + and None. Default value is "2024-11-01". 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 Retry-After header is present. """ def __init__( - self, credential: "TokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any + self, + credential: "TokenCredential", + subscription_id: str, + base_url: Optional[str] = None, + *, + cloud_setting: Optional["AzureClouds"] = None, + **kwargs: Any ) -> None: - _cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore + _endpoint = "{endpoint}" + _cloud = cloud_setting or settings.current.azure_cloud # type: ignore _endpoints = get_arm_endpoints(_cloud) if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) self._config = KubernetesConfigurationExtensionsMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, credential_scopes=credential_scopes, **kwargs + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + cloud_setting=cloud_setting, + credential_scopes=credential_scopes, + **kwargs ) _policies = kwargs.pop("policies", None) @@ -78,24 +94,23 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, base_url), policies=_policies, **kwargs) + self._client: ARMPipelineClient = ARMPipelineClient(base_url=cast(str, _endpoint), policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) + self._serialize = Serializer() + self._deserialize = Deserializer() self._serialize.client_side_validation = False self.extensions = ExtensionsOperations(self._client, self._config, self._serialize, self._deserialize) self.operation_status = OperationStatusOperations( self._client, self._config, self._serialize, self._deserialize ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest >>> request = HttpRequest("GET", "https://www.example.org/") - >>> response = client._send_request(request) + >>> response = client.send_request(request) For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request @@ -108,7 +123,11 @@ def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: """ request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_configuration.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_configuration.py index 5f4acdaabb70..9f92bf75586b 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_configuration.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_configuration.py @@ -3,11 +3,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, TYPE_CHECKING +from typing import Any, Optional, TYPE_CHECKING from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -15,6 +15,7 @@ from ._version import VERSION if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials import TokenCredential @@ -24,16 +25,29 @@ class KubernetesConfigurationExtensionsMgmtClientConfiguration: # pylint: disab Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: The ID of the target subscription. Required. + :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 "2024-11-01". Note that overriding this - default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :param cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is + None. + :type cloud_setting: ~azure.core.AzureClouds + :keyword api_version: The API version to use for this operation. Known values are "2024-11-01" + and None. Default value is "2024-11-01". 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: + def __init__( + self, + credential: "TokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + cloud_setting: Optional["AzureClouds"] = None, + **kwargs: Any + ) -> None: api_version: str = kwargs.pop("api_version", "2024-11-01") if credential is None: @@ -43,6 +57,8 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url + self.cloud_setting = cloud_setting self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-kubernetesconfiguration-extensions/{}".format(VERSION)) diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_patch.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_patch.py index 8bcb627aa475..87676c65a8f0 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_patch.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_patch.py @@ -7,9 +7,9 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/__init__.py index 0af9b28f6607..8026245c2abc 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/__init__.py @@ -1,6 +1,6 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/model_base.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/model_base.py new file mode 100644 index 000000000000..7b7f8ba67b53 --- /dev/null +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/model_base.py @@ -0,0 +1,1350 @@ +# pylint: disable=line-too-long,useless-suppression,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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=protected-access, broad-except + +import copy +import calendar +import decimal +import functools +import sys +import logging +import base64 +import re +import typing +import enum +import email.utils +from datetime import datetime, date, time, timedelta, timezone +from json import JSONEncoder +import xml.etree.ElementTree as ET +from collections.abc import MutableMapping +from typing_extensions import Self +import isodate +from azure.core.exceptions import DeserializationError +from azure.core import CaseInsensitiveEnumMeta +from azure.core.pipeline import PipelineResponse +from azure.core.serialization import _Null +from azure.core.rest import HttpResponse + +_LOGGER = logging.getLogger(__name__) + +__all__ = ["SdkJSONEncoder", "Model", "rest_field", "rest_discriminator"] + +TZ_UTC = timezone.utc +_T = typing.TypeVar("_T") +_NONE_TYPE = type(None) + + +def _timedelta_as_isostr(td: timedelta) -> str: + """Converts a datetime.timedelta object into an ISO 8601 formatted string, e.g. 'P4DT12H30M05S' + + Function adapted from the Tin Can Python project: https://github.com/RusticiSoftware/TinCanPython + + :param timedelta td: The timedelta to convert + :rtype: str + :return: ISO8601 version of this timedelta + """ + + # Split seconds to larger units + seconds = td.total_seconds() + minutes, seconds = divmod(seconds, 60) + hours, minutes = divmod(minutes, 60) + days, hours = divmod(hours, 24) + + days, hours, minutes = list(map(int, (days, hours, minutes))) + seconds = round(seconds, 6) + + # Build date + date_str = "" + if days: + date_str = "%sD" % days + + if hours or minutes or seconds: + # Build time + time_str = "T" + + # Hours + bigger_exists = date_str or hours + if bigger_exists: + time_str += "{:02}H".format(hours) + + # Minutes + bigger_exists = bigger_exists or minutes + if bigger_exists: + time_str += "{:02}M".format(minutes) + + # Seconds + try: + if seconds.is_integer(): + seconds_string = "{:02}".format(int(seconds)) + else: + # 9 chars long w/ leading 0, 6 digits after decimal + seconds_string = "%09.6f" % seconds + # Remove trailing zeros + seconds_string = seconds_string.rstrip("0") + except AttributeError: # int.is_integer() raises + seconds_string = "{:02}".format(seconds) + + time_str += "{}S".format(seconds_string) + else: + time_str = "" + + return "P" + date_str + time_str + + +def _serialize_bytes(o, format: typing.Optional[str] = None) -> str: + encoded = base64.b64encode(o).decode() + if format == "base64url": + return encoded.strip("=").replace("+", "-").replace("/", "_") + return encoded + + +def _serialize_datetime(o, format: typing.Optional[str] = None): + if hasattr(o, "year") and hasattr(o, "hour"): + if format == "rfc7231": + return email.utils.format_datetime(o, usegmt=True) + if format == "unix-timestamp": + return int(calendar.timegm(o.utctimetuple())) + + # astimezone() fails for naive times in Python 2.7, so make make sure o is aware (tzinfo is set) + if not o.tzinfo: + iso_formatted = o.replace(tzinfo=TZ_UTC).isoformat() + else: + iso_formatted = o.astimezone(TZ_UTC).isoformat() + # Replace the trailing "+00:00" UTC offset with "Z" (RFC 3339: https://www.ietf.org/rfc/rfc3339.txt) + return iso_formatted.replace("+00:00", "Z") + # Next try datetime.date or datetime.time + return o.isoformat() + + +def _is_readonly(p): + try: + return p._visibility == ["read"] + except AttributeError: + return False + + +class SdkJSONEncoder(JSONEncoder): + """A JSON encoder that's capable of serializing datetime objects and bytes.""" + + def __init__(self, *args, exclude_readonly: bool = False, format: typing.Optional[str] = None, **kwargs): + super().__init__(*args, **kwargs) + self.exclude_readonly = exclude_readonly + self.format = format + + def default(self, o): # pylint: disable=too-many-return-statements + if _is_model(o): + if self.exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + return {k: v for k, v in o.items() if k not in readonly_props} + return dict(o.items()) + try: + return super(SdkJSONEncoder, self).default(o) + except TypeError: + if isinstance(o, _Null): + return None + if isinstance(o, decimal.Decimal): + return float(o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, self.format) + try: + # First try datetime.datetime + return _serialize_datetime(o, self.format) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _timedelta_as_isostr(o) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return super(SdkJSONEncoder, self).default(o) + + +_VALID_DATE = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" + r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") +_VALID_RFC7231 = re.compile( + r"(Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s\d{2}\s" + r"(Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\s\d{4}\s\d{2}:\d{2}:\d{2}\sGMT" +) + +_ARRAY_ENCODE_MAPPING = { + "pipeDelimited": "|", + "spaceDelimited": " ", + "commaDelimited": ",", + "newlineDelimited": "\n", +} + + +def _deserialize_array_encoded(delimit: str, attr): + if isinstance(attr, str): + if attr == "": + return [] + return attr.split(delimit) + return attr + + +def _deserialize_datetime(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize ISO-8601 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + attr = attr.upper() + match = _VALID_DATE.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + check_decimal = attr.split(".") + if len(check_decimal) > 1: + decimal_str = "" + for digit in check_decimal[1]: + if digit.isdigit(): + decimal_str += digit + else: + break + if len(decimal_str) > 6: + attr = attr.replace(decimal_str, decimal_str[0:6]) + + date_obj = isodate.parse_datetime(attr) + test_utc = date_obj.utctimetuple() + if test_utc.tm_year > 9999 or test_utc.tm_year < 1: + raise OverflowError("Hit max or min date") + return date_obj # type: ignore[no-any-return] + + +def _deserialize_datetime_rfc7231(attr: typing.Union[str, datetime]) -> datetime: + """Deserialize RFC7231 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + match = _VALID_RFC7231.match(attr) + if not match: + raise ValueError("Invalid datetime string: " + attr) + + return email.utils.parsedate_to_datetime(attr) + + +def _deserialize_datetime_unix_timestamp(attr: typing.Union[float, datetime]) -> datetime: + """Deserialize unix timestamp into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: ~datetime.datetime + :returns: The datetime object from that input + """ + if isinstance(attr, datetime): + # i'm already deserialized + return attr + return datetime.fromtimestamp(attr, TZ_UTC) + + +def _deserialize_date(attr: typing.Union[str, date]) -> date: + """Deserialize ISO-8601 formatted string into Date object. + :param str attr: response string to be deserialized. + :rtype: date + :returns: The date object from that input + """ + # This must NOT use defaultmonth/defaultday. Using None ensure this raises an exception. + if isinstance(attr, date): + return attr + return isodate.parse_date(attr, defaultmonth=None, defaultday=None) # type: ignore + + +def _deserialize_time(attr: typing.Union[str, time]) -> time: + """Deserialize ISO-8601 formatted string into time object. + + :param str attr: response string to be deserialized. + :rtype: datetime.time + :returns: The time object from that input + """ + if isinstance(attr, time): + return attr + return isodate.parse_time(attr) # type: ignore[no-any-return] + + +def _deserialize_bytes(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + return bytes(base64.b64decode(attr)) + + +def _deserialize_bytes_base64(attr): + if isinstance(attr, (bytes, bytearray)): + return attr + padding = "=" * (3 - (len(attr) + 3) % 4) # type: ignore + attr = attr + padding # type: ignore + encoded = attr.replace("-", "+").replace("_", "/") + return bytes(base64.b64decode(encoded)) + + +def _deserialize_duration(attr): + if isinstance(attr, timedelta): + return attr + return isodate.parse_duration(attr) + + +def _deserialize_decimal(attr): + if isinstance(attr, decimal.Decimal): + return attr + return decimal.Decimal(str(attr)) + + +def _deserialize_int_as_str(attr): + if isinstance(attr, int): + return attr + return int(attr) + + +_DESERIALIZE_MAPPING = { + datetime: _deserialize_datetime, + date: _deserialize_date, + time: _deserialize_time, + bytes: _deserialize_bytes, + bytearray: _deserialize_bytes, + timedelta: _deserialize_duration, + typing.Any: lambda x: x, + decimal.Decimal: _deserialize_decimal, +} + +_DESERIALIZE_MAPPING_WITHFORMAT = { + "rfc3339": _deserialize_datetime, + "rfc7231": _deserialize_datetime_rfc7231, + "unix-timestamp": _deserialize_datetime_unix_timestamp, + "base64": _deserialize_bytes, + "base64url": _deserialize_bytes_base64, +} + + +def get_deserializer(annotation: typing.Any, rf: typing.Optional["_RestField"] = None): + if annotation is int and rf and rf._format == "str": + return _deserialize_int_as_str + if annotation is str and rf and rf._format in _ARRAY_ENCODE_MAPPING: + return functools.partial(_deserialize_array_encoded, _ARRAY_ENCODE_MAPPING[rf._format]) + if rf and rf._format: + return _DESERIALIZE_MAPPING_WITHFORMAT.get(rf._format) + return _DESERIALIZE_MAPPING.get(annotation) # pyright: ignore + + +def _get_type_alias_type(module_name: str, alias_name: str): + types = { + k: v + for k, v in sys.modules[module_name].__dict__.items() + if isinstance(v, typing._GenericAlias) # type: ignore + } + if alias_name not in types: + return alias_name + return types[alias_name] + + +def _get_model(module_name: str, model_name: str): + models = {k: v for k, v in sys.modules[module_name].__dict__.items() if isinstance(v, type)} + module_end = module_name.rsplit(".", 1)[0] + models.update({k: v for k, v in sys.modules[module_end].__dict__.items() if isinstance(v, type)}) + if isinstance(model_name, str): + model_name = model_name.split(".")[-1] + if model_name not in models: + return model_name + return models[model_name] + + +_UNSET = object() + + +class _MyMutableMapping(MutableMapping[str, typing.Any]): + def __init__(self, data: dict[str, typing.Any]) -> None: + self._data = data + + def __contains__(self, key: typing.Any) -> bool: + return key in self._data + + def __getitem__(self, key: str) -> typing.Any: + # If this key has been deserialized (for mutable types), we need to handle serialization + if hasattr(self, "_attr_to_rest_field"): + cache_attr = f"_deserialized_{key}" + if hasattr(self, cache_attr): + rf = _get_rest_field(getattr(self, "_attr_to_rest_field"), key) + if rf: + value = self._data.get(key) + if isinstance(value, (dict, list, set)): + # For mutable types, serialize and return + # But also update _data with serialized form and clear flag + # so mutations via this returned value affect _data + serialized = _serialize(value, rf._format) + # If serialized form is same type (no transformation needed), + # return _data directly so mutations work + if isinstance(serialized, type(value)) and serialized == value: + return self._data.get(key) + # Otherwise return serialized copy and clear flag + try: + object.__delattr__(self, cache_attr) + except AttributeError: + pass + # Store serialized form back + self._data[key] = serialized + return serialized + return self._data.__getitem__(key) + + def __setitem__(self, key: str, value: typing.Any) -> None: + # Clear any cached deserialized value when setting through dictionary access + cache_attr = f"_deserialized_{key}" + try: + object.__delattr__(self, cache_attr) + except AttributeError: + pass + self._data.__setitem__(key, value) + + def __delitem__(self, key: str) -> None: + self._data.__delitem__(key) + + def __iter__(self) -> typing.Iterator[typing.Any]: + return self._data.__iter__() + + def __len__(self) -> int: + return self._data.__len__() + + def __ne__(self, other: typing.Any) -> bool: + return not self.__eq__(other) + + def keys(self) -> typing.KeysView[str]: + """ + :returns: a set-like object providing a view on D's keys + :rtype: ~typing.KeysView + """ + return self._data.keys() + + def values(self) -> typing.ValuesView[typing.Any]: + """ + :returns: an object providing a view on D's values + :rtype: ~typing.ValuesView + """ + return self._data.values() + + def items(self) -> typing.ItemsView[str, typing.Any]: + """ + :returns: set-like object providing a view on D's items + :rtype: ~typing.ItemsView + """ + return self._data.items() + + def get(self, key: str, default: typing.Any = None) -> typing.Any: + """ + Get the value for key if key is in the dictionary, else default. + :param str key: The key to look up. + :param any default: The value to return if key is not in the dictionary. Defaults to None + :returns: D[k] if k in D, else d. + :rtype: any + """ + try: + return self[key] + except KeyError: + return default + + @typing.overload + def pop(self, key: str) -> typing.Any: ... # pylint: disable=arguments-differ + + @typing.overload + def pop(self, key: str, default: _T) -> _T: ... # pylint: disable=signature-differs + + @typing.overload + def pop(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def pop(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Removes specified key and return the corresponding value. + :param str key: The key to pop. + :param any default: The value to return if key is not in the dictionary + :returns: The value corresponding to the key. + :rtype: any + :raises KeyError: If key is not found and default is not given. + """ + if default is _UNSET: + return self._data.pop(key) + return self._data.pop(key, default) + + def popitem(self) -> tuple[str, typing.Any]: + """ + Removes and returns some (key, value) pair + :returns: The (key, value) pair. + :rtype: tuple + :raises KeyError: if D is empty. + """ + return self._data.popitem() + + def clear(self) -> None: + """ + Remove all items from D. + """ + self._data.clear() + + def update(self, *args: typing.Any, **kwargs: typing.Any) -> None: # pylint: disable=arguments-differ + """ + Updates D from mapping/iterable E and F. + :param any args: Either a mapping object or an iterable of key-value pairs. + """ + self._data.update(*args, **kwargs) + + @typing.overload + def setdefault(self, key: str, default: None = None) -> None: ... + + @typing.overload + def setdefault(self, key: str, default: typing.Any) -> typing.Any: ... # pylint: disable=signature-differs + + def setdefault(self, key: str, default: typing.Any = _UNSET) -> typing.Any: + """ + Same as calling D.get(k, d), and setting D[k]=d if k not found + :param str key: The key to look up. + :param any default: The value to set if key is not in the dictionary + :returns: D[k] if k in D, else d. + :rtype: any + """ + if default is _UNSET: + return self._data.setdefault(key) + return self._data.setdefault(key, default) + + def __eq__(self, other: typing.Any) -> bool: + if isinstance(other, _MyMutableMapping): + return self._data == other._data + try: + other_model = self.__class__(other) + except Exception: + return False + return self._data == other_model._data + + def __repr__(self) -> str: + return str(self._data) + + +def _is_model(obj: typing.Any) -> bool: + return getattr(obj, "_is_model", False) + + +def _serialize(o, format: typing.Optional[str] = None): # pylint: disable=too-many-return-statements + if isinstance(o, list): + if format in _ARRAY_ENCODE_MAPPING and all(isinstance(x, str) for x in o): + return _ARRAY_ENCODE_MAPPING[format].join(o) + return [_serialize(x, format) for x in o] + if isinstance(o, dict): + return {k: _serialize(v, format) for k, v in o.items()} + if isinstance(o, set): + return {_serialize(x, format) for x in o} + if isinstance(o, tuple): + return tuple(_serialize(x, format) for x in o) + if isinstance(o, (bytes, bytearray)): + return _serialize_bytes(o, format) + if isinstance(o, decimal.Decimal): + return float(o) + if isinstance(o, enum.Enum): + return o.value + if isinstance(o, int): + if format == "str": + return str(o) + return o + try: + # First try datetime.datetime + return _serialize_datetime(o, format) + except AttributeError: + pass + # Last, try datetime.timedelta + try: + return _timedelta_as_isostr(o) + except AttributeError: + # This will be raised when it hits value.total_seconds in the method above + pass + return o + + +def _get_rest_field(attr_to_rest_field: dict[str, "_RestField"], rest_name: str) -> typing.Optional["_RestField"]: + try: + return next(rf for rf in attr_to_rest_field.values() if rf._rest_name == rest_name) + except StopIteration: + return None + + +def _create_value(rf: typing.Optional["_RestField"], value: typing.Any) -> typing.Any: + if not rf: + return _serialize(value, None) + if rf._is_multipart_file_input: + return value + if rf._is_model: + return _deserialize(rf._type, value) + if isinstance(value, ET.Element): + value = _deserialize(rf._type, value) + return _serialize(value, rf._format) + + +class Model(_MyMutableMapping): + _is_model = True + # label whether current class's _attr_to_rest_field has been calculated + # could not see _attr_to_rest_field directly because subclass inherits it from parent class + _calculated: set[str] = set() + + def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: + class_name = self.__class__.__name__ + if len(args) > 1: + raise TypeError(f"{class_name}.__init__() takes 2 positional arguments but {len(args) + 1} were given") + dict_to_pass = { + rest_field._rest_name: rest_field._default + for rest_field in self._attr_to_rest_field.values() + if rest_field._default is not _UNSET + } + if args: # pylint: disable=too-many-nested-blocks + if isinstance(args[0], ET.Element): + existed_attr_keys = [] + model_meta = getattr(self, "_xml", {}) + + for rf in self._attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and args[0].get(xml_name) is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + # unwrapped array could either use prop items meta/prop meta + if prop_meta.get("itemsName"): + xml_name = prop_meta.get("itemsName") + xml_ns = prop_meta.get("itemNs") + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = args[0].findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, items) + elif not rf._is_optional: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = [] + continue + + # text element is primitive type + if prop_meta.get("text", False): + if args[0].text is not None: + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].text) + continue + + # wrapped element could be normal property or array, it should only have one element + item = args[0].find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in args[0]: + if e.tag not in existed_attr_keys: + dict_to_pass[e.tag] = _convert_element(e) + else: + dict_to_pass.update( + {k: _create_value(_get_rest_field(self._attr_to_rest_field, k), v) for k, v in args[0].items()} + ) + else: + non_attr_kwargs = [k for k in kwargs if k not in self._attr_to_rest_field] + if non_attr_kwargs: + # actual type errors only throw the first wrong keyword arg they see, so following that. + raise TypeError(f"{class_name}.__init__() got an unexpected keyword argument '{non_attr_kwargs[0]}'") + dict_to_pass.update( + { + self._attr_to_rest_field[k]._rest_name: _create_value(self._attr_to_rest_field[k], v) + for k, v in kwargs.items() + if v is not None + } + ) + super().__init__(dict_to_pass) + + def copy(self) -> "Model": + return Model(self.__dict__) + + def __new__(cls, *args: typing.Any, **kwargs: typing.Any) -> Self: + if f"{cls.__module__}.{cls.__qualname__}" not in cls._calculated: + # we know the last nine classes in mro are going to be 'Model', '_MyMutableMapping', 'MutableMapping', + # 'Mapping', 'Collection', 'Sized', 'Iterable', 'Container' and 'object' + mros = cls.__mro__[:-9][::-1] # ignore parents, and reverse the mro order + attr_to_rest_field: dict[str, _RestField] = { # map attribute name to rest_field property + k: v for mro_class in mros for k, v in mro_class.__dict__.items() if k[0] != "_" and hasattr(v, "_type") + } + annotations = { + k: v + for mro_class in mros + if hasattr(mro_class, "__annotations__") + for k, v in mro_class.__annotations__.items() + } + for attr, rf in attr_to_rest_field.items(): + rf._module = cls.__module__ + if not rf._type: + rf._type = rf._get_deserialize_callable_from_annotation(annotations.get(attr, None)) + if not rf._rest_name_input: + rf._rest_name_input = attr + cls._attr_to_rest_field: dict[str, _RestField] = dict(attr_to_rest_field.items()) + cls._calculated.add(f"{cls.__module__}.{cls.__qualname__}") + + return super().__new__(cls) + + def __init_subclass__(cls, discriminator: typing.Optional[str] = None) -> None: + for base in cls.__bases__: + if hasattr(base, "__mapping__"): + base.__mapping__[discriminator or cls.__name__] = cls # type: ignore + + @classmethod + def _get_discriminator(cls, exist_discriminators) -> typing.Optional["_RestField"]: + for v in cls.__dict__.values(): + if isinstance(v, _RestField) and v._is_discriminator and v._rest_name not in exist_discriminators: + return v + return None + + @classmethod + def _deserialize(cls, data, exist_discriminators): + if not hasattr(cls, "__mapping__"): + return cls(data) + discriminator = cls._get_discriminator(exist_discriminators) + if discriminator is None: + return cls(data) + exist_discriminators.append(discriminator._rest_name) + if isinstance(data, ET.Element): + model_meta = getattr(cls, "_xml", {}) + prop_meta = getattr(discriminator, "_xml", {}) + xml_name = prop_meta.get("name", discriminator._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + if data.get(xml_name) is not None: + discriminator_value = data.get(xml_name) + else: + discriminator_value = data.find(xml_name).text # pyright: ignore + else: + discriminator_value = data.get(discriminator._rest_name) + mapped_cls = cls.__mapping__.get(discriminator_value, cls) # pyright: ignore # pylint: disable=no-member + return mapped_cls._deserialize(data, exist_discriminators) + + def as_dict(self, *, exclude_readonly: bool = False) -> dict[str, typing.Any]: + """Return a dict that can be turned into json using json.dump. + + :keyword bool exclude_readonly: Whether to remove the readonly properties. + :returns: A dict JSON compatible object + :rtype: dict + """ + + result = {} + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in self._attr_to_rest_field.values() if _is_readonly(p)] + for k, v in self.items(): + if exclude_readonly and k in readonly_props: # pyright: ignore + continue + is_multipart_file_input = False + try: + is_multipart_file_input = next( + rf for rf in self._attr_to_rest_field.values() if rf._rest_name == k + )._is_multipart_file_input + except StopIteration: + pass + result[k] = v if is_multipart_file_input else Model._as_dict_value(v, exclude_readonly=exclude_readonly) + return result + + @staticmethod + def _as_dict_value(v: typing.Any, exclude_readonly: bool = False) -> typing.Any: + if v is None or isinstance(v, _Null): + return None + if isinstance(v, (list, tuple, set)): + return type(v)(Model._as_dict_value(x, exclude_readonly=exclude_readonly) for x in v) + if isinstance(v, dict): + return {dk: Model._as_dict_value(dv, exclude_readonly=exclude_readonly) for dk, dv in v.items()} + return v.as_dict(exclude_readonly=exclude_readonly) if hasattr(v, "as_dict") else v + + +def _deserialize_model(model_deserializer: typing.Optional[typing.Callable], obj): + if _is_model(obj): + return obj + return _deserialize(model_deserializer, obj) + + +def _deserialize_with_optional(if_obj_deserializer: typing.Optional[typing.Callable], obj): + if obj is None: + return obj + return _deserialize_with_callable(if_obj_deserializer, obj) + + +def _deserialize_with_union(deserializers, obj): + for deserializer in deserializers: + try: + return _deserialize(deserializer, obj) + except DeserializationError: + pass + raise DeserializationError() + + +def _deserialize_dict( + value_deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj: dict[typing.Any, typing.Any], +): + if obj is None: + return obj + if isinstance(obj, ET.Element): + obj = {child.tag: child for child in obj} + return {k: _deserialize(value_deserializer, v, module) for k, v in obj.items()} + + +def _deserialize_multiple_sequence( + entry_deserializers: list[typing.Optional[typing.Callable]], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + return type(obj)(_deserialize(deserializer, entry, module) for entry, deserializer in zip(obj, entry_deserializers)) + + +def _is_array_encoded_deserializer(deserializer: functools.partial) -> bool: + return ( + isinstance(deserializer, functools.partial) + and isinstance(deserializer.args[0], functools.partial) + and deserializer.args[0].func == _deserialize_array_encoded # pylint: disable=comparison-with-callable + ) + + +def _deserialize_sequence( + deserializer: typing.Optional[typing.Callable], + module: typing.Optional[str], + obj, +): + if obj is None: + return obj + if isinstance(obj, ET.Element): + obj = list(obj) + + # encoded string may be deserialized to sequence + if isinstance(obj, str) and isinstance(deserializer, functools.partial): + # for list[str] + if _is_array_encoded_deserializer(deserializer): + return deserializer(obj) + + # for list[Union[...]] + if isinstance(deserializer.args[0], list): + for sub_deserializer in deserializer.args[0]: + if _is_array_encoded_deserializer(sub_deserializer): + return sub_deserializer(obj) + + return type(obj)(_deserialize(deserializer, entry, module) for entry in obj) + + +def _sorted_annotations(types: list[typing.Any]) -> list[typing.Any]: + return sorted( + types, + key=lambda x: hasattr(x, "__name__") and x.__name__.lower() in ("str", "float", "int", "bool"), + ) + + +def _get_deserialize_callable_from_annotation( # pylint: disable=too-many-return-statements, too-many-statements, too-many-branches + annotation: typing.Any, + module: typing.Optional[str], + rf: typing.Optional["_RestField"] = None, +) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + if not annotation: + return None + + # is it a type alias? + if isinstance(annotation, str): + if module is not None: + annotation = _get_type_alias_type(module, annotation) + + # is it a forward ref / in quotes? + if isinstance(annotation, (str, typing.ForwardRef)): + try: + model_name = annotation.__forward_arg__ # type: ignore + except AttributeError: + model_name = annotation + if module is not None: + annotation = _get_model(module, model_name) # type: ignore + + try: + if module and _is_model(annotation): + if rf: + rf._is_model = True + + return functools.partial(_deserialize_model, annotation) # pyright: ignore + except Exception: + pass + + # is it a literal? + try: + if annotation.__origin__ is typing.Literal: # pyright: ignore + return None + except AttributeError: + pass + + # is it optional? + try: + if any(a is _NONE_TYPE for a in annotation.__args__): # pyright: ignore + if rf: + rf._is_optional = True + if len(annotation.__args__) <= 2: # pyright: ignore + if_obj_deserializer = _get_deserialize_callable_from_annotation( + next(a for a in annotation.__args__ if a is not _NONE_TYPE), module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_with_optional, if_obj_deserializer) + # the type is Optional[Union[...]], we need to remove the None type from the Union + annotation_copy = copy.copy(annotation) + annotation_copy.__args__ = [a for a in annotation_copy.__args__ if a is not _NONE_TYPE] # pyright: ignore + return _get_deserialize_callable_from_annotation(annotation_copy, module, rf) + except AttributeError: + pass + + # is it union? + if getattr(annotation, "__origin__", None) is typing.Union: + # initial ordering is we make `string` the last deserialization option, because it is often them most generic + deserializers = [ + _get_deserialize_callable_from_annotation(arg, module, rf) + for arg in _sorted_annotations(annotation.__args__) # pyright: ignore + ] + + return functools.partial(_deserialize_with_union, deserializers) + + try: + annotation_name = ( + annotation.__name__ if hasattr(annotation, "__name__") else annotation._name # pyright: ignore + ) + if annotation_name.lower() == "dict": + value_deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[1], module, rf # pyright: ignore + ) + + return functools.partial( + _deserialize_dict, + value_deserializer, + module, + ) + except (AttributeError, IndexError): + pass + try: + annotation_name = ( + annotation.__name__ if hasattr(annotation, "__name__") else annotation._name # pyright: ignore + ) + if annotation_name.lower() in ["list", "set", "tuple", "sequence"]: + if len(annotation.__args__) > 1: # pyright: ignore + entry_deserializers = [ + _get_deserialize_callable_from_annotation(dt, module, rf) + for dt in annotation.__args__ # pyright: ignore + ] + return functools.partial(_deserialize_multiple_sequence, entry_deserializers, module) + deserializer = _get_deserialize_callable_from_annotation( + annotation.__args__[0], module, rf # pyright: ignore + ) + + return functools.partial(_deserialize_sequence, deserializer, module) + except (TypeError, IndexError, AttributeError, SyntaxError): + pass + + def _deserialize_default( + deserializer, + obj, + ): + if obj is None: + return obj + try: + return _deserialize_with_callable(deserializer, obj) + except Exception: + pass + return obj + + if get_deserializer(annotation, rf): + return functools.partial(_deserialize_default, get_deserializer(annotation, rf)) + + return functools.partial(_deserialize_default, annotation) + + +def _deserialize_with_callable( + deserializer: typing.Optional[typing.Callable[[typing.Any], typing.Any]], + value: typing.Any, +): # pylint: disable=too-many-return-statements + try: + if value is None or isinstance(value, _Null): + return None + if isinstance(value, ET.Element): + if deserializer is str: + return value.text or "" + if deserializer is int: + return int(value.text) if value.text else None + if deserializer is float: + return float(value.text) if value.text else None + if deserializer is bool: + return value.text == "true" if value.text else None + if deserializer and deserializer in _DESERIALIZE_MAPPING.values(): + return deserializer(value.text) if value.text else None + if deserializer and deserializer in _DESERIALIZE_MAPPING_WITHFORMAT.values(): + return deserializer(value.text) if value.text else None + if deserializer is None: + return value + if deserializer in [int, float, bool]: + return deserializer(value) + if isinstance(deserializer, CaseInsensitiveEnumMeta): + try: + return deserializer(value.text if isinstance(value, ET.Element) else value) + except ValueError: + # for unknown value, return raw value + return value.text if isinstance(value, ET.Element) else value + if isinstance(deserializer, type) and issubclass(deserializer, Model): + return deserializer._deserialize(value, []) + return typing.cast(typing.Callable[[typing.Any], typing.Any], deserializer)(value) + except Exception as e: + raise DeserializationError() from e + + +def _deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + if isinstance(value, PipelineResponse): + value = value.http_response.json() + if rf is None and format: + rf = _RestField(format=format) + if not isinstance(deserializer, functools.partial): + deserializer = _get_deserialize_callable_from_annotation(deserializer, module, rf) + return _deserialize_with_callable(deserializer, value) + + +def _failsafe_deserialize( + deserializer: typing.Any, + response: HttpResponse, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, response.json(), module, rf, format) + except Exception: # pylint: disable=broad-except + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +def _failsafe_deserialize_xml( + deserializer: typing.Any, + response: HttpResponse, +) -> typing.Any: + try: + return _deserialize_xml(deserializer, response.text()) + except Exception: # pylint: disable=broad-except + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +# pylint: disable=too-many-instance-attributes +class _RestField: + def __init__( + self, + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + is_discriminator: bool = False, + visibility: typing.Optional[list[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[dict[str, typing.Any]] = None, + ): + self._type = type + self._rest_name_input = name + self._module: typing.Optional[str] = None + self._is_discriminator = is_discriminator + self._visibility = visibility + self._is_model = False + self._is_optional = False + self._default = default + self._format = format + self._is_multipart_file_input = is_multipart_file_input + self._xml = xml if xml is not None else {} + + @property + def _class_type(self) -> typing.Any: + result = getattr(self._type, "args", [None])[0] + # type may be wrapped by nested functools.partial so we need to check for that + if isinstance(result, functools.partial): + return getattr(result, "args", [None])[0] + return result + + @property + def _rest_name(self) -> str: + if self._rest_name_input is None: + raise ValueError("Rest name was never set") + return self._rest_name_input + + def __get__(self, obj: Model, type=None): # pylint: disable=redefined-builtin + # by this point, type and rest_name will have a value bc we default + # them in __new__ of the Model class + # Use _data.get() directly to avoid triggering __getitem__ which clears the cache + item = obj._data.get(self._rest_name) + if item is None: + return item + if self._is_model: + return item + + # For mutable types, we want mutations to directly affect _data + # Check if we've already deserialized this value + cache_attr = f"_deserialized_{self._rest_name}" + if hasattr(obj, cache_attr): + # Return the value from _data directly (it's been deserialized in place) + return obj._data.get(self._rest_name) + + deserialized = _deserialize(self._type, _serialize(item, self._format), rf=self) + + # For mutable types, store the deserialized value back in _data + # so mutations directly affect _data + if isinstance(deserialized, (dict, list, set)): + obj._data[self._rest_name] = deserialized + object.__setattr__(obj, cache_attr, True) # Mark as deserialized + return deserialized + + return deserialized + + def __set__(self, obj: Model, value) -> None: + # Clear the cached deserialized object when setting a new value + cache_attr = f"_deserialized_{self._rest_name}" + if hasattr(obj, cache_attr): + object.__delattr__(obj, cache_attr) + + if value is None: + # we want to wipe out entries if users set attr to None + try: + obj.__delitem__(self._rest_name) + except KeyError: + pass + return + if self._is_model: + if not _is_model(value): + value = _deserialize(self._type, value) + obj.__setitem__(self._rest_name, value) + return + obj.__setitem__(self._rest_name, _serialize(value, self._format)) + + def _get_deserialize_callable_from_annotation( + self, annotation: typing.Any + ) -> typing.Optional[typing.Callable[[typing.Any], typing.Any]]: + return _get_deserialize_callable_from_annotation(annotation, self._module, self) + + +def rest_field( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[list[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField( + name=name, + type=type, + visibility=visibility, + default=default, + format=format, + is_multipart_file_input=is_multipart_file_input, + xml=xml, + ) + + +def rest_discriminator( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[list[str]] = None, + xml: typing.Optional[dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField(name=name, type=type, is_discriminator=True, visibility=visibility, xml=xml) + + +def serialize_xml(model: Model, exclude_readonly: bool = False) -> str: + """Serialize a model to XML. + + :param Model model: The model to serialize. + :param bool exclude_readonly: Whether to exclude readonly properties. + :returns: The XML representation of the model. + :rtype: str + """ + return ET.tostring(_get_element(model, exclude_readonly), encoding="unicode") # type: ignore + + +def _get_element( + o: typing.Any, + exclude_readonly: bool = False, + parent_meta: typing.Optional[dict[str, typing.Any]] = None, + wrapped_element: typing.Optional[ET.Element] = None, +) -> typing.Union[ET.Element, list[ET.Element]]: + if _is_model(o): + model_meta = getattr(o, "_xml", {}) + + # if prop is a model, then use the prop element directly, else generate a wrapper of model + if wrapped_element is None: + wrapped_element = _create_xml_element( + model_meta.get("name", o.__class__.__name__), + model_meta.get("prefix"), + model_meta.get("ns"), + ) + + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + + for k, v in o.items(): + # do not serialize readonly properties + if exclude_readonly and k in readonly_props: + continue + + prop_rest_field = _get_rest_field(o._attr_to_rest_field, k) + if prop_rest_field: + prop_meta = getattr(prop_rest_field, "_xml").copy() + # use the wire name as xml name if no specific name is set + if prop_meta.get("name") is None: + prop_meta["name"] = k + else: + # additional properties will not have rest field, use the wire name as xml name + prop_meta = {"name": k} + + # if no ns for prop, use model's + if prop_meta.get("ns") is None and model_meta.get("ns"): + prop_meta["ns"] = model_meta.get("ns") + prop_meta["prefix"] = model_meta.get("prefix") + + if prop_meta.get("unwrapped", False): + # unwrapped could only set on array + wrapped_element.extend(_get_element(v, exclude_readonly, prop_meta)) + elif prop_meta.get("text", False): + # text could only set on primitive type + wrapped_element.text = _get_primitive_type_value(v) + elif prop_meta.get("attribute", False): + xml_name = prop_meta.get("name", k) + if prop_meta.get("ns"): + ET.register_namespace(prop_meta.get("prefix"), prop_meta.get("ns")) # pyright: ignore + xml_name = "{" + prop_meta.get("ns") + "}" + xml_name # pyright: ignore + # attribute should be primitive type + wrapped_element.set(xml_name, _get_primitive_type_value(v)) + else: + # other wrapped prop element + wrapped_element.append(_get_wrapped_element(v, exclude_readonly, prop_meta)) + return wrapped_element + if isinstance(o, list): + return [_get_element(x, exclude_readonly, parent_meta) for x in o] # type: ignore + if isinstance(o, dict): + result = [] + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": parent_meta.get("ns") if parent_meta else None, + "prefix": parent_meta.get("prefix") if parent_meta else None, + }, + ) + ) + return result + + # primitive case need to create element based on parent_meta + if parent_meta: + return _get_wrapped_element( + o, + exclude_readonly, + { + "name": parent_meta.get("itemsName", parent_meta.get("name")), + "prefix": parent_meta.get("itemsPrefix", parent_meta.get("prefix")), + "ns": parent_meta.get("itemsNs", parent_meta.get("ns")), + }, + ) + + raise ValueError("Could not serialize value into xml: " + o) + + +def _get_wrapped_element( + v: typing.Any, + exclude_readonly: bool, + meta: typing.Optional[dict[str, typing.Any]], +) -> ET.Element: + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, meta.get("ns") if meta else None + ) + if isinstance(v, (dict, list)): + wrapped_element.extend(_get_element(v, exclude_readonly, meta)) + elif _is_model(v): + _get_element(v, exclude_readonly, meta, wrapped_element) + else: + wrapped_element.text = _get_primitive_type_value(v) + return wrapped_element # type: ignore[no-any-return] + + +def _get_primitive_type_value(v) -> str: + if v is True: + return "true" + if v is False: + return "false" + if isinstance(v, _Null): + return "" + return str(v) + + +def _create_xml_element( + tag: typing.Any, prefix: typing.Optional[str] = None, ns: typing.Optional[str] = None +) -> ET.Element: + if prefix and ns: + ET.register_namespace(prefix, ns) + if ns: + return ET.Element("{" + ns + "}" + tag) + return ET.Element(tag) + + +def _deserialize_xml( + deserializer: typing.Any, + value: str, +) -> typing.Any: + element = ET.fromstring(value) # nosec + return _deserialize(deserializer, element) + + +def _convert_element(e: ET.Element): + # dict case + if len(e.attrib) > 0 or len({child.tag for child in e}) > 1: + dict_result: dict[str, typing.Any] = {} + for child in e: + if dict_result.get(child.tag) is not None: + if isinstance(dict_result[child.tag], list): + dict_result[child.tag].append(_convert_element(child)) + else: + dict_result[child.tag] = [dict_result[child.tag], _convert_element(child)] + else: + dict_result[child.tag] = _convert_element(child) + dict_result.update(e.attrib) + return dict_result + # array case + if len(e) > 0: + array_result: list[typing.Any] = [] + for child in e: + array_result.append(_convert_element(child)) + return array_result + # primitive case + return e.text diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/serialization.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/serialization.py index f5187701d7be..81ec1de5922b 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/serialization.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_utils/serialization.py @@ -3,7 +3,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -21,7 +21,6 @@ import sys import codecs from typing import ( - Dict, Any, cast, Optional, @@ -31,7 +30,6 @@ Mapping, Callable, MutableMapping, - List, ) try: @@ -229,12 +227,12 @@ class Model: serialization and deserialization. """ - _subtype_map: Dict[str, Dict[str, Any]] = {} - _attribute_map: Dict[str, Dict[str, Any]] = {} - _validation: Dict[str, Dict[str, Any]] = {} + _subtype_map: dict[str, dict[str, Any]] = {} + _attribute_map: dict[str, dict[str, Any]] = {} + _validation: dict[str, dict[str, Any]] = {} def __init__(self, **kwargs: Any) -> None: - self.additional_properties: Optional[Dict[str, Any]] = {} + self.additional_properties: Optional[dict[str, Any]] = {} for k in kwargs: # pylint: disable=consider-using-dict-items if k not in self._attribute_map: _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) @@ -311,7 +309,7 @@ def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: def as_dict( self, keep_readonly: bool = True, - key_transformer: Callable[[str, Dict[str, Any], Any], Any] = attribute_transformer, + key_transformer: Callable[[str, dict[str, Any], Any], Any] = attribute_transformer, **kwargs: Any ) -> JSON: """Return a dict that can be serialized using json.dump. @@ -380,7 +378,7 @@ def deserialize(cls, data: Any, content_type: Optional[str] = None) -> Self: def from_dict( cls, data: Any, - key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, + key_extractors: Optional[Callable[[str, dict[str, Any], Any], Any]] = None, content_type: Optional[str] = None, ) -> Self: """Parse a dict using given key extractor return a model. @@ -414,7 +412,7 @@ def _flatten_subtype(cls, key, objects): return {} result = dict(cls._subtype_map[key]) for valuetype in cls._subtype_map[key].values(): - result.update(objects[valuetype]._flatten_subtype(key, objects)) # pylint: disable=protected-access + result |= objects[valuetype]._flatten_subtype(key, objects) # pylint: disable=protected-access return result @classmethod @@ -528,7 +526,7 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: "[]": self.serialize_iter, "{}": self.serialize_dict, } - self.dependencies: Dict[str, type] = dict(classes) if classes else {} + self.dependencies: dict[str, type] = dict(classes) if classes else {} self.key_transformer = full_restapi_key_transformer self.client_side_validation = True @@ -579,7 +577,7 @@ def _serialize( # pylint: disable=too-many-nested-blocks, too-many-branches, to if attr_name == "additional_properties" and attr_desc["key"] == "": if target_obj.additional_properties is not None: - serialized.update(target_obj.additional_properties) + serialized |= target_obj.additional_properties continue try: @@ -789,7 +787,7 @@ def serialize_data(self, data, data_type, **kwargs): # If dependencies is empty, try with current data class # It has to be a subclass of Enum anyway - enum_type = self.dependencies.get(data_type, data.__class__) + enum_type = self.dependencies.get(data_type, cast(type, data.__class__)) if issubclass(enum_type, Enum): return Serializer.serialize_enum(data, enum_obj=enum_type) @@ -823,13 +821,20 @@ def serialize_basic(cls, data, data_type, **kwargs): :param str data_type: Type of object in the iterable. :rtype: str, int, float, bool :return: serialized object + :raises TypeError: raise if data_type is not one of str, int, float, bool. """ custom_serializer = cls._get_custom_serializers(data_type, **kwargs) if custom_serializer: return custom_serializer(data) if data_type == "str": return cls.serialize_unicode(data) - return eval(data_type)(data) # nosec # pylint: disable=eval-used + if data_type == "int": + return int(data) + if data_type == "float": + return float(data) + if data_type == "bool": + return bool(data) + raise TypeError("Unknown basic data type: {}".format(data_type)) @classmethod def serialize_unicode(cls, data): @@ -1184,7 +1189,7 @@ def rest_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argumen while "." in key: # Need the cast, as for some reasons "split" is typed as list[str | Any] - dict_keys = cast(List[str], _FLATTEN.split(key)) + dict_keys = cast(list[str], _FLATTEN.split(key)) if len(dict_keys) == 1: key = _decode_attribute_map_key(dict_keys[0]) break @@ -1386,7 +1391,7 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: "duration": (isodate.Duration, datetime.timedelta), "iso-8601": (datetime.datetime), } - self.dependencies: Dict[str, type] = dict(classes) if classes else {} + self.dependencies: dict[str, type] = dict(classes) if classes else {} self.key_extractors = [rest_key_extractor, xml_key_extractor] # Additional properties only works if the "rest_key_extractor" is used to # extract the keys. Making it to work whatever the key extractor is too much @@ -1759,7 +1764,7 @@ def deserialize_basic(self, attr, data_type): # pylint: disable=too-many-return :param str data_type: deserialization data type. :return: Deserialized basic type. :rtype: str, int, float or bool - :raises TypeError: if string format is not valid. + :raises TypeError: if string format is not valid or data_type is not one of str, int, float, bool. """ # If we're here, data is supposed to be a basic type. # If it's still an XML node, take the text @@ -1785,7 +1790,11 @@ def deserialize_basic(self, attr, data_type): # pylint: disable=too-many-return if data_type == "str": return self.deserialize_unicode(attr) - return eval(data_type)(attr) # nosec # pylint: disable=eval-used + if data_type == "int": + return int(attr) + if data_type == "float": + return float(attr) + raise TypeError("Unknown basic data type: {}".format(data_type)) @staticmethod def deserialize_unicode(data): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_version.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_version.py index e5754a47ce68..bbcd28b4aa67 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_version.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/_version.py @@ -2,8 +2,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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.0.0b1" +VERSION = "1.0.0b2" diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/__init__.py index a403fe4a267a..83ab88ccbe8b 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/__init__.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._kubernetes_configuration_extensions_mgmt_client import KubernetesConfigurationExtensionsMgmtClient # type: ignore +from ._client import KubernetesConfigurationExtensionsMgmtClient # type: ignore try: from ._patch import __all__ as _patch_all diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_kubernetes_configuration_extensions_mgmt_client.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_client.py similarity index 73% rename from sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_kubernetes_configuration_extensions_mgmt_client.py rename to sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_client.py index 8588a477c3e0..925d708d57f6 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_kubernetes_configuration_extensions_mgmt_client.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_client.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -17,17 +17,17 @@ from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from .. import models as _models from .._utils.serialization import Deserializer, Serializer from ._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration from .operations import ExtensionsOperations, OperationStatusOperations if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials_async import AsyncTokenCredential class KubernetesConfigurationExtensionsMgmtClient: # pylint: disable=name-too-long - """KubernetesConfiguration Extensions Client. + """Use these APIs to create extension resources through ARM, for Kubernetes Clusters. :ivar extensions: ExtensionsOperations operations :vartype extensions: @@ -35,29 +35,45 @@ class KubernetesConfigurationExtensionsMgmtClient: # pylint: disable=name-too-l :ivar operation_status: OperationStatusOperations operations :vartype operation_status: azure.mgmt.kubernetesconfiguration.extensions.aio.operations.OperationStatusOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: The ID of the target subscription. Required. + :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is None. + :param base_url: Service host. Default value is None. :type base_url: str - :keyword api_version: Api Version. Default value is "2024-11-01". Note that overriding this - default value may result in unsupported behavior. + :keyword cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is + None. + :paramtype cloud_setting: ~azure.core.AzureClouds + :keyword api_version: The API version to use for this operation. Known values are "2024-11-01" + and None. Default value is "2024-11-01". 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 Retry-After header is present. """ def __init__( - self, credential: "AsyncTokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any + self, + credential: "AsyncTokenCredential", + subscription_id: str, + base_url: Optional[str] = None, + *, + cloud_setting: Optional["AzureClouds"] = None, + **kwargs: Any ) -> None: - _cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud # type: ignore + _endpoint = "{endpoint}" + _cloud = cloud_setting or settings.current.azure_cloud # type: ignore _endpoints = get_arm_endpoints(_cloud) if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) self._config = KubernetesConfigurationExtensionsMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, credential_scopes=credential_scopes, **kwargs + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + cloud_setting=cloud_setting, + credential_scopes=credential_scopes, + **kwargs ) _policies = kwargs.pop("policies", None) @@ -79,19 +95,18 @@ def __init__( self._config.http_logging_policy, ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient( - base_url=cast(str, base_url), policies=_policies, **kwargs + base_url=cast(str, _endpoint), policies=_policies, **kwargs ) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} - self._serialize = Serializer(client_models) - self._deserialize = Deserializer(client_models) + self._serialize = Serializer() + self._deserialize = Deserializer() self._serialize.client_side_validation = False self.extensions = ExtensionsOperations(self._client, self._config, self._serialize, self._deserialize) self.operation_status = OperationStatusOperations( self._client, self._config, self._serialize, self._deserialize ) - def _send_request( + def send_request( self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -99,7 +114,7 @@ def _send_request( >>> from azure.core.rest import HttpRequest >>> request = HttpRequest("GET", "https://www.example.org/") - >>> response = await client._send_request(request) + >>> response = await client.send_request(request) For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request @@ -112,7 +127,11 @@ def _send_request( """ request_copy = deepcopy(request) - request_copy.url = self._client.format_url(request_copy.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_configuration.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_configuration.py index a80905989478..a6b6f78b4ce4 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_configuration.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_configuration.py @@ -3,11 +3,11 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, TYPE_CHECKING +from typing import Any, Optional, TYPE_CHECKING from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -15,6 +15,7 @@ from .._version import VERSION if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials_async import AsyncTokenCredential @@ -24,16 +25,29 @@ class KubernetesConfigurationExtensionsMgmtClientConfiguration: # pylint: disab Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: The ID of the target subscription. Required. + :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 "2024-11-01". Note that overriding this - default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :param cloud_setting: The cloud setting for which to get the ARM endpoint. Default value is + None. + :type cloud_setting: ~azure.core.AzureClouds + :keyword api_version: The API version to use for this operation. Known values are "2024-11-01" + and None. Default value is "2024-11-01". 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: + def __init__( + self, + credential: "AsyncTokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + cloud_setting: Optional["AzureClouds"] = None, + **kwargs: Any + ) -> None: api_version: str = kwargs.pop("api_version", "2024-11-01") if credential is None: @@ -43,6 +57,8 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url + self.cloud_setting = cloud_setting self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-kubernetesconfiguration-extensions/{}".format(VERSION)) diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_patch.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_patch.py index 8bcb627aa475..87676c65a8f0 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_patch.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/_patch.py @@ -7,9 +7,9 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/__init__.py index a43141e42aa8..811e5748c43d 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -12,8 +12,8 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._extensions_operations import ExtensionsOperations # type: ignore -from ._operation_status_operations import OperationStatusOperations # type: ignore +from ._operations import ExtensionsOperations # type: ignore +from ._operations import OperationStatusOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operation_status_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operation_status_operations.py deleted file mode 100644 index f9b3f75c0939..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operation_status_operations.py +++ /dev/null @@ -1,133 +0,0 @@ -# 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 collections.abc import MutableMapping -from typing import Any, Callable, Dict, Optional, TypeVar - -from azure.core import AsyncPipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator_async import distributed_trace_async -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ..._utils.serialization import Deserializer, Serializer -from ...operations._operation_status_operations import build_get_request -from .._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class OperationStatusOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.kubernetesconfiguration.extensions.aio.KubernetesConfigurationExtensionsMgmtClient`'s - :attr:`operation_status` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: KubernetesConfigurationExtensionsMgmtClientConfiguration = ( - input_args.pop(0) if input_args else kwargs.pop("config") - ) - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace_async - async def get( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - operation_id: str, - **kwargs: Any - ) -> _models.OperationStatusResult: - """Get Async Operation status. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :param operation_id: operation Id. Required. - :type operation_id: str - :return: OperationStatusResult or the result of cls(response) - :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.OperationStatusResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _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.OperationStatusResult] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - operation_id=operation_id, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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) - - deserialized = self._deserialize("OperationStatusResult", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_extensions_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operations.py similarity index 64% rename from sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_extensions_operations.py rename to sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operations.py index 258104f3c9d2..9b033b45c509 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_extensions_operations.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_operations.py @@ -1,13 +1,15 @@ +# 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from collections.abc import MutableMapping from io import IOBase -from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, TypeVar, Union, cast, overload +import json +from typing import Any, AsyncIterator, Callable, IO, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core import AsyncPipelineClient @@ -32,18 +34,22 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models +from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ..._utils.serialization import Deserializer, Serializer -from ...operations._extensions_operations import ( - build_create_request, - build_delete_request, - build_get_request, - build_list_request, - build_update_request, +from ...operations._operations import ( + build_extensions_create_request, + build_extensions_delete_request, + build_extensions_get_request, + build_extensions_list_request, + build_extensions_update_request, + build_operation_status_get_request, ) from .._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration +JSON = MutableMapping[str, Any] T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] +List = list class ExtensionsOperations: @@ -56,8 +62,6 @@ class ExtensionsOperations: :attr:`extensions` attribute. """ - models = _models - def __init__(self, *args, **kwargs) -> None: input_args = list(args) self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") @@ -67,6 +71,95 @@ def __init__(self, *args, **kwargs) -> None: self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + @distributed_trace_async + async def get( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + **kwargs: Any + ) -> _models.Extension: + """Gets Kubernetes Cluster Extension. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :return: Extension. The Extension is compatible with MutableMapping + :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.Extension] = kwargs.pop("cls", None) + + _request = build_extensions_get_request( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.Extension, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + async def _create_initial( self, resource_group_name: str, @@ -74,7 +167,7 @@ async def _create_initial( cluster_resource_name: str, cluster_name: str, extension_name: str, - extension: Union[_models.Extension, IO[bytes]], + extension: Union[_models.Extension, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { @@ -86,35 +179,35 @@ async def _create_initial( error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" - _json = None _content = None if isinstance(extension, (IOBase, bytes)): _content = extension else: - _json = self._serialize.body(extension, "Extension") + _content = json.dumps(extension, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore - _request = build_create_request( + _request = build_extensions_create_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, subscription_id=self._config.subscription_id, - api_version=api_version, content_type=content_type, - json=_json, + api_version=self._config.api_version, content=_content, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _decompress = kwargs.pop("decompress", True) _stream = True @@ -130,13 +223,23 @@ async def _create_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) + response_headers = {} + if response.status_code == 201: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized # type: ignore @@ -173,8 +276,48 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + extension: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.Extension]: + """Create a new Kubernetes Cluster Extension. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param extension: Properties necessary to Create an Extension. Required. + :type extension: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -213,8 +356,8 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -228,7 +371,7 @@ async def begin_create( cluster_resource_name: str, cluster_name: str, extension_name: str, - extension: Union[_models.Extension, IO[bytes]], + extension: Union[_models.Extension, JSON, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Extension]: """Create a new Kubernetes Cluster Extension. @@ -246,19 +389,19 @@ async def begin_create( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param extension: Properties necessary to Create an Extension. Is either a Extension type or a - IO[bytes] type. Required. - :type extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension or IO[bytes] - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + :param extension: Properties necessary to Create an Extension. Is one of the following types: + Extension, JSON, IO[bytes] Required. + :type extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension or JSON or + IO[bytes] + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Extension] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) @@ -272,7 +415,6 @@ async def begin_create( cluster_name=cluster_name, extension_name=extension_name, extension=extension, - api_version=api_version, content_type=content_type, cls=lambda x, y, z: x, headers=_headers, @@ -283,15 +425,19 @@ async def begin_create( kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Extension", pipeline_response.http_response) + response = pipeline_response.http_response + deserialized = _deserialize(_models.Extension, response.json()) if cls: return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, - AsyncARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs), + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) @@ -308,89 +454,14 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) - @distributed_trace_async - async def get( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - **kwargs: Any - ) -> _models.Extension: - """Gets Kubernetes Cluster Extension. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :return: Extension or the result of cls(response) - :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _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.Extension] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _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) - - deserialized = self._deserialize("Extension", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _delete_initial( + async def _update_initial( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - force_delete: Optional[bool] = None, + patch_extension: Union[_models.PatchExtension, JSON, IO[bytes]], **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { @@ -401,25 +472,36 @@ async def _delete_initial( } error_map.update(kwargs.pop("error_map", {}) or {}) - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - _request = build_delete_request( + content_type = content_type or "application/json" + _content = None + if isinstance(patch_extension, (IOBase, bytes)): + _content = patch_extension + else: + _content = json.dumps(patch_extension, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_extensions_update_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, subscription_id=self._config.subscription_id, - force_delete=force_delete, - api_version=api_version, + content_type=content_type, + api_version=self._config.api_version, + content=_content, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _decompress = kwargs.pop("decompress", True) _stream = True @@ -429,35 +511,46 @@ async def _delete_initial( response = pipeline_response.http_response - if response.status_code not in [200, 202, 204]: + if response.status_code not in [200, 202]: try: await response.read() # Load the body in memory and close the socket except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) + response_headers = {} + if response.status_code == 202: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized # type: ignore - @distributed_trace_async - async def begin_delete( + @overload + async def begin_update( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - force_delete: Optional[bool] = None, + patch_extension: _models.PatchExtension, + *, + content_type: str = "application/json", **kwargs: Any - ) -> AsyncLROPoller[None]: - """Delete a Kubernetes Cluster Extension. This will cause the Agent to Uninstall the extension - from the cluster. + ) -> AsyncLROPoller[_models.Extension]: + """Patch an existing Kubernetes Cluster Extension. :param resource_group_name: The name of the resource group. The name is case insensitive. Required. @@ -472,132 +565,17 @@ async def begin_delete( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param force_delete: Delete the extension resource in Azure - not the normal asynchronous - delete. Default value is None. - :type force_delete: bool - :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[None] + :param patch_extension: Properties to Patch in an existing Extension. Required. + :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :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[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._delete_initial( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - force_delete=force_delete, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, - AsyncARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs), - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - async def _update_initial( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - patch_extension: Union[_models.PatchExtension, IO[bytes]], - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(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)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(patch_extension, (IOBase, bytes)): - _content = patch_extension - else: - _json = self._serialize.body(patch_extension, "PatchExtension") - - _request = build_update_request( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - 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, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - 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) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore @overload async def begin_update( @@ -607,7 +585,7 @@ async def begin_update( cluster_resource_name: str, cluster_name: str, extension_name: str, - patch_extension: _models.PatchExtension, + patch_extension: JSON, *, content_type: str = "application/json", **kwargs: Any @@ -628,12 +606,12 @@ async def begin_update( :param extension_name: Name of the Extension. Required. :type extension_name: str :param patch_extension: Properties to Patch in an existing Extension. Required. - :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension + :type patch_extension: JSON :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -672,8 +650,8 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -687,7 +665,7 @@ async def begin_update( cluster_resource_name: str, cluster_name: str, extension_name: str, - patch_extension: Union[_models.PatchExtension, IO[bytes]], + patch_extension: Union[_models.PatchExtension, JSON, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Extension]: """Patch an existing Kubernetes Cluster Extension. @@ -705,20 +683,19 @@ async def begin_update( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param patch_extension: Properties to Patch in an existing Extension. Is either a - PatchExtension type or a IO[bytes] type. Required. + :param patch_extension: Properties to Patch in an existing Extension. Is one of the following + types: PatchExtension, JSON, IO[bytes] Required. :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension or - IO[bytes] - :return: An instance of AsyncLROPoller that returns either Extension or the result of - cls(response) + JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Extension] = kwargs.pop("cls", None) polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) @@ -732,7 +709,6 @@ async def begin_update( cluster_name=cluster_name, extension_name=extension_name, patch_extension=patch_extension, - api_version=api_version, content_type=content_type, cls=lambda x, y, z: x, headers=_headers, @@ -743,15 +719,19 @@ async def begin_update( kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Extension", pipeline_response.http_response) + response = pipeline_response.http_response + deserialized = _deserialize(_models.Extension, response.json()) if cls: return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, - AsyncARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs), + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) ) elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) @@ -768,10 +748,168 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) + async def _delete_initial( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + *, + force_delete: Optional[bool] = None, + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_extensions_delete_request( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + subscription_id=self._config.subscription_id, + force_delete=force_delete, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _decompress = kwargs.pop("decompress", True) + _stream = True + 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, 202, 204]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + *, + force_delete: Optional[bool] = None, + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Delete a Kubernetes Cluster Extension. This will cause the Agent to Uninstall the extension + from the cluster. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :keyword force_delete: Delete the extension resource in Azure - not the normal asynchronous + delete. Default value is None. + :paramtype force_delete: bool + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._delete_initial( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + force_delete=force_delete, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + @distributed_trace def list( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, **kwargs: Any - ) -> AsyncIterable["_models.Extension"]: + ) -> AsyncItemPaged["_models.Extension"]: """List all Extensions in the cluster. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -785,16 +923,15 @@ def list( :type cluster_resource_name: str :param cluster_name: The name of the kubernetes cluster. Required. :type cluster_name: str - :return: An iterator like instance of either Extension or the result of cls(response) + :return: An iterator like instance of Extension :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ExtensionsList] = kwargs.pop("cls", None) + cls: ClsType[List[_models.Extension]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -807,17 +944,22 @@ def list( def prepare_request(next_link=None): if not next_link: - _request = build_list_request( + _request = build_extensions_list_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, subscription_id=self._config.subscription_id, - api_version=api_version, + api_version=self._config.api_version, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -832,16 +974,24 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("ExtensionsList", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Extension], + deserialized.get("value", []), + ) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) @@ -854,9 +1004,125 @@ async def get_next(next_link=None): 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) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return AsyncItemPaged(get_next, extract_data) + + +class OperationStatusOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.kubernetesconfiguration.extensions.aio.KubernetesConfigurationExtensionsMgmtClient`'s + :attr:`operation_status` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: KubernetesConfigurationExtensionsMgmtClientConfiguration = ( + input_args.pop(0) if input_args else kwargs.pop("config") + ) + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace_async + async def get( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + operation_id: str, + **kwargs: Any + ) -> _models.OperationStatusResult: + """Get Async Operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param operation_id: Required. + :type operation_id: str + :return: OperationStatusResult. The OperationStatusResult is compatible with MutableMapping + :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.OperationStatusResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OperationStatusResult] = kwargs.pop("cls", None) + + _request = build_operation_status_get_request( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + operation_id=operation_id, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("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]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.OperationStatusResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_patch.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_patch.py index 8bcb627aa475..87676c65a8f0 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_patch.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/aio/operations/_patch.py @@ -7,9 +7,9 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/__init__.py index df74a11fab45..2944c681fb03 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -13,17 +13,18 @@ from ._patch import * # pylint: disable=unused-wildcard-import -from ._models_py3 import ( # type: ignore +from ._models import ( # type: ignore ErrorAdditionalInfo, ErrorDetail, ErrorResponse, Extension, + ExtensionProperties, ExtensionPropertiesAksAssignedIdentity, ExtensionStatus, - ExtensionsList, Identity, OperationStatusResult, PatchExtension, + PatchExtensionProperties, Plan, ProxyResource, Resource, @@ -33,11 +34,12 @@ SystemData, ) -from ._kubernetes_configuration_extensions_mgmt_client_enums import ( # type: ignore +from ._enums import ( # type: ignore AKSIdentityType, CreatedByType, LevelType, ProvisioningState, + ResourceIdentityType, ) from ._patch import __all__ as _patch_all from ._patch import * @@ -48,12 +50,13 @@ "ErrorDetail", "ErrorResponse", "Extension", + "ExtensionProperties", "ExtensionPropertiesAksAssignedIdentity", "ExtensionStatus", - "ExtensionsList", "Identity", "OperationStatusResult", "PatchExtension", + "PatchExtensionProperties", "Plan", "ProxyResource", "Resource", @@ -65,6 +68,7 @@ "CreatedByType", "LevelType", "ProvisioningState", + "ResourceIdentityType", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_kubernetes_configuration_extensions_mgmt_client_enums.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_enums.py similarity index 64% rename from sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_kubernetes_configuration_extensions_mgmt_client_enums.py rename to sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_enums.py index 185640eb5da4..8103ccec7863 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_kubernetes_configuration_extensions_mgmt_client_enums.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_enums.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -14,32 +14,54 @@ class AKSIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The identity type.""" SYSTEM_ASSIGNED = "SystemAssigned" + """SystemAssigned.""" USER_ASSIGNED = "UserAssigned" + """UserAssigned.""" class CreatedByType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """The type of identity that created the resource.""" + """The kind of entity that created the resource.""" USER = "User" + """The entity was created by a user.""" APPLICATION = "Application" + """The entity was created by an application.""" MANAGED_IDENTITY = "ManagedIdentity" + """The entity was created by a managed identity.""" KEY = "Key" + """The entity was created by a key.""" class LevelType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Level of the status.""" ERROR = "Error" + """Error.""" WARNING = "Warning" + """Warning.""" INFORMATION = "Information" + """Information.""" class ProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The provisioning state of the resource.""" SUCCEEDED = "Succeeded" + """Succeeded.""" FAILED = "Failed" + """Failed.""" CANCELED = "Canceled" + """Canceled.""" CREATING = "Creating" + """Creating.""" UPDATING = "Updating" + """Updating.""" DELETING = "Deleting" + """Deleting.""" + + +class ResourceIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """ResourceIdentityType.""" + + SYSTEM_ASSIGNED = "SystemAssigned" + """SYSTEM_ASSIGNED.""" diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models.py new file mode 100644 index 000000000000..4092ba410eb8 --- /dev/null +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models.py @@ -0,0 +1,863 @@ +# pylint: disable=line-too-long,useless-suppression +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=useless-super-delegation + +import datetime +from typing import Any, Mapping, Optional, TYPE_CHECKING, Union, overload + +from .._utils.model_base import Model as _Model, rest_field + +if TYPE_CHECKING: + from .. import models as _models + + +class ErrorAdditionalInfo(_Model): + """The resource management error additional info. + + :ivar type: The additional info type. + :vartype type: str + :ivar info: The additional info. + :vartype info: any + """ + + type: Optional[str] = rest_field(visibility=["read"]) + """The additional info type.""" + info: Optional[Any] = rest_field(visibility=["read"]) + """The additional info.""" + + +class ErrorDetail(_Model): + """The error detail. + + :ivar code: The error code. + :vartype code: str + :ivar message: The error message. + :vartype message: str + :ivar target: The error target. + :vartype target: str + :ivar details: The error details. + :vartype details: list[~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail] + :ivar additional_info: The error additional info. + :vartype additional_info: + list[~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorAdditionalInfo] + """ + + code: Optional[str] = rest_field(visibility=["read"]) + """The error code.""" + message: Optional[str] = rest_field(visibility=["read"]) + """The error message.""" + target: Optional[str] = rest_field(visibility=["read"]) + """The error target.""" + details: Optional[list["_models.ErrorDetail"]] = rest_field(visibility=["read"]) + """The error details.""" + additional_info: Optional[list["_models.ErrorAdditionalInfo"]] = rest_field( + name="additionalInfo", visibility=["read"] + ) + """The error additional info.""" + + +class ErrorResponse(_Model): + """Error response. + + :ivar error: The error object. + :vartype error: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail + """ + + error: Optional["_models.ErrorDetail"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The error object.""" + + @overload + def __init__( + self, + *, + error: Optional["_models.ErrorDetail"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Resource(_Model): + """Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.kubernetesconfiguration.extensions.models.SystemData + """ + + id: Optional[str] = rest_field(visibility=["read"]) + """Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}.""" + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the resource.""" + type: Optional[str] = rest_field(visibility=["read"]) + """The type of the resource. E.g. \"Microsoft.Compute/virtualMachines\" or + \"Microsoft.Storage/storageAccounts\".""" + system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"]) + """Azure Resource Manager metadata containing createdBy and modifiedBy information.""" + + +class ProxyResource(Resource): + """Proxy Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.kubernetesconfiguration.extensions.models.SystemData + """ + + +class Extension(ProxyResource): + """The Extension object. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.kubernetesconfiguration.extensions.models.SystemData + :ivar properties: Properties of an Extension resource. + :vartype properties: ~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionProperties + :ivar identity: Identity of the Extension resource. + :vartype identity: ~azure.mgmt.kubernetesconfiguration.extensions.models.Identity + :ivar plan: Details of the resource plan. + :vartype plan: ~azure.mgmt.kubernetesconfiguration.extensions.models.Plan + """ + + properties: Optional["_models.ExtensionProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of an Extension resource.""" + identity: Optional["_models.Identity"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Identity of the Extension resource.""" + plan: Optional["_models.Plan"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Details of the resource plan.""" + + __flattened_items = [ + "extension_type", + "auto_upgrade_minor_version", + "release_train", + "version", + "scope", + "configuration_settings", + "configuration_protected_settings", + "current_version", + "provisioning_state", + "statuses", + "error_info", + "custom_location_settings", + "package_uri", + "aks_assigned_identity", + "is_system_extension", + ] + + @overload + def __init__( + self, + *, + properties: Optional["_models.ExtensionProperties"] = None, + identity: Optional["_models.Identity"] = None, + plan: Optional["_models.Plan"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class ExtensionProperties(_Model): + """Properties of an Extension resource. + + :ivar extension_type: Type of the Extension, of which this resource is an instance of. It must + be one of the Extension Types registered with Microsoft.KubernetesConfiguration by the + Extension publisher. + :vartype extension_type: str + :ivar auto_upgrade_minor_version: Flag to note if this extension participates in auto upgrade + of minor version, or not. + :vartype auto_upgrade_minor_version: bool + :ivar release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, + Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. + :vartype release_train: str + :ivar version: User-specified version of the extension for this extension to 'pin'. To use + 'version', autoUpgradeMinorVersion must be 'false'. + :vartype version: str + :ivar scope: Scope at which the extension is installed. + :vartype scope: ~azure.mgmt.kubernetesconfiguration.extensions.models.Scope + :ivar configuration_settings: Configuration settings, as name-value pairs for configuring this + extension. + :vartype configuration_settings: dict[str, str] + :ivar configuration_protected_settings: Configuration settings that are sensitive, as + name-value pairs for configuring this extension. + :vartype configuration_protected_settings: dict[str, str] + :ivar current_version: Currently installed version of the extension. + :vartype current_version: str + :ivar provisioning_state: Status of installation of this extension. Known values are: + "Succeeded", "Failed", "Canceled", "Creating", "Updating", and "Deleting". + :vartype provisioning_state: str or + ~azure.mgmt.kubernetesconfiguration.extensions.models.ProvisioningState + :ivar statuses: Status from this extension. + :vartype statuses: list[~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionStatus] + :ivar error_info: Error information from the Agent - e.g. errors during installation. + :vartype error_info: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail + :ivar custom_location_settings: Custom Location settings properties. + :vartype custom_location_settings: dict[str, str] + :ivar package_uri: Uri of the Helm package. + :vartype package_uri: str + :ivar aks_assigned_identity: Identity of the Extension resource in an AKS cluster. + :vartype aks_assigned_identity: + ~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionPropertiesAksAssignedIdentity + :ivar is_system_extension: Flag to note if this extension is a system extension. + :vartype is_system_extension: bool + """ + + extension_type: Optional[str] = rest_field( + name="extensionType", visibility=["read", "create", "update", "delete", "query"] + ) + """Type of the Extension, of which this resource is an instance of. It must be one of the + Extension Types registered with Microsoft.KubernetesConfiguration by the Extension publisher.""" + auto_upgrade_minor_version: Optional[bool] = rest_field( + name="autoUpgradeMinorVersion", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag to note if this extension participates in auto upgrade of minor version, or not.""" + release_train: Optional[str] = rest_field( + name="releaseTrain", visibility=["read", "create", "update", "delete", "query"] + ) + """ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, Preview, etc.) - + only if autoUpgradeMinorVersion is 'true'.""" + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """User-specified version of the extension for this extension to 'pin'. To use 'version', + autoUpgradeMinorVersion must be 'false'.""" + scope: Optional["_models.Scope"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Scope at which the extension is installed.""" + configuration_settings: Optional[dict[str, str]] = rest_field( + name="configurationSettings", visibility=["read", "create", "update", "delete", "query"] + ) + """Configuration settings, as name-value pairs for configuring this extension.""" + configuration_protected_settings: Optional[dict[str, str]] = rest_field( + name="configurationProtectedSettings", visibility=["read", "create", "update", "delete", "query"] + ) + """Configuration settings that are sensitive, as name-value pairs for configuring this extension.""" + current_version: Optional[str] = rest_field(name="currentVersion", visibility=["read"]) + """Currently installed version of the extension.""" + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Status of installation of this extension. Known values are: \"Succeeded\", \"Failed\", + \"Canceled\", \"Creating\", \"Updating\", and \"Deleting\".""" + statuses: Optional[list["_models.ExtensionStatus"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Status from this extension.""" + error_info: Optional["_models.ErrorDetail"] = rest_field(name="errorInfo", visibility=["read"]) + """Error information from the Agent - e.g. errors during installation.""" + custom_location_settings: Optional[dict[str, str]] = rest_field(name="customLocationSettings", visibility=["read"]) + """Custom Location settings properties.""" + package_uri: Optional[str] = rest_field(name="packageUri", visibility=["read"]) + """Uri of the Helm package.""" + aks_assigned_identity: Optional["_models.ExtensionPropertiesAksAssignedIdentity"] = rest_field( + name="aksAssignedIdentity", visibility=["read", "create", "update", "delete", "query"] + ) + """Identity of the Extension resource in an AKS cluster.""" + is_system_extension: Optional[bool] = rest_field(name="isSystemExtension", visibility=["read"]) + """Flag to note if this extension is a system extension.""" + + @overload + def __init__( + self, + *, + extension_type: Optional[str] = None, + auto_upgrade_minor_version: Optional[bool] = None, + release_train: Optional[str] = None, + version: Optional[str] = None, + scope: Optional["_models.Scope"] = None, + configuration_settings: Optional[dict[str, str]] = None, + configuration_protected_settings: Optional[dict[str, str]] = None, + statuses: Optional[list["_models.ExtensionStatus"]] = None, + aks_assigned_identity: Optional["_models.ExtensionPropertiesAksAssignedIdentity"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ExtensionPropertiesAksAssignedIdentity(_Model): + """Identity of the Extension resource in an AKS cluster. + + :ivar principal_id: The principal ID of resource identity. + :vartype principal_id: str + :ivar tenant_id: The tenant ID of resource. + :vartype tenant_id: str + :ivar type: The identity type. Known values are: "SystemAssigned" and "UserAssigned". + :vartype type: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.AKSIdentityType + """ + + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The principal ID of resource identity.""" + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The tenant ID of resource.""" + type: Optional[Union[str, "_models.AKSIdentityType"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The identity type. Known values are: \"SystemAssigned\" and \"UserAssigned\".""" + + @overload + def __init__( + self, + *, + type: Optional[Union[str, "_models.AKSIdentityType"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ExtensionStatus(_Model): + """Status from the extension. + + :ivar code: Status code provided by the Extension. + :vartype code: str + :ivar display_status: Short description of status of the extension. + :vartype display_status: str + :ivar level: Level of the status. Known values are: "Error", "Warning", and "Information". + :vartype level: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.LevelType + :ivar message: Detailed message of the status from the Extension. + :vartype message: str + :ivar time: DateLiteral (per ISO8601) noting the time of installation status. + :vartype time: str + """ + + code: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Status code provided by the Extension.""" + display_status: Optional[str] = rest_field( + name="displayStatus", visibility=["read", "create", "update", "delete", "query"] + ) + """Short description of status of the extension.""" + level: Optional[Union[str, "_models.LevelType"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Level of the status. Known values are: \"Error\", \"Warning\", and \"Information\".""" + message: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Detailed message of the status from the Extension.""" + time: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """DateLiteral (per ISO8601) noting the time of installation status.""" + + @overload + def __init__( + self, + *, + code: Optional[str] = None, + display_status: Optional[str] = None, + level: Optional[Union[str, "_models.LevelType"]] = None, + message: Optional[str] = None, + time: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Identity(_Model): + """Identity for the resource. + + :ivar principal_id: The principal ID of resource identity. The value must be an UUID. + :vartype principal_id: str + :ivar tenant_id: The tenant ID of resource. The value must be an UUID. + :vartype tenant_id: str + :ivar type: The identity type. "SystemAssigned" + :vartype type: str or + ~azure.mgmt.kubernetesconfiguration.extensions.models.ResourceIdentityType + """ + + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The principal ID of resource identity. The value must be an UUID.""" + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The tenant ID of resource. The value must be an UUID.""" + type: Optional[Union[str, "_models.ResourceIdentityType"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The identity type. \"SystemAssigned\"""" + + @overload + def __init__( + self, + *, + type: Optional[Union[str, "_models.ResourceIdentityType"]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class OperationStatusResult(_Model): + """The current status of an async operation. + + :ivar id: Fully qualified ID for the async operation. + :vartype id: str + :ivar name: Name of the async operation. + :vartype name: str + :ivar status: Operation status. Required. + :vartype status: str + :ivar properties: Additional information, if available. + :vartype properties: dict[str, str] + :ivar error: If present, details of the operation error. + :vartype error: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail + """ + + id: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Fully qualified ID for the async operation.""" + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Name of the async operation.""" + status: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Operation status. Required.""" + properties: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Additional information, if available.""" + error: Optional["_models.ErrorDetail"] = rest_field(visibility=["read"]) + """If present, details of the operation error.""" + + @overload + def __init__( + self, + *, + status: str, + id: Optional[str] = None, # pylint: disable=redefined-builtin + name: Optional[str] = None, + properties: Optional[dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class PatchExtension(_Model): + """The Extension Patch Request object. + + :ivar properties: Updatable properties of an Extension Patch Request. + :vartype properties: + ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtensionProperties + """ + + properties: Optional["_models.PatchExtensionProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Updatable properties of an Extension Patch Request.""" + + __flattened_items = [ + "auto_upgrade_minor_version", + "release_train", + "version", + "configuration_settings", + "configuration_protected_settings", + ] + + @overload + def __init__( + self, + *, + properties: Optional["_models.PatchExtensionProperties"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + _flattened_input = {k: kwargs.pop(k) for k in kwargs.keys() & self.__flattened_items} + super().__init__(*args, **kwargs) + for k, v in _flattened_input.items(): + setattr(self, k, v) + + def __getattr__(self, name: str) -> Any: + if name in self.__flattened_items: + if self.properties is None: + return None + return getattr(self.properties, name) + raise AttributeError(f"'{self.__class__.__name__}' object has no attribute '{name}'") + + def __setattr__(self, key: str, value: Any) -> None: + if key in self.__flattened_items: + if self.properties is None: + self.properties = self._attr_to_rest_field["properties"]._class_type() + setattr(self.properties, key, value) + else: + super().__setattr__(key, value) + + +class PatchExtensionProperties(_Model): + """Updatable properties of an Extension Patch Request. + + :ivar auto_upgrade_minor_version: Flag to note if this extension participates in auto upgrade + of minor version, or not. + :vartype auto_upgrade_minor_version: bool + :ivar release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, + Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. + :vartype release_train: str + :ivar version: Version of the extension for this extension, if it is 'pinned' to a specific + version. autoUpgradeMinorVersion must be 'false'. + :vartype version: str + :ivar configuration_settings: Configuration settings, as name-value pairs for configuring this + extension. + :vartype configuration_settings: dict[str, str] + :ivar configuration_protected_settings: Configuration settings that are sensitive, as + name-value pairs for configuring this extension. + :vartype configuration_protected_settings: dict[str, str] + """ + + auto_upgrade_minor_version: Optional[bool] = rest_field( + name="autoUpgradeMinorVersion", visibility=["read", "create", "update", "delete", "query"] + ) + """Flag to note if this extension participates in auto upgrade of minor version, or not.""" + release_train: Optional[str] = rest_field( + name="releaseTrain", visibility=["read", "create", "update", "delete", "query"] + ) + """ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, Preview, etc.) - + only if autoUpgradeMinorVersion is 'true'.""" + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Version of the extension for this extension, if it is 'pinned' to a specific version. + autoUpgradeMinorVersion must be 'false'.""" + configuration_settings: Optional[dict[str, str]] = rest_field( + name="configurationSettings", visibility=["read", "create", "update", "delete", "query"] + ) + """Configuration settings, as name-value pairs for configuring this extension.""" + configuration_protected_settings: Optional[dict[str, str]] = rest_field( + name="configurationProtectedSettings", visibility=["read", "create", "update", "delete", "query"] + ) + """Configuration settings that are sensitive, as name-value pairs for configuring this extension.""" + + @overload + def __init__( + self, + *, + auto_upgrade_minor_version: Optional[bool] = None, + release_train: Optional[str] = None, + version: Optional[str] = None, + configuration_settings: Optional[dict[str, str]] = None, + configuration_protected_settings: Optional[dict[str, str]] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Plan(_Model): + """Plan for the resource. + + :ivar name: A user defined name of the 3rd Party Artifact that is being procured. Required. + :vartype name: str + :ivar publisher: The publisher of the 3rd Party Artifact that is being bought. E.g. NewRelic. + Required. + :vartype publisher: str + :ivar product: The 3rd Party artifact that is being procured. E.g. NewRelic. Product maps to + the OfferID specified for the artifact at the time of Data Market onboarding. Required. + :vartype product: str + :ivar promotion_code: A publisher provided promotion code as provisioned in Data Market for the + said product/artifact. + :vartype promotion_code: str + :ivar version: The version of the desired product/artifact. + :vartype version: str + """ + + name: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """A user defined name of the 3rd Party Artifact that is being procured. Required.""" + publisher: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The publisher of the 3rd Party Artifact that is being bought. E.g. NewRelic. Required.""" + product: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The 3rd Party artifact that is being procured. E.g. NewRelic. Product maps to the OfferID + specified for the artifact at the time of Data Market onboarding. Required.""" + promotion_code: Optional[str] = rest_field( + name="promotionCode", visibility=["read", "create", "update", "delete", "query"] + ) + """A publisher provided promotion code as provisioned in Data Market for the said + product/artifact.""" + version: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The version of the desired product/artifact.""" + + @overload + def __init__( + self, + *, + name: str, + publisher: str, + product: str, + promotion_code: Optional[str] = None, + version: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class Scope(_Model): + """Scope of the extension. It can be either Cluster or Namespace; but not both. + + :ivar cluster: Specifies that the scope of the extension is Cluster. + :vartype cluster: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeCluster + :ivar namespace: Specifies that the scope of the extension is Namespace. + :vartype namespace: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeNamespace + """ + + cluster: Optional["_models.ScopeCluster"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Specifies that the scope of the extension is Cluster.""" + namespace: Optional["_models.ScopeNamespace"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Specifies that the scope of the extension is Namespace.""" + + @overload + def __init__( + self, + *, + cluster: Optional["_models.ScopeCluster"] = None, + namespace: Optional["_models.ScopeNamespace"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ScopeCluster(_Model): + """Specifies that the scope of the extension is Cluster. + + :ivar release_namespace: Namespace where the extension Release must be placed, for a Cluster + scoped extension. If this namespace does not exist, it will be created. + :vartype release_namespace: str + """ + + release_namespace: Optional[str] = rest_field( + name="releaseNamespace", visibility=["read", "create", "update", "delete", "query"] + ) + """Namespace where the extension Release must be placed, for a Cluster scoped extension. If this + namespace does not exist, it will be created.""" + + @overload + def __init__( + self, + *, + release_namespace: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class ScopeNamespace(_Model): + """Specifies that the scope of the extension is Namespace. + + :ivar target_namespace: Namespace where the extension will be created for an Namespace scoped + extension. If this namespace does not exist, it will be created. + :vartype target_namespace: str + """ + + target_namespace: Optional[str] = rest_field( + name="targetNamespace", visibility=["read", "create", "update", "delete", "query"] + ) + """Namespace where the extension will be created for an Namespace scoped extension. If this + namespace does not exist, it will be created.""" + + @overload + def __init__( + self, + *, + target_namespace: Optional[str] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class SystemData(_Model): + """Metadata pertaining to creation and last modification of the resource. + + :ivar created_by: The identity that created the resource. + :vartype created_by: str + :ivar created_by_type: The type of identity that created the resource. Known values are: + "User", "Application", "ManagedIdentity", and "Key". + :vartype created_by_type: str or + ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType + :ivar created_at: The timestamp of resource creation (UTC). + :vartype created_at: ~datetime.datetime + :ivar last_modified_by: The identity that last modified the resource. + :vartype last_modified_by: str + :ivar last_modified_by_type: The type of identity that last modified the resource. Known values + are: "User", "Application", "ManagedIdentity", and "Key". + :vartype last_modified_by_type: str or + ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType + :ivar last_modified_at: The timestamp of resource last modification (UTC). + :vartype last_modified_at: ~datetime.datetime + """ + + created_by: Optional[str] = rest_field(name="createdBy", visibility=["read", "create", "update", "delete", "query"]) + """The identity that created the resource.""" + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="createdByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that created the resource. Known values are: \"User\", \"Application\", + \"ManagedIdentity\", and \"Key\".""" + created_at: Optional[datetime.datetime] = rest_field( + name="createdAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource creation (UTC).""" + last_modified_by: Optional[str] = rest_field( + name="lastModifiedBy", visibility=["read", "create", "update", "delete", "query"] + ) + """The identity that last modified the resource.""" + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = rest_field( + name="lastModifiedByType", visibility=["read", "create", "update", "delete", "query"] + ) + """The type of identity that last modified the resource. Known values are: \"User\", + \"Application\", \"ManagedIdentity\", and \"Key\".""" + last_modified_at: Optional[datetime.datetime] = rest_field( + name="lastModifiedAt", visibility=["read", "create", "update", "delete", "query"], format="rfc3339" + ) + """The timestamp of resource last modification (UTC).""" + + @overload + def __init__( + self, + *, + created_by: Optional[str] = None, + created_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + created_at: Optional[datetime.datetime] = None, + last_modified_by: Optional[str] = None, + last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, + last_modified_at: Optional[datetime.datetime] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models_py3.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models_py3.py deleted file mode 100644 index ac6c4016e57a..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_models_py3.py +++ /dev/null @@ -1,818 +0,0 @@ -# pylint: disable=line-too-long,useless-suppression -# 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. -# -------------------------------------------------------------------------- - -import datetime -from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union - -from .._utils import serialization as _serialization - -if TYPE_CHECKING: - from .. import models as _models - - -class ErrorAdditionalInfo(_serialization.Model): - """The resource management error additional info. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar type: The additional info type. - :vartype type: str - :ivar info: The additional info. - :vartype info: JSON - """ - - _validation = { - "type": {"readonly": True}, - "info": {"readonly": True}, - } - - _attribute_map = { - "type": {"key": "type", "type": "str"}, - "info": {"key": "info", "type": "object"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.type: Optional[str] = None - self.info: Optional[JSON] = None - - -class ErrorDetail(_serialization.Model): - """The error detail. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar code: The error code. - :vartype code: str - :ivar message: The error message. - :vartype message: str - :ivar target: The error target. - :vartype target: str - :ivar details: The error details. - :vartype details: list[~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail] - :ivar additional_info: The error additional info. - :vartype additional_info: - list[~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorAdditionalInfo] - """ - - _validation = { - "code": {"readonly": True}, - "message": {"readonly": True}, - "target": {"readonly": True}, - "details": {"readonly": True}, - "additional_info": {"readonly": True}, - } - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "message": {"key": "message", "type": "str"}, - "target": {"key": "target", "type": "str"}, - "details": {"key": "details", "type": "[ErrorDetail]"}, - "additional_info": {"key": "additionalInfo", "type": "[ErrorAdditionalInfo]"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.code: Optional[str] = None - self.message: Optional[str] = None - self.target: Optional[str] = None - self.details: Optional[List["_models.ErrorDetail"]] = None - self.additional_info: Optional[List["_models.ErrorAdditionalInfo"]] = None - - -class ErrorResponse(_serialization.Model): - """Common error response for all Azure Resource Manager APIs to return error details for failed - operations. (This also follows the OData error response format.). - - :ivar error: The error object. - :vartype error: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail - """ - - _attribute_map = { - "error": {"key": "error", "type": "ErrorDetail"}, - } - - def __init__(self, *, error: Optional["_models.ErrorDetail"] = None, **kwargs: Any) -> None: - """ - :keyword error: The error object. - :paramtype error: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail - """ - super().__init__(**kwargs) - self.error = error - - -class Resource(_serialization.Model): - """Common fields that are returned in the response for all Azure Resource Manager resources. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Fully qualified resource ID for the resource. Ex - - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. - :vartype id: str - :ivar name: The name of the resource. - :vartype name: str - :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or - "Microsoft.Storage/storageAccounts". - :vartype type: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.id: Optional[str] = None - self.name: Optional[str] = None - self.type: Optional[str] = None - - -class ProxyResource(Resource): - """The resource model definition for a Azure Resource Manager proxy resource. It will not have - tags and a location. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Fully qualified resource ID for the resource. Ex - - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. - :vartype id: str - :ivar name: The name of the resource. - :vartype name: str - :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or - "Microsoft.Storage/storageAccounts". - :vartype type: str - """ - - -class Extension(ProxyResource): - """The Extension object. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Fully qualified resource ID for the resource. Ex - - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. - :vartype id: str - :ivar name: The name of the resource. - :vartype name: str - :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or - "Microsoft.Storage/storageAccounts". - :vartype type: str - :ivar identity: Identity of the Extension resource. - :vartype identity: ~azure.mgmt.kubernetesconfiguration.extensions.models.Identity - :ivar system_data: Top level metadata - https://github.com/Azure/azure-resource-manager-rpc/blob/master/v1.0/common-api-contracts.md#system-metadata-for-all-azure-resources. - :vartype system_data: ~azure.mgmt.kubernetesconfiguration.extensions.models.SystemData - :ivar plan: The plan information. - :vartype plan: ~azure.mgmt.kubernetesconfiguration.extensions.models.Plan - :ivar extension_type: Type of the Extension, of which this resource is an instance of. It must - be one of the Extension Types registered with Microsoft.KubernetesConfiguration by the - Extension publisher. - :vartype extension_type: str - :ivar auto_upgrade_minor_version: Flag to note if this extension participates in auto upgrade - of minor version, or not. - :vartype auto_upgrade_minor_version: bool - :ivar release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, - Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. - :vartype release_train: str - :ivar version: User-specified version of the extension for this extension to 'pin'. To use - 'version', autoUpgradeMinorVersion must be 'false'. - :vartype version: str - :ivar scope: Scope at which the extension is installed. - :vartype scope: ~azure.mgmt.kubernetesconfiguration.extensions.models.Scope - :ivar configuration_settings: Configuration settings, as name-value pairs for configuring this - extension. - :vartype configuration_settings: dict[str, str] - :ivar configuration_protected_settings: Configuration settings that are sensitive, as - name-value pairs for configuring this extension. - :vartype configuration_protected_settings: dict[str, str] - :ivar current_version: Currently installed version of the extension. - :vartype current_version: str - :ivar provisioning_state: Status of installation of this extension. Known values are: - "Succeeded", "Failed", "Canceled", "Creating", "Updating", and "Deleting". - :vartype provisioning_state: str or - ~azure.mgmt.kubernetesconfiguration.extensions.models.ProvisioningState - :ivar statuses: Status from this extension. - :vartype statuses: list[~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionStatus] - :ivar error_info: Error information from the Agent - e.g. errors during installation. - :vartype error_info: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail - :ivar custom_location_settings: Custom Location settings properties. - :vartype custom_location_settings: dict[str, str] - :ivar package_uri: Uri of the Helm package. - :vartype package_uri: str - :ivar aks_assigned_identity: Identity of the Extension resource in an AKS cluster. - :vartype aks_assigned_identity: - ~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionPropertiesAksAssignedIdentity - :ivar is_system_extension: Flag to note if this extension is a system extension. - :vartype is_system_extension: bool - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "current_version": {"readonly": True}, - "provisioning_state": {"readonly": True}, - "error_info": {"readonly": True}, - "custom_location_settings": {"readonly": True}, - "package_uri": {"readonly": True}, - "is_system_extension": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "identity": {"key": "identity", "type": "Identity"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "plan": {"key": "plan", "type": "Plan"}, - "extension_type": {"key": "properties.extensionType", "type": "str"}, - "auto_upgrade_minor_version": {"key": "properties.autoUpgradeMinorVersion", "type": "bool"}, - "release_train": {"key": "properties.releaseTrain", "type": "str"}, - "version": {"key": "properties.version", "type": "str"}, - "scope": {"key": "properties.scope", "type": "Scope"}, - "configuration_settings": {"key": "properties.configurationSettings", "type": "{str}"}, - "configuration_protected_settings": {"key": "properties.configurationProtectedSettings", "type": "{str}"}, - "current_version": {"key": "properties.currentVersion", "type": "str"}, - "provisioning_state": {"key": "properties.provisioningState", "type": "str"}, - "statuses": {"key": "properties.statuses", "type": "[ExtensionStatus]"}, - "error_info": {"key": "properties.errorInfo", "type": "ErrorDetail"}, - "custom_location_settings": {"key": "properties.customLocationSettings", "type": "{str}"}, - "package_uri": {"key": "properties.packageUri", "type": "str"}, - "aks_assigned_identity": { - "key": "properties.aksAssignedIdentity", - "type": "ExtensionPropertiesAksAssignedIdentity", - }, - "is_system_extension": {"key": "properties.isSystemExtension", "type": "bool"}, - } - - def __init__( - self, - *, - identity: Optional["_models.Identity"] = None, - plan: Optional["_models.Plan"] = None, - extension_type: Optional[str] = None, - auto_upgrade_minor_version: bool = True, - release_train: str = "Stable", - version: Optional[str] = None, - scope: Optional["_models.Scope"] = None, - configuration_settings: Optional[Dict[str, str]] = None, - configuration_protected_settings: Optional[Dict[str, str]] = None, - statuses: Optional[List["_models.ExtensionStatus"]] = None, - aks_assigned_identity: Optional["_models.ExtensionPropertiesAksAssignedIdentity"] = None, - **kwargs: Any - ) -> None: - """ - :keyword identity: Identity of the Extension resource. - :paramtype identity: ~azure.mgmt.kubernetesconfiguration.extensions.models.Identity - :keyword plan: The plan information. - :paramtype plan: ~azure.mgmt.kubernetesconfiguration.extensions.models.Plan - :keyword extension_type: Type of the Extension, of which this resource is an instance of. It - must be one of the Extension Types registered with Microsoft.KubernetesConfiguration by the - Extension publisher. - :paramtype extension_type: str - :keyword auto_upgrade_minor_version: Flag to note if this extension participates in auto - upgrade of minor version, or not. - :paramtype auto_upgrade_minor_version: bool - :keyword release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. - Stable, Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. - :paramtype release_train: str - :keyword version: User-specified version of the extension for this extension to 'pin'. To use - 'version', autoUpgradeMinorVersion must be 'false'. - :paramtype version: str - :keyword scope: Scope at which the extension is installed. - :paramtype scope: ~azure.mgmt.kubernetesconfiguration.extensions.models.Scope - :keyword configuration_settings: Configuration settings, as name-value pairs for configuring - this extension. - :paramtype configuration_settings: dict[str, str] - :keyword configuration_protected_settings: Configuration settings that are sensitive, as - name-value pairs for configuring this extension. - :paramtype configuration_protected_settings: dict[str, str] - :keyword statuses: Status from this extension. - :paramtype statuses: - list[~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionStatus] - :keyword aks_assigned_identity: Identity of the Extension resource in an AKS cluster. - :paramtype aks_assigned_identity: - ~azure.mgmt.kubernetesconfiguration.extensions.models.ExtensionPropertiesAksAssignedIdentity - """ - super().__init__(**kwargs) - self.identity = identity - self.system_data: Optional["_models.SystemData"] = None - self.plan = plan - self.extension_type = extension_type - self.auto_upgrade_minor_version = auto_upgrade_minor_version - self.release_train = release_train - self.version = version - self.scope = scope - self.configuration_settings = configuration_settings - self.configuration_protected_settings = configuration_protected_settings - self.current_version: Optional[str] = None - self.provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = None - self.statuses = statuses - self.error_info: Optional["_models.ErrorDetail"] = None - self.custom_location_settings: Optional[Dict[str, str]] = None - self.package_uri: Optional[str] = None - self.aks_assigned_identity = aks_assigned_identity - self.is_system_extension: Optional[bool] = None - - -class ExtensionPropertiesAksAssignedIdentity(_serialization.Model): - """Identity of the Extension resource in an AKS cluster. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar principal_id: The principal ID of resource identity. - :vartype principal_id: str - :ivar tenant_id: The tenant ID of resource. - :vartype tenant_id: str - :ivar type: The identity type. Known values are: "SystemAssigned" and "UserAssigned". - :vartype type: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.AKSIdentityType - """ - - _validation = { - "principal_id": {"readonly": True}, - "tenant_id": {"readonly": True}, - } - - _attribute_map = { - "principal_id": {"key": "principalId", "type": "str"}, - "tenant_id": {"key": "tenantId", "type": "str"}, - "type": {"key": "type", "type": "str"}, - } - - def __init__(self, *, type: Optional[Union[str, "_models.AKSIdentityType"]] = None, **kwargs: Any) -> None: - """ - :keyword type: The identity type. Known values are: "SystemAssigned" and "UserAssigned". - :paramtype type: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.AKSIdentityType - """ - super().__init__(**kwargs) - self.principal_id: Optional[str] = None - self.tenant_id: Optional[str] = None - self.type = type - - -class ExtensionsList(_serialization.Model): - """Result of the request to list Extensions. It contains a list of Extension objects and a URL - link to get the next set of results. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of Extensions within a Kubernetes cluster. - :vartype value: list[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] - :ivar next_link: URL to get the next set of extension objects, if any. - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[Extension]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value: Optional[List["_models.Extension"]] = None - self.next_link: Optional[str] = None - - -class ExtensionStatus(_serialization.Model): - """Status from the extension. - - :ivar code: Status code provided by the Extension. - :vartype code: str - :ivar display_status: Short description of status of the extension. - :vartype display_status: str - :ivar level: Level of the status. Known values are: "Error", "Warning", and "Information". - :vartype level: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.LevelType - :ivar message: Detailed message of the status from the Extension. - :vartype message: str - :ivar time: DateLiteral (per ISO8601) noting the time of installation status. - :vartype time: str - """ - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "display_status": {"key": "displayStatus", "type": "str"}, - "level": {"key": "level", "type": "str"}, - "message": {"key": "message", "type": "str"}, - "time": {"key": "time", "type": "str"}, - } - - def __init__( - self, - *, - code: Optional[str] = None, - display_status: Optional[str] = None, - level: Union[str, "_models.LevelType"] = "Information", - message: Optional[str] = None, - time: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword code: Status code provided by the Extension. - :paramtype code: str - :keyword display_status: Short description of status of the extension. - :paramtype display_status: str - :keyword level: Level of the status. Known values are: "Error", "Warning", and "Information". - :paramtype level: str or ~azure.mgmt.kubernetesconfiguration.extensions.models.LevelType - :keyword message: Detailed message of the status from the Extension. - :paramtype message: str - :keyword time: DateLiteral (per ISO8601) noting the time of installation status. - :paramtype time: str - """ - super().__init__(**kwargs) - self.code = code - self.display_status = display_status - self.level = level - self.message = message - self.time = time - - -class Identity(_serialization.Model): - """Identity for the resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar principal_id: The principal ID of resource identity. - :vartype principal_id: str - :ivar tenant_id: The tenant ID of resource. - :vartype tenant_id: str - :ivar type: The identity type. Default value is "SystemAssigned". - :vartype type: str - """ - - _validation = { - "principal_id": {"readonly": True}, - "tenant_id": {"readonly": True}, - } - - _attribute_map = { - "principal_id": {"key": "principalId", "type": "str"}, - "tenant_id": {"key": "tenantId", "type": "str"}, - "type": {"key": "type", "type": "str"}, - } - - def __init__(self, *, type: Optional[Literal["SystemAssigned"]] = None, **kwargs: Any) -> None: - """ - :keyword type: The identity type. Default value is "SystemAssigned". - :paramtype type: str - """ - super().__init__(**kwargs) - self.principal_id: Optional[str] = None - self.tenant_id: Optional[str] = None - self.type = type - - -class OperationStatusResult(_serialization.Model): - """The current status of an async operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - All required parameters must be populated in order to send to server. - - :ivar id: Fully qualified ID for the async operation. - :vartype id: str - :ivar name: Name of the async operation. - :vartype name: str - :ivar status: Operation status. Required. - :vartype status: str - :ivar properties: Additional information, if available. - :vartype properties: dict[str, str] - :ivar error: If present, details of the operation error. - :vartype error: ~azure.mgmt.kubernetesconfiguration.extensions.models.ErrorDetail - """ - - _validation = { - "status": {"required": True}, - "error": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "status": {"key": "status", "type": "str"}, - "properties": {"key": "properties", "type": "{str}"}, - "error": {"key": "error", "type": "ErrorDetail"}, - } - - def __init__( - self, - *, - status: str, - id: Optional[str] = None, # pylint: disable=redefined-builtin - name: Optional[str] = None, - properties: Optional[Dict[str, str]] = None, - **kwargs: Any - ) -> None: - """ - :keyword id: Fully qualified ID for the async operation. - :paramtype id: str - :keyword name: Name of the async operation. - :paramtype name: str - :keyword status: Operation status. Required. - :paramtype status: str - :keyword properties: Additional information, if available. - :paramtype properties: dict[str, str] - """ - super().__init__(**kwargs) - self.id = id - self.name = name - self.status = status - self.properties = properties - self.error: Optional["_models.ErrorDetail"] = None - - -class PatchExtension(_serialization.Model): - """The Extension Patch Request object. - - :ivar auto_upgrade_minor_version: Flag to note if this extension participates in auto upgrade - of minor version, or not. - :vartype auto_upgrade_minor_version: bool - :ivar release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. Stable, - Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. - :vartype release_train: str - :ivar version: Version of the extension for this extension, if it is 'pinned' to a specific - version. autoUpgradeMinorVersion must be 'false'. - :vartype version: str - :ivar configuration_settings: Configuration settings, as name-value pairs for configuring this - extension. - :vartype configuration_settings: dict[str, str] - :ivar configuration_protected_settings: Configuration settings that are sensitive, as - name-value pairs for configuring this extension. - :vartype configuration_protected_settings: dict[str, str] - """ - - _attribute_map = { - "auto_upgrade_minor_version": {"key": "properties.autoUpgradeMinorVersion", "type": "bool"}, - "release_train": {"key": "properties.releaseTrain", "type": "str"}, - "version": {"key": "properties.version", "type": "str"}, - "configuration_settings": {"key": "properties.configurationSettings", "type": "{str}"}, - "configuration_protected_settings": {"key": "properties.configurationProtectedSettings", "type": "{str}"}, - } - - def __init__( - self, - *, - auto_upgrade_minor_version: bool = True, - release_train: str = "Stable", - version: Optional[str] = None, - configuration_settings: Optional[Dict[str, str]] = None, - configuration_protected_settings: Optional[Dict[str, str]] = None, - **kwargs: Any - ) -> None: - """ - :keyword auto_upgrade_minor_version: Flag to note if this extension participates in auto - upgrade of minor version, or not. - :paramtype auto_upgrade_minor_version: bool - :keyword release_train: ReleaseTrain this extension participates in for auto-upgrade (e.g. - Stable, Preview, etc.) - only if autoUpgradeMinorVersion is 'true'. - :paramtype release_train: str - :keyword version: Version of the extension for this extension, if it is 'pinned' to a specific - version. autoUpgradeMinorVersion must be 'false'. - :paramtype version: str - :keyword configuration_settings: Configuration settings, as name-value pairs for configuring - this extension. - :paramtype configuration_settings: dict[str, str] - :keyword configuration_protected_settings: Configuration settings that are sensitive, as - name-value pairs for configuring this extension. - :paramtype configuration_protected_settings: dict[str, str] - """ - super().__init__(**kwargs) - self.auto_upgrade_minor_version = auto_upgrade_minor_version - self.release_train = release_train - self.version = version - self.configuration_settings = configuration_settings - self.configuration_protected_settings = configuration_protected_settings - - -class Plan(_serialization.Model): - """Plan for the resource. - - All required parameters must be populated in order to send to server. - - :ivar name: A user defined name of the 3rd Party Artifact that is being procured. Required. - :vartype name: str - :ivar publisher: The publisher of the 3rd Party Artifact that is being bought. E.g. NewRelic. - Required. - :vartype publisher: str - :ivar product: The 3rd Party artifact that is being procured. E.g. NewRelic. Product maps to - the OfferID specified for the artifact at the time of Data Market onboarding. Required. - :vartype product: str - :ivar promotion_code: A publisher provided promotion code as provisioned in Data Market for the - said product/artifact. - :vartype promotion_code: str - :ivar version: The version of the desired product/artifact. - :vartype version: str - """ - - _validation = { - "name": {"required": True}, - "publisher": {"required": True}, - "product": {"required": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "publisher": {"key": "publisher", "type": "str"}, - "product": {"key": "product", "type": "str"}, - "promotion_code": {"key": "promotionCode", "type": "str"}, - "version": {"key": "version", "type": "str"}, - } - - def __init__( - self, - *, - name: str, - publisher: str, - product: str, - promotion_code: Optional[str] = None, - version: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: A user defined name of the 3rd Party Artifact that is being procured. Required. - :paramtype name: str - :keyword publisher: The publisher of the 3rd Party Artifact that is being bought. E.g. - NewRelic. Required. - :paramtype publisher: str - :keyword product: The 3rd Party artifact that is being procured. E.g. NewRelic. Product maps to - the OfferID specified for the artifact at the time of Data Market onboarding. Required. - :paramtype product: str - :keyword promotion_code: A publisher provided promotion code as provisioned in Data Market for - the said product/artifact. - :paramtype promotion_code: str - :keyword version: The version of the desired product/artifact. - :paramtype version: str - """ - super().__init__(**kwargs) - self.name = name - self.publisher = publisher - self.product = product - self.promotion_code = promotion_code - self.version = version - - -class Scope(_serialization.Model): - """Scope of the extension. It can be either Cluster or Namespace; but not both. - - :ivar cluster: Specifies that the scope of the extension is Cluster. - :vartype cluster: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeCluster - :ivar namespace: Specifies that the scope of the extension is Namespace. - :vartype namespace: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeNamespace - """ - - _attribute_map = { - "cluster": {"key": "cluster", "type": "ScopeCluster"}, - "namespace": {"key": "namespace", "type": "ScopeNamespace"}, - } - - def __init__( - self, - *, - cluster: Optional["_models.ScopeCluster"] = None, - namespace: Optional["_models.ScopeNamespace"] = None, - **kwargs: Any - ) -> None: - """ - :keyword cluster: Specifies that the scope of the extension is Cluster. - :paramtype cluster: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeCluster - :keyword namespace: Specifies that the scope of the extension is Namespace. - :paramtype namespace: ~azure.mgmt.kubernetesconfiguration.extensions.models.ScopeNamespace - """ - super().__init__(**kwargs) - self.cluster = cluster - self.namespace = namespace - - -class ScopeCluster(_serialization.Model): - """Specifies that the scope of the extension is Cluster. - - :ivar release_namespace: Namespace where the extension Release must be placed, for a Cluster - scoped extension. If this namespace does not exist, it will be created. - :vartype release_namespace: str - """ - - _attribute_map = { - "release_namespace": {"key": "releaseNamespace", "type": "str"}, - } - - def __init__(self, *, release_namespace: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword release_namespace: Namespace where the extension Release must be placed, for a Cluster - scoped extension. If this namespace does not exist, it will be created. - :paramtype release_namespace: str - """ - super().__init__(**kwargs) - self.release_namespace = release_namespace - - -class ScopeNamespace(_serialization.Model): - """Specifies that the scope of the extension is Namespace. - - :ivar target_namespace: Namespace where the extension will be created for an Namespace scoped - extension. If this namespace does not exist, it will be created. - :vartype target_namespace: str - """ - - _attribute_map = { - "target_namespace": {"key": "targetNamespace", "type": "str"}, - } - - def __init__(self, *, target_namespace: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword target_namespace: Namespace where the extension will be created for an Namespace - scoped extension. If this namespace does not exist, it will be created. - :paramtype target_namespace: str - """ - super().__init__(**kwargs) - self.target_namespace = target_namespace - - -class SystemData(_serialization.Model): - """Metadata pertaining to creation and last modification of the resource. - - :ivar created_by: The identity that created the resource. - :vartype created_by: str - :ivar created_by_type: The type of identity that created the resource. Known values are: - "User", "Application", "ManagedIdentity", and "Key". - :vartype created_by_type: str or - ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType - :ivar created_at: The timestamp of resource creation (UTC). - :vartype created_at: ~datetime.datetime - :ivar last_modified_by: The identity that last modified the resource. - :vartype last_modified_by: str - :ivar last_modified_by_type: The type of identity that last modified the resource. Known values - are: "User", "Application", "ManagedIdentity", and "Key". - :vartype last_modified_by_type: str or - ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType - :ivar last_modified_at: The timestamp of resource last modification (UTC). - :vartype last_modified_at: ~datetime.datetime - """ - - _attribute_map = { - "created_by": {"key": "createdBy", "type": "str"}, - "created_by_type": {"key": "createdByType", "type": "str"}, - "created_at": {"key": "createdAt", "type": "iso-8601"}, - "last_modified_by": {"key": "lastModifiedBy", "type": "str"}, - "last_modified_by_type": {"key": "lastModifiedByType", "type": "str"}, - "last_modified_at": {"key": "lastModifiedAt", "type": "iso-8601"}, - } - - def __init__( - self, - *, - created_by: Optional[str] = None, - created_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, - created_at: Optional[datetime.datetime] = None, - last_modified_by: Optional[str] = None, - last_modified_by_type: Optional[Union[str, "_models.CreatedByType"]] = None, - last_modified_at: Optional[datetime.datetime] = None, - **kwargs: Any - ) -> None: - """ - :keyword created_by: The identity that created the resource. - :paramtype created_by: str - :keyword created_by_type: The type of identity that created the resource. Known values are: - "User", "Application", "ManagedIdentity", and "Key". - :paramtype created_by_type: str or - ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType - :keyword created_at: The timestamp of resource creation (UTC). - :paramtype created_at: ~datetime.datetime - :keyword last_modified_by: The identity that last modified the resource. - :paramtype last_modified_by: str - :keyword last_modified_by_type: The type of identity that last modified the resource. Known - values are: "User", "Application", "ManagedIdentity", and "Key". - :paramtype last_modified_by_type: str or - ~azure.mgmt.kubernetesconfiguration.extensions.models.CreatedByType - :keyword last_modified_at: The timestamp of resource last modification (UTC). - :paramtype last_modified_at: ~datetime.datetime - """ - super().__init__(**kwargs) - self.created_by = created_by - self.created_by_type = created_by_type - self.created_at = created_at - self.last_modified_by = last_modified_by - self.last_modified_by_type = last_modified_by_type - self.last_modified_at = last_modified_at diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_patch.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_patch.py index 8bcb627aa475..87676c65a8f0 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_patch.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/models/_patch.py @@ -7,9 +7,9 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/__init__.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/__init__.py index a43141e42aa8..811e5748c43d 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/__init__.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/__init__.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pylint: disable=wrong-import-position @@ -12,8 +12,8 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._extensions_operations import ExtensionsOperations # type: ignore -from ._operation_status_operations import OperationStatusOperations # type: ignore +from ._operations import ExtensionsOperations # type: ignore +from ._operations import OperationStatusOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operation_status_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operation_status_operations.py deleted file mode 100644 index 4bb9c135a0ba..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operation_status_operations.py +++ /dev/null @@ -1,182 +0,0 @@ -# pylint: disable=line-too-long,useless-suppression -# 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 collections.abc import MutableMapping -from typing import Any, Callable, Dict, Optional, TypeVar - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import HttpRequest, HttpResponse -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 .._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration -from .._utils.serialization import Deserializer, Serializer - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_get_request( - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - operation_id: str, - subscription_id: str, - **kwargs: Any -) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}/operations/{operationId}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), - "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), - "operationId": _SERIALIZER.url("operation_id", operation_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationStatusOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.kubernetesconfiguration.extensions.KubernetesConfigurationExtensionsMgmtClient`'s - :attr:`operation_status` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") - self._config: KubernetesConfigurationExtensionsMgmtClientConfiguration = ( - input_args.pop(0) if input_args else kwargs.pop("config") - ) - self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def get( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - operation_id: str, - **kwargs: Any - ) -> _models.OperationStatusResult: - """Get Async Operation status. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :param operation_id: operation Id. Required. - :type operation_id: str - :return: OperationStatusResult or the result of cls(response) - :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.OperationStatusResult - :raises ~azure.core.exceptions.HttpResponseError: - """ - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _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.OperationStatusResult] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - operation_id=operation_id, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = 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) - - deserialized = self._deserialize("OperationStatusResult", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_extensions_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operations.py similarity index 64% rename from sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_extensions_operations.py rename to sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operations.py index f29f78d380b8..d437d8d7ff78 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_extensions_operations.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_operations.py @@ -3,12 +3,13 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- from collections.abc import MutableMapping from io import IOBase -from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload +import json +from typing import Any, Callable, IO, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core import PipelineClient @@ -33,16 +34,19 @@ from .. import models as _models from .._configuration import KubernetesConfigurationExtensionsMgmtClientConfiguration +from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._utils.serialization import Deserializer, Serializer +JSON = MutableMapping[str, Any] T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, dict[str, Any]], Any]] +List = list _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_create_request( +def build_extensions_get_request( resource_group_name: str, cluster_rp: str, cluster_resource_name: str, @@ -55,24 +59,54 @@ def build_create_request( _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), + "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_extensions_create_request( + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}", - ) + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}" path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), } @@ -89,7 +123,7 @@ def build_create_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_get_request( +def build_extensions_update_request( resource_group_name: str, cluster_rp: str, cluster_resource_name: str, @@ -101,24 +135,18 @@ def build_get_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}", - ) + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}" path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), } @@ -128,12 +156,14 @@ def build_get_request( _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_delete_request( +def build_extensions_delete_request( resource_group_name: str, cluster_rp: str, cluster_resource_name: str, @@ -144,27 +174,17 @@ def build_delete_request( force_delete: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) - accept = _headers.pop("Accept", "application/json") - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}", - ) + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}" path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), } @@ -175,18 +195,14 @@ def build_delete_request( if force_delete is not None: _params["forceDelete"] = _SERIALIZER.query("force_delete", force_delete, "bool") - # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_update_request( +def build_extensions_list_request( resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, - extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: @@ -194,25 +210,16 @@ def build_update_request( _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2024-11-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}", - ) + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions" path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), - "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), } _url: str = _url.format(**path_format_arguments) # type: ignore @@ -221,18 +228,18 @@ def build_update_request( _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_list_request( +def build_operation_status_get_request( resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, + extension_name: str, + operation_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: @@ -243,20 +250,15 @@ def build_list_request( accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions", - ) + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{clusterRp}/{clusterResourceName}/{clusterName}/providers/Microsoft.KubernetesConfiguration/extensions/{extensionName}/operations/{operationId}" path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str", min_length=1), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), "clusterRp": _SERIALIZER.url("cluster_rp", cluster_rp, "str"), - "clusterResourceName": _SERIALIZER.url( - "cluster_resource_name", cluster_resource_name, "str", pattern=r"^[a-zA-Z]*$" - ), - "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str", pattern=r"^.*"), + "clusterResourceName": _SERIALIZER.url("cluster_resource_name", cluster_resource_name, "str"), + "clusterName": _SERIALIZER.url("cluster_name", cluster_name, "str"), + "extensionName": _SERIALIZER.url("extension_name", extension_name, "str"), + "operationId": _SERIALIZER.url("operation_id", operation_id, "str"), } _url: str = _url.format(**path_format_arguments) # type: ignore @@ -280,9 +282,7 @@ class ExtensionsOperations: :attr:`extensions` attribute. """ - models = _models - - def __init__(self, *args, **kwargs): + def __init__(self, *args, **kwargs) -> None: input_args = list(args) self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") self._config: KubernetesConfigurationExtensionsMgmtClientConfiguration = ( @@ -291,6 +291,95 @@ def __init__(self, *args, **kwargs): self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + @distributed_trace + def get( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + **kwargs: Any + ) -> _models.Extension: + """Gets Kubernetes Cluster Extension. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :return: Extension. The Extension is compatible with MutableMapping + :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.Extension] = kwargs.pop("cls", None) + + _request = build_extensions_get_request( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.Extension, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + def _create_initial( self, resource_group_name: str, @@ -298,7 +387,7 @@ def _create_initial( cluster_resource_name: str, cluster_name: str, extension_name: str, - extension: Union[_models.Extension, IO[bytes]], + extension: Union[_models.Extension, JSON, IO[bytes]], **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { @@ -310,35 +399,35 @@ def _create_initial( error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" - _json = None _content = None if isinstance(extension, (IOBase, bytes)): _content = extension else: - _json = self._serialize.body(extension, "Extension") + _content = json.dumps(extension, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore - _request = build_create_request( + _request = build_extensions_create_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, subscription_id=self._config.subscription_id, - api_version=api_version, content_type=content_type, - json=_json, + api_version=self._config.api_version, content=_content, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _decompress = kwargs.pop("decompress", True) _stream = True @@ -354,13 +443,23 @@ def _create_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) + response_headers = {} + if response.status_code == 201: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized # type: ignore @@ -397,7 +496,48 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :return: An instance of LROPoller that returns either Extension or the result of cls(response) + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + extension: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.Extension]: + """Create a new Kubernetes Cluster Extension. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param extension: Properties necessary to Create an Extension. Required. + :type extension: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -436,7 +576,8 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :return: An instance of LROPoller that returns either Extension or the result of cls(response) + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: @@ -450,7 +591,7 @@ def begin_create( cluster_resource_name: str, cluster_name: str, extension_name: str, - extension: Union[_models.Extension, IO[bytes]], + extension: Union[_models.Extension, JSON, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Extension]: """Create a new Kubernetes Cluster Extension. @@ -468,18 +609,19 @@ def begin_create( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param extension: Properties necessary to Create an Extension. Is either a Extension type or a - IO[bytes] type. Required. - :type extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension or IO[bytes] - :return: An instance of LROPoller that returns either Extension or the result of cls(response) + :param extension: Properties necessary to Create an Extension. Is one of the following types: + Extension, JSON, IO[bytes] Required. + :type extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension or JSON or + IO[bytes] + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[_models.Extension] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) @@ -493,7 +635,6 @@ def begin_create( cluster_name=cluster_name, extension_name=extension_name, extension=extension, - api_version=api_version, content_type=content_type, cls=lambda x, y, z: x, headers=_headers, @@ -504,14 +645,19 @@ def begin_create( kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Extension", pipeline_response.http_response) + response = pipeline_response.http_response + deserialized = _deserialize(_models.Extension, response.json()) if cls: return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + if polling is True: polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs) + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) @@ -528,35 +674,16 @@ def get_long_running_output(pipeline_response): self._client, raw_result, get_long_running_output, polling_method # type: ignore ) - @distributed_trace - def get( + def _update_initial( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, + patch_extension: Union[_models.PatchExtension, JSON, IO[bytes]], **kwargs: Any - ) -> _models.Extension: - """Gets Kubernetes Cluster Extension. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :return: Extension or the result of cls(response) - :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.Extension - :raises ~azure.core.exceptions.HttpResponseError: - """ + ) -> Iterator[bytes]: error_map: MutableMapping = { 401: ClientAuthenticationError, 404: ResourceNotFoundError, @@ -565,119 +692,203 @@ def get( } error_map.update(kwargs.pop("error_map", {}) or {}) - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.Extension] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(patch_extension, (IOBase, bytes)): + _content = patch_extension + else: + _content = json.dumps(patch_extension, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore - _request = build_get_request( + _request = build_extensions_update_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, subscription_id=self._config.subscription_id, - api_version=api_version, + content_type=content_type, + api_version=self._config.api_version, + content=_content, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) - _stream = False + _decompress = kwargs.pop("decompress", True) + _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access _request, stream=_stream, **kwargs ) response = pipeline_response.http_response - if response.status_code not in [200]: + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("Extension", pipeline_response.http_response) + response_headers = {} + if response.status_code == 202: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized # type: ignore - def _delete_initial( + @overload + def begin_update( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - force_delete: Optional[bool] = None, + patch_extension: _models.PatchExtension, + *, + content_type: str = "application/json", **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) + ) -> LROPoller[_models.Extension]: + """Patch an existing Kubernetes Cluster Extension. - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param patch_extension: Properties to Patch in an existing Extension. Required. + :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :raises ~azure.core.exceptions.HttpResponseError: + """ - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - cluster_rp=cluster_rp, - cluster_resource_name=cluster_resource_name, - cluster_name=cluster_name, - extension_name=extension_name, - subscription_id=self._config.subscription_id, - force_delete=force_delete, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202, 204]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - 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) + @overload + def begin_update( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + patch_extension: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.Extension]: + """Patch an existing Kubernetes Cluster Extension. - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param patch_extension: Properties to Patch in an existing Extension. Required. + :type patch_extension: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :raises ~azure.core.exceptions.HttpResponseError: + """ - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + @overload + def begin_update( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + patch_extension: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.Extension]: + """Patch an existing Kubernetes Cluster Extension. - return deserialized # type: ignore + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param patch_extension: Properties to Patch in an existing Extension. Required. + :type patch_extension: IO[bytes] + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :raises ~azure.core.exceptions.HttpResponseError: + """ @distributed_trace - def begin_delete( + def begin_update( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - force_delete: Optional[bool] = None, + patch_extension: Union[_models.PatchExtension, JSON, IO[bytes]], **kwargs: Any - ) -> LROPoller[None]: - """Delete a Kubernetes Cluster Extension. This will cause the Agent to Uninstall the extension - from the cluster. + ) -> LROPoller[_models.Extension]: + """Patch an existing Kubernetes Cluster Extension. :param resource_group_name: The name of the resource group. The name is case insensitive. Required. @@ -692,30 +903,33 @@ def begin_delete( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param force_delete: Delete the extension resource in Azure - not the normal asynchronous - delete. Default value is None. - :type force_delete: bool - :return: An instance of LROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.LROPoller[None] + :param patch_extension: Properties to Patch in an existing Extension. Is one of the following + types: PatchExtension, JSON, IO[bytes] Required. + :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension or + JSON or IO[bytes] + :return: An instance of LROPoller that returns Extension. The Extension is compatible with + MutableMapping + :rtype: + ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Extension] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: - raw_result = self._delete_initial( + raw_result = self._update_initial( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, - force_delete=force_delete, - api_version=api_version, + patch_extension=patch_extension, + content_type=content_type, cls=lambda x, y, z: x, headers=_headers, params=_params, @@ -724,35 +938,45 @@ def begin_delete( raw_result.http_response.read() # type: ignore kwargs.pop("error_map", None) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.Extension, response.json()) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } if polling is True: polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs) + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: polling_method = polling if cont_token: - return LROPoller[None].from_continuation_token( + return LROPoller[_models.Extension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return LROPoller[_models.Extension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) - def _update_initial( + def _delete_initial( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - patch_extension: Union[_models.PatchExtension, IO[bytes]], + *, + force_delete: Optional[bool] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { @@ -763,36 +987,27 @@ def _update_initial( } error_map.update(kwargs.pop("error_map", {}) or {}) - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(patch_extension, (IOBase, bytes)): - _content = patch_extension - else: - _json = self._serialize.body(patch_extension, "PatchExtension") - - _request = build_update_request( + _request = build_extensions_delete_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, + force_delete=force_delete, + api_version=self._config.api_version, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) _decompress = kwargs.pop("decompress", True) _stream = True @@ -802,112 +1017,46 @@ def _update_initial( response = pipeline_response.http_response - if response.status_code not in [200, 202]: + if response.status_code not in [200, 202, 204]: try: response.read() # Load the body in memory and close the socket except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) + response_headers = {} + if response.status_code == 202: + response_headers["Azure-AsyncOperation"] = self._deserialize( + "str", response.headers.get("Azure-AsyncOperation") + ) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() if _decompress else response.iter_raw() if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, response_headers) # type: ignore return deserialized # type: ignore - @overload - def begin_update( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - patch_extension: _models.PatchExtension, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.Extension]: - """Patch an existing Kubernetes Cluster Extension. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :param patch_extension: Properties to Patch in an existing Extension. Required. - :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either Extension or the result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_update( - self, - resource_group_name: str, - cluster_rp: str, - cluster_resource_name: str, - cluster_name: str, - extension_name: str, - patch_extension: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.Extension]: - """Patch an existing Kubernetes Cluster Extension. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, - Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. - :type cluster_rp: str - :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, - connectedClusters, provisionedClusters, appliances. Required. - :type cluster_resource_name: str - :param cluster_name: The name of the kubernetes cluster. Required. - :type cluster_name: str - :param extension_name: Name of the Extension. Required. - :type extension_name: str - :param patch_extension: Properties to Patch in an existing Extension. Required. - :type patch_extension: IO[bytes] - :keyword content_type: Body Parameter content-type. Content type parameter for binary body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either Extension or the result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] - :raises ~azure.core.exceptions.HttpResponseError: - """ - @distributed_trace - def begin_update( + def begin_delete( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, extension_name: str, - patch_extension: Union[_models.PatchExtension, IO[bytes]], + *, + force_delete: Optional[bool] = None, **kwargs: Any - ) -> LROPoller[_models.Extension]: - """Patch an existing Kubernetes Cluster Extension. + ) -> LROPoller[None]: + """Delete a Kubernetes Cluster Extension. This will cause the Agent to Uninstall the extension + from the cluster. :param resource_group_name: The name of the resource group. The name is case insensitive. Required. @@ -922,34 +1071,28 @@ def begin_update( :type cluster_name: str :param extension_name: Name of the Extension. Required. :type extension_name: str - :param patch_extension: Properties to Patch in an existing Extension. Is either a - PatchExtension type or a IO[bytes] type. Required. - :type patch_extension: ~azure.mgmt.kubernetesconfiguration.extensions.models.PatchExtension or - IO[bytes] - :return: An instance of LROPoller that returns either Extension or the result of cls(response) - :rtype: - ~azure.core.polling.LROPoller[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] + :keyword force_delete: Delete the extension resource in Azure - not the normal asynchronous + delete. Default value is None. + :paramtype force_delete: bool + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: """ - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Extension] = kwargs.pop("cls", None) + cls: ClsType[None] = kwargs.pop("cls", None) polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: - raw_result = self._update_initial( + raw_result = self._delete_initial( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, extension_name=extension_name, - patch_extension=patch_extension, - api_version=api_version, - content_type=content_type, + force_delete=force_delete, cls=lambda x, y, z: x, headers=_headers, params=_params, @@ -958,35 +1101,35 @@ def begin_update( raw_result.http_response.read() # type: ignore kwargs.pop("error_map", None) - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Extension", pipeline_response.http_response) + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } if polling is True: polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "azure-async-operation"}, **kwargs) + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) ) elif polling is False: polling_method = cast(PollingMethod, NoPolling()) else: polling_method = polling if cont_token: - return LROPoller[_models.Extension].from_continuation_token( + return LROPoller[None].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller[_models.Extension]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list( self, resource_group_name: str, cluster_rp: str, cluster_resource_name: str, cluster_name: str, **kwargs: Any - ) -> Iterable["_models.Extension"]: + ) -> ItemPaged["_models.Extension"]: """List all Extensions in the cluster. :param resource_group_name: The name of the resource group. The name is case insensitive. @@ -1000,16 +1143,15 @@ def list( :type cluster_resource_name: str :param cluster_name: The name of the kubernetes cluster. Required. :type cluster_name: str - :return: An iterator like instance of either Extension or the result of cls(response) + :return: An iterator like instance of Extension :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.kubernetesconfiguration.extensions.models.Extension] :raises ~azure.core.exceptions.HttpResponseError: """ _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + _params = kwargs.pop("params", {}) or {} - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.ExtensionsList] = kwargs.pop("cls", None) + cls: ClsType[List[_models.Extension]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -1022,17 +1164,22 @@ def list( def prepare_request(next_link=None): if not next_link: - _request = build_list_request( + _request = build_extensions_list_request( resource_group_name=resource_group_name, cluster_rp=cluster_rp, cluster_resource_name=cluster_resource_name, cluster_name=cluster_name, subscription_id=self._config.subscription_id, - api_version=api_version, + api_version=self._config.api_version, headers=_headers, params=_params, ) - _request.url = self._client.format_url(_request.url) + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) else: # make call to next link with the client's api-version @@ -1047,16 +1194,24 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request def extract_data(pipeline_response): - deserialized = self._deserialize("ExtensionsList", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Extension], + deserialized.get("value", []), + ) if cls: list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) + return deserialized.get("nextLink") or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) @@ -1069,9 +1224,125 @@ def get_next(next_link=None): 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) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return ItemPaged(get_next, extract_data) + + +class OperationStatusOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.kubernetesconfiguration.extensions.KubernetesConfigurationExtensionsMgmtClient`'s + :attr:`operation_status` attribute. + """ + + def __init__(self, *args, **kwargs) -> None: + input_args = list(args) + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: KubernetesConfigurationExtensionsMgmtClientConfiguration = ( + input_args.pop(0) if input_args else kwargs.pop("config") + ) + self._serialize: Serializer = input_args.pop(0) if input_args else kwargs.pop("serializer") + self._deserialize: Deserializer = input_args.pop(0) if input_args else kwargs.pop("deserializer") + + @distributed_trace + def get( + self, + resource_group_name: str, + cluster_rp: str, + cluster_resource_name: str, + cluster_name: str, + extension_name: str, + operation_id: str, + **kwargs: Any + ) -> _models.OperationStatusResult: + """Get Async Operation status. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param cluster_rp: The Kubernetes cluster RP - i.e. Microsoft.ContainerService, + Microsoft.Kubernetes, Microsoft.HybridContainerService. Required. + :type cluster_rp: str + :param cluster_resource_name: The Kubernetes cluster resource name - i.e. managedClusters, + connectedClusters, provisionedClusters, appliances. Required. + :type cluster_resource_name: str + :param cluster_name: The name of the kubernetes cluster. Required. + :type cluster_name: str + :param extension_name: Name of the Extension. Required. + :type extension_name: str + :param operation_id: Required. + :type operation_id: str + :return: OperationStatusResult. The OperationStatusResult is compatible with MutableMapping + :rtype: ~azure.mgmt.kubernetesconfiguration.extensions.models.OperationStatusResult + :raises ~azure.core.exceptions.HttpResponseError: + """ + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[_models.OperationStatusResult] = kwargs.pop("cls", None) + + _request = build_operation_status_get_request( + resource_group_name=resource_group_name, + cluster_rp=cluster_rp, + cluster_resource_name=cluster_resource_name, + cluster_name=cluster_name, + extension_name=extension_name, + operation_id=operation_id, + subscription_id=self._config.subscription_id, + api_version=self._config.api_version, + headers=_headers, + params=_params, + ) + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + + _decompress = kwargs.pop("decompress", True) + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.ErrorResponse, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() if _decompress else response.iter_raw() + else: + deserialized = _deserialize(_models.OperationStatusResult, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_patch.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_patch.py index 8bcb627aa475..87676c65a8f0 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_patch.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/azure/mgmt/kubernetesconfiguration/extensions/operations/_patch.py @@ -7,9 +7,9 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize """ -from typing import List -__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level def patch_sdk(): diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension.py index e1aab3df588f..2fb6e670243d 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.begin_create( @@ -54,6 +53,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/CreateExtension.json +# x-ms-original-file: 2024-11-01/CreateExtension.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension_with_plan.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension_with_plan.py index 9510bc1c84d9..a0023d48d24f 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension_with_plan.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/create_extension_with_plan.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.begin_create( @@ -49,6 +48,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/CreateExtensionWithPlan.json +# x-ms-original-file: 2024-11-01/CreateExtensionWithPlan.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/delete_extension.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/delete_extension.py index 773d11aa214c..a497351de86e 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/delete_extension.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/delete_extension.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) client.extensions.begin_delete( @@ -40,6 +39,6 @@ def main(): ).result() -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/DeleteExtension.json +# x-ms-original-file: 2024-11-01/DeleteExtension.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension.py index fcc1ed806737..f4b9b93381e9 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.get( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/GetExtension.json +# x-ms-original-file: 2024-11-01/GetExtension.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_async_operation_status.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_async_operation_status.py index ffed4540834b..4448e02c2884 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_async_operation_status.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_async_operation_status.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.operation_status.get( @@ -42,6 +41,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/GetExtensionAsyncOperationStatus.json +# x-ms-original-file: 2024-11-01/GetExtensionAsyncOperationStatus.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_with_plan.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_with_plan.py index 21d8963241c0..35131178dde4 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_with_plan.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/get_extension_with_plan.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.get( @@ -41,6 +40,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/GetExtensionWithPlan.json +# x-ms-original-file: 2024-11-01/GetExtensionWithPlan.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/list_extensions.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/list_extensions.py index 3fefe69ffa2d..4f21362fb81b 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/list_extensions.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/list_extensions.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.list( @@ -41,6 +40,6 @@ def main(): print(item) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/ListExtensions.json +# x-ms-original-file: 2024-11-01/ListExtensions.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/patch_extension.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/patch_extension.py index 6e4b7896f7b1..86b5d20110b1 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/patch_extension.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_samples/patch_extension.py @@ -1,9 +1,8 @@ -# pylint: disable=line-too-long,useless-suppression # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -28,7 +27,7 @@ def main(): client = KubernetesConfigurationExtensionsMgmtClient( credential=DefaultAzureCredential(), - subscription_id="subId1", + subscription_id="SUBSCRIPTION_ID", ) response = client.extensions.begin_update( @@ -52,6 +51,6 @@ def main(): print(response) -# x-ms-original-file: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions/stable/2024-11-01/examples/PatchExtension.json +# x-ms-original-file: 2024-11-01/PatchExtension.json if __name__ == "__main__": main() diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/conftest.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/conftest.py index 97c89863bec1..c373774ecda8 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/conftest.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/conftest.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import os diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations.py index eaa52d483bf7..8476b955fc22 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest @@ -18,6 +18,20 @@ class TestKubernetesConfigurationExtensionsMgmtExtensionsOperations(AzureMgmtRec def setup_method(self, method): self.client = self.create_mgmt_client(KubernetesConfigurationExtensionsMgmtClient) + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_extensions_get(self, resource_group): + response = self.client.extensions.get( + resource_group_name=resource_group.name, + cluster_rp="str", + cluster_resource_name="str", + cluster_name="str", + extension_name="str", + ) + + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_extensions_begin_create(self, resource_group): @@ -28,32 +42,35 @@ def test_extensions_begin_create(self, resource_group): cluster_name="str", extension_name="str", extension={ - "aksAssignedIdentity": {"principalId": "str", "tenantId": "str", "type": "str"}, - "autoUpgradeMinorVersion": True, - "configurationProtectedSettings": {"str": "str"}, - "configurationSettings": {"str": "str"}, - "currentVersion": "str", - "customLocationSettings": {"str": "str"}, - "errorInfo": { - "additionalInfo": [{"info": {}, "type": "str"}], - "code": "str", - "details": [...], - "message": "str", - "target": "str", - }, - "extensionType": "str", "id": "str", - "identity": {"principalId": "str", "tenantId": "str", "type": "SystemAssigned"}, - "isSystemExtension": False, + "identity": {"principalId": "str", "tenantId": "str", "type": "str"}, "name": "str", - "packageUri": "str", "plan": {"name": "str", "product": "str", "publisher": "str", "promotionCode": "str", "version": "str"}, - "provisioningState": "str", - "releaseTrain": "Stable", - "scope": {"cluster": {"releaseNamespace": "str"}, "namespace": {"targetNamespace": "str"}}, - "statuses": [ - {"code": "str", "displayStatus": "str", "level": "Information", "message": "str", "time": "str"} - ], + "properties": { + "aksAssignedIdentity": {"principalId": "str", "tenantId": "str", "type": "str"}, + "autoUpgradeMinorVersion": bool, + "configurationProtectedSettings": {"str": "str"}, + "configurationSettings": {"str": "str"}, + "currentVersion": "str", + "customLocationSettings": {"str": "str"}, + "errorInfo": { + "additionalInfo": [{"info": {}, "type": "str"}], + "code": "str", + "details": [...], + "message": "str", + "target": "str", + }, + "extensionType": "str", + "isSystemExtension": bool, + "packageUri": "str", + "provisioningState": "str", + "releaseTrain": "str", + "scope": {"cluster": {"releaseNamespace": "str"}, "namespace": {"targetNamespace": "str"}}, + "statuses": [ + {"code": "str", "displayStatus": "str", "level": "str", "message": "str", "time": "str"} + ], + "version": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -63,9 +80,7 @@ def test_extensions_begin_create(self, resource_group): "lastModifiedByType": "str", }, "type": "str", - "version": "str", }, - api_version="2024-11-01", ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself @@ -73,15 +88,23 @@ def test_extensions_begin_create(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_extensions_get(self, resource_group): - response = self.client.extensions.get( + def test_extensions_begin_update(self, resource_group): + response = self.client.extensions.begin_update( resource_group_name=resource_group.name, cluster_rp="str", cluster_resource_name="str", cluster_name="str", extension_name="str", - api_version="2024-11-01", - ) + patch_extension={ + "properties": { + "autoUpgradeMinorVersion": bool, + "configurationProtectedSettings": {"str": "str"}, + "configurationSettings": {"str": "str"}, + "releaseTrain": "str", + "version": "str", + } + }, + ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself # ... @@ -95,29 +118,6 @@ def test_extensions_begin_delete(self, resource_group): cluster_resource_name="str", cluster_name="str", extension_name="str", - api_version="2024-11-01", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_extensions_begin_update(self, resource_group): - response = self.client.extensions.begin_update( - resource_group_name=resource_group.name, - cluster_rp="str", - cluster_resource_name="str", - cluster_name="str", - extension_name="str", - patch_extension={ - "autoUpgradeMinorVersion": True, - "configurationProtectedSettings": {"str": "str"}, - "configurationSettings": {"str": "str"}, - "releaseTrain": "Stable", - "version": "str", - }, - api_version="2024-11-01", ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself @@ -131,7 +131,6 @@ def test_extensions_list(self, resource_group): cluster_rp="str", cluster_resource_name="str", cluster_name="str", - api_version="2024-11-01", ) result = [r for r in response] # please add some check logic here by yourself diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations_async.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations_async.py index c1d7c390373d..531d9a6e9489 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations_async.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_extensions_operations_async.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest @@ -19,6 +19,20 @@ class TestKubernetesConfigurationExtensionsMgmtExtensionsOperationsAsync(AzureMg def setup_method(self, method): self.client = self.create_mgmt_client(KubernetesConfigurationExtensionsMgmtClient, is_async=True) + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_extensions_get(self, resource_group): + response = await self.client.extensions.get( + resource_group_name=resource_group.name, + cluster_rp="str", + cluster_resource_name="str", + cluster_name="str", + extension_name="str", + ) + + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async async def test_extensions_begin_create(self, resource_group): @@ -30,25 +44,9 @@ async def test_extensions_begin_create(self, resource_group): cluster_name="str", extension_name="str", extension={ - "aksAssignedIdentity": {"principalId": "str", "tenantId": "str", "type": "str"}, - "autoUpgradeMinorVersion": True, - "configurationProtectedSettings": {"str": "str"}, - "configurationSettings": {"str": "str"}, - "currentVersion": "str", - "customLocationSettings": {"str": "str"}, - "errorInfo": { - "additionalInfo": [{"info": {}, "type": "str"}], - "code": "str", - "details": [...], - "message": "str", - "target": "str", - }, - "extensionType": "str", "id": "str", - "identity": {"principalId": "str", "tenantId": "str", "type": "SystemAssigned"}, - "isSystemExtension": False, + "identity": {"principalId": "str", "tenantId": "str", "type": "str"}, "name": "str", - "packageUri": "str", "plan": { "name": "str", "product": "str", @@ -56,12 +54,31 @@ async def test_extensions_begin_create(self, resource_group): "promotionCode": "str", "version": "str", }, - "provisioningState": "str", - "releaseTrain": "Stable", - "scope": {"cluster": {"releaseNamespace": "str"}, "namespace": {"targetNamespace": "str"}}, - "statuses": [ - {"code": "str", "displayStatus": "str", "level": "Information", "message": "str", "time": "str"} - ], + "properties": { + "aksAssignedIdentity": {"principalId": "str", "tenantId": "str", "type": "str"}, + "autoUpgradeMinorVersion": bool, + "configurationProtectedSettings": {"str": "str"}, + "configurationSettings": {"str": "str"}, + "currentVersion": "str", + "customLocationSettings": {"str": "str"}, + "errorInfo": { + "additionalInfo": [{"info": {}, "type": "str"}], + "code": "str", + "details": [...], + "message": "str", + "target": "str", + }, + "extensionType": "str", + "isSystemExtension": bool, + "packageUri": "str", + "provisioningState": "str", + "releaseTrain": "str", + "scope": {"cluster": {"releaseNamespace": "str"}, "namespace": {"targetNamespace": "str"}}, + "statuses": [ + {"code": "str", "displayStatus": "str", "level": "str", "message": "str", "time": "str"} + ], + "version": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -71,9 +88,7 @@ async def test_extensions_begin_create(self, resource_group): "lastModifiedByType": "str", }, "type": "str", - "version": "str", }, - api_version="2024-11-01", ) ).result() # call '.result()' to poll until service return final result @@ -82,30 +97,23 @@ async def test_extensions_begin_create(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_extensions_get(self, resource_group): - response = await self.client.extensions.get( - resource_group_name=resource_group.name, - cluster_rp="str", - cluster_resource_name="str", - cluster_name="str", - extension_name="str", - api_version="2024-11-01", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_extensions_begin_delete(self, resource_group): + async def test_extensions_begin_update(self, resource_group): response = await ( - await self.client.extensions.begin_delete( + await self.client.extensions.begin_update( resource_group_name=resource_group.name, cluster_rp="str", cluster_resource_name="str", cluster_name="str", extension_name="str", - api_version="2024-11-01", + patch_extension={ + "properties": { + "autoUpgradeMinorVersion": bool, + "configurationProtectedSettings": {"str": "str"}, + "configurationSettings": {"str": "str"}, + "releaseTrain": "str", + "version": "str", + } + }, ) ).result() # call '.result()' to poll until service return final result @@ -114,22 +122,14 @@ async def test_extensions_begin_delete(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_extensions_begin_update(self, resource_group): + async def test_extensions_begin_delete(self, resource_group): response = await ( - await self.client.extensions.begin_update( + await self.client.extensions.begin_delete( resource_group_name=resource_group.name, cluster_rp="str", cluster_resource_name="str", cluster_name="str", extension_name="str", - patch_extension={ - "autoUpgradeMinorVersion": True, - "configurationProtectedSettings": {"str": "str"}, - "configurationSettings": {"str": "str"}, - "releaseTrain": "Stable", - "version": "str", - }, - api_version="2024-11-01", ) ).result() # call '.result()' to poll until service return final result @@ -144,7 +144,6 @@ async def test_extensions_list(self, resource_group): cluster_rp="str", cluster_resource_name="str", cluster_name="str", - api_version="2024-11-01", ) result = [r async for r in response] # please add some check logic here by yourself diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations.py index 2b94611a92fd..80a424684c99 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest @@ -28,7 +28,6 @@ def test_operation_status_get(self, resource_group): cluster_name="str", extension_name="str", operation_id="str", - api_version="2024-11-01", ) # please add some check logic here by yourself diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations_async.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations_async.py index 1bab80b267b0..7d07694ba4ca 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations_async.py +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/generated_tests/test_kubernetes_configuration_extensions_mgmt_operation_status_operations_async.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # 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. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest @@ -29,7 +29,6 @@ async def test_operation_status_get(self, resource_group): cluster_name="str", extension_name="str", operation_id="str", - api_version="2024-11-01", ) # please add some check logic here by yourself diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/pyproject.toml b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/pyproject.toml index ee542ddce18b..7778b74b022a 100644 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/pyproject.toml +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/pyproject.toml @@ -1,4 +1,87 @@ +[build-system] +requires = [ + "setuptools>=77.0.3", + "wheel", +] +build-backend = "setuptools.build_meta" + +[project] +name = "azure-mgmt-kubernetesconfiguration-extensions" +authors = [ + { name = "Microsoft Corporation", email = "azpysdkhelp@microsoft.com" }, +] +description = "Microsoft Azure Extensions Management Client Library for Python" +license = "MIT" +classifiers = [ + "Development Status :: 4 - Beta", + "Programming Language :: Python", + "Programming Language :: Python :: 3 :: Only", + "Programming Language :: Python :: 3", + "Programming Language :: Python :: 3.9", + "Programming Language :: Python :: 3.10", + "Programming Language :: Python :: 3.11", + "Programming Language :: Python :: 3.12", + "Programming Language :: Python :: 3.13", +] +requires-python = ">=3.9" +keywords = [ + "azure", + "azure sdk", +] +dependencies = [ + "isodate>=0.6.1", + "azure-mgmt-core>=1.6.0", + "typing-extensions>=4.6.0", +] +dynamic = [ + "version", + "readme", +] + +[project.urls] +repository = "https://github.com/Azure/azure-sdk-for-python" + +[tool.setuptools.dynamic.version] +attr = "azure.mgmt.kubernetesconfiguration.extensions._version.VERSION" + +[tool.setuptools.dynamic.readme] +file = [ + "README.md", + "CHANGELOG.md", +] +content-type = "text/markdown" + +[tool.setuptools.packages.find] +exclude = [ + "tests*", + "generated_tests*", + "samples*", + "generated_samples*", + "doc*", + "azure", + "azure.mgmt", + "azure.mgmt.kubernetesconfiguration", +] + +[tool.setuptools.package-data] +pytyped = [ + "py.typed", +] + [tool.azure-sdk-build] breaking = false mypy = false pyright = false + +[packaging] +package_name = "azure-mgmt-kubernetesconfiguration-extensions" +package_nspkg = "azure-mgmt-kubernetesconfiguration-nspkg" +package_pprint_name = "Kubernetesconfiguration-extensions Management" +package_doc_id = "" +is_stable = false +is_arm = true +need_msrestazure = false +need_azuremgmtcore = true +sample_link = "" +exclude_folders = "" +title = "KubernetesConfigurationExtensionsMgmtClient" diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/sdk_packaging.toml b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/sdk_packaging.toml deleted file mode 100644 index 4d593a6ca657..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/sdk_packaging.toml +++ /dev/null @@ -1,12 +0,0 @@ -[packaging] -package_name = "azure-mgmt-kubernetesconfiguration-extensions" -package_nspkg = "azure-mgmt-kubernetesconfiguration-nspkg" -package_pprint_name = "Kubernetesconfiguration-extensions Management" -package_doc_id = "" -is_stable = true -is_arm = true -need_msrestazure = false -need_azuremgmtcore = true -sample_link = "" -exclude_folders = "" -title = "KubernetesConfigurationExtensionsMgmtClient" diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/setup.py b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/setup.py deleted file mode 100644 index ba737eaa55b7..000000000000 --- a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/setup.py +++ /dev/null @@ -1,83 +0,0 @@ -#!/usr/bin/env python - -# ------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for -# license information. -# -------------------------------------------------------------------------- - -import re -import os.path -from io import open -from setuptools import find_packages, setup - -# Change the PACKAGE_NAME only to change folder and different name -PACKAGE_NAME = "azure-mgmt-kubernetesconfiguration-extensions" -PACKAGE_PPRINT_NAME = "Kubernetesconfiguration-extensions Management" - -# a-b-c => a/b/c -package_folder_path = PACKAGE_NAME.replace("-", "/") -# a-b-c => a.b.c -namespace_name = PACKAGE_NAME.replace("-", ".") - -# Version extraction inspired from 'requests' -with open( - os.path.join(package_folder_path, "version.py") - if os.path.exists(os.path.join(package_folder_path, "version.py")) - else os.path.join(package_folder_path, "_version.py"), - "r", -) as fd: - version = re.search(r'^VERSION\s*=\s*[\'"]([^\'"]*)[\'"]', fd.read(), re.MULTILINE).group(1) - -if not version: - raise RuntimeError("Cannot find version information") - -with open("README.md", encoding="utf-8") as f: - readme = f.read() -with open("CHANGELOG.md", encoding="utf-8") as f: - changelog = f.read() - -setup( - name=PACKAGE_NAME, - version=version, - description="Microsoft Azure {} Client Library for Python".format(PACKAGE_PPRINT_NAME), - long_description=readme + "\n\n" + changelog, - long_description_content_type="text/markdown", - license="MIT License", - author="Microsoft Corporation", - author_email="azpysdkhelp@microsoft.com", - url="https://github.com/Azure/azure-sdk-for-python", - keywords="azure, azure sdk", # update with search keywords relevant to the azure service / product - classifiers=[ - "Development Status :: 5 - Production/Stable", - "Programming Language :: Python", - "Programming Language :: Python :: 3 :: Only", - "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3.9", - "Programming Language :: Python :: 3.10", - "Programming Language :: Python :: 3.11", - "Programming Language :: Python :: 3.12", - "License :: OSI Approved :: MIT License", - ], - zip_safe=False, - packages=find_packages( - exclude=[ - "tests", - # Exclude packages that will be covered by PEP420 or nspkg - "azure", - "azure.mgmt", - "azure.mgmt.kubernetesconfiguration", - ] - ), - include_package_data=True, - package_data={ - "pytyped": ["py.typed"], - }, - install_requires=[ - "isodate>=0.6.1", - "typing-extensions>=4.6.0", - "azure-common>=1.1", - "azure-mgmt-core>=1.5.0", - ], - python_requires=">=3.9", -) diff --git a/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/tsp-location.yaml b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/tsp-location.yaml new file mode 100644 index 000000000000..e80f76cc3f9b --- /dev/null +++ b/sdk/kubernetesconfiguration/azure-mgmt-kubernetesconfiguration-extensions/tsp-location.yaml @@ -0,0 +1,4 @@ +directory: specification/kubernetesconfiguration/resource-manager/Microsoft.KubernetesConfiguration/extensions +commit: bf2f971b4319da9f5624da7f3c13d487be477897 +repo: Azure/azure-rest-api-specs +additionalDirectories: