From 243e7ab6a2ece7edd7f85a6a7a1060404d06877f Mon Sep 17 00:00:00 2001 From: azure-sdk Date: Fri, 22 May 2026 16:01:27 +0000 Subject: [PATCH 1/6] Configurations: 'specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity/tspconfig.yaml', API Version: 2025-05-31-preview, SDK Release Type: beta, and CommitSHA: '7811c124444dfbd4bd149ad88f2648854dd9201b' in SpecRepo: 'https://github.com/Azure/azure-rest-api-specs' Pipeline run: https://dev.azure.com/azure-sdk/internal/_build/results?buildId=6340545 Refer to https://eng.ms/docs/products/azure-developer-experience/develop/sdk-release/sdk-release-prerequisites to prepare for SDK release. --- sdk/resources/azure-mgmt-msi/CHANGELOG.md | 24 + sdk/resources/azure-mgmt-msi/MANIFEST.in | 8 +- sdk/resources/azure-mgmt-msi/README.md | 4 +- sdk/resources/azure-mgmt-msi/_metadata.json | 15 +- .../azure-mgmt-msi/apiview-properties.json | 88 +- .../azure-mgmt-msi/azure/__init__.py | 2 +- .../azure-mgmt-msi/azure/mgmt/__init__.py | 2 +- .../azure-mgmt-msi/azure/mgmt/msi/__init__.py | 6 +- ..._service_identity_client.py => _client.py} | 81 +- .../azure/mgmt/msi/_configuration.py | 37 +- .../azure-mgmt-msi/azure/mgmt/msi/_patch.py | 38 +- .../azure/mgmt/msi/_utils/__init__.py | 2 +- .../azure/mgmt/msi/_utils/model_base.py | 1737 +++++++++++++++++ .../azure/mgmt/msi/_utils/serialization.py | 74 +- .../azure-mgmt-msi/azure/mgmt/msi/_version.py | 4 +- .../azure/mgmt/msi/aio/__init__.py | 6 +- ..._service_identity_client.py => _client.py} | 83 +- .../azure/mgmt/msi/aio/_configuration.py | 37 +- .../azure/mgmt/msi/aio/_patch.py | 3 +- .../azure/mgmt/msi/aio/operations/__init__.py | 12 +- ...derated_identity_credentials_operations.py | 431 ---- .../mgmt/msi/aio/operations/_operations.py | 1322 ++++++++++++- .../azure/mgmt/msi/aio/operations/_patch.py | 3 +- .../_system_assigned_identities_operations.py | 105 - .../_user_assigned_identities_operations.py | 583 ------ .../azure/mgmt/msi/models/__init__.py | 28 +- ...ice_identity_client_enums.py => _enums.py} | 17 +- .../azure/mgmt/msi/models/_models.py | 767 ++++++++ .../azure/mgmt/msi/models/_models_py3.py | 685 ------- .../azure/mgmt/msi/models/_patch.py | 3 +- .../azure/mgmt/msi/operations/__init__.py | 12 +- ...derated_identity_credentials_operations.py | 602 ------ .../azure/mgmt/msi/operations/_operations.py | 1644 +++++++++++++++- .../azure/mgmt/msi/operations/_patch.py | 3 +- .../_system_assigned_identities_operations.py | 131 -- .../_user_assigned_identities_operations.py | 762 -------- .../federated_identity_credential_create.py | 11 +- .../federated_identity_credential_delete.py | 11 +- .../federated_identity_credential_get.py | 11 +- .../federated_identity_credential_list.py | 11 +- ...le_federated_identity_credential_create.py | 53 + ...xible_federated_identity_credential_get.py | 43 + .../generated_samples/identity_create.py | 11 +- .../generated_samples/identity_delete.py | 11 +- .../generated_samples/identity_get.py | 11 +- .../identity_list_by_resource_group.py | 11 +- .../identity_list_by_subscription.py | 11 +- .../generated_samples/identity_update.py | 20 +- .../generated_samples/msi_operations_list.py | 9 +- .../system_assigned_identity_get.py | 10 +- .../generated_tests/conftest.py | 24 +- ...erated_identity_credentials_operations.py} | 39 +- ..._identity_credentials_operations_async.py} | 39 +- ...py => test_managed_identity_operations.py} | 12 +- ...test_managed_identity_operations_async.py} | 12 +- ..._system_assigned_identities_operations.py} | 9 +- ...m_assigned_identities_operations_async.py} | 9 +- ...ty_user_assigned_identities_operations.py} | 67 +- ...r_assigned_identities_operations_async.py} | 67 +- sdk/resources/azure-mgmt-msi/pyproject.toml | 70 + sdk/resources/azure-mgmt-msi/setup.py | 83 - .../azure-mgmt-msi/tsp-location.yaml | 4 + 62 files changed, 6159 insertions(+), 3871 deletions(-) rename sdk/resources/azure-mgmt-msi/azure/mgmt/msi/{_managed_service_identity_client.py => _client.py} (70%) create mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/model_base.py rename sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/{_managed_service_identity_client.py => _client.py} (71%) delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_federated_identity_credentials_operations.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_system_assigned_identities_operations.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_user_assigned_identities_operations.py rename sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/{_managed_service_identity_client_enums.py => _enums.py} (54%) create mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models_py3.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_federated_identity_credentials_operations.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_system_assigned_identities_operations.py delete mode 100644 sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_user_assigned_identities_operations.py create mode 100644 sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py create mode 100644 sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_federated_identity_credentials_operations.py => test_managed_identity_federated_identity_credentials_operations.py} (84%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_federated_identity_credentials_operations_async.py => test_managed_identity_federated_identity_credentials_operations_async.py} (84%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_operations.py => test_managed_identity_operations.py} (72%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_operations_async.py => test_managed_identity_operations_async.py} (72%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_system_assigned_identities_operations.py => test_managed_identity_system_assigned_identities_operations.py} (75%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_system_assigned_identities_operations_async.py => test_managed_identity_system_assigned_identities_operations_async.py} (75%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_user_assigned_identities_operations.py => test_managed_identity_user_assigned_identities_operations.py} (82%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_service_identity_user_assigned_identities_operations_async.py => test_managed_identity_user_assigned_identities_operations_async.py} (82%) delete mode 100644 sdk/resources/azure-mgmt-msi/setup.py create mode 100644 sdk/resources/azure-mgmt-msi/tsp-location.yaml diff --git a/sdk/resources/azure-mgmt-msi/CHANGELOG.md b/sdk/resources/azure-mgmt-msi/CHANGELOG.md index 066331444780..b7e0c8d65234 100644 --- a/sdk/resources/azure-mgmt-msi/CHANGELOG.md +++ b/sdk/resources/azure-mgmt-msi/CHANGELOG.md @@ -1,5 +1,29 @@ # Release History +## 8.0.0b2 (2026-05-22) + +### Features Added + + - Model `Identity` added property `properties` + - Model `IdentityUpdate` added property `properties` + - Added model `AssignmentRestrictions` + - Added model `ClaimsMatchingExpression` + - Added model `CloudError` + - Added model `ExtensionResource` + - Added enum `IsolationScope` + - Added model `UserAssignedIdentityProperties` + +### Breaking Changes + + - Deleted or renamed client `ManagedServiceIdentityClient` + - Model `Identity` deleted or renamed its instance variable `tenant_id` + - Model `Identity` deleted or renamed its instance variable `principal_id` + - Model `Identity` deleted or renamed its instance variable `client_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `tenant_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `principal_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `client_id` + - Method `FederatedIdentityCredentialsOperations.list` changed its parameter `skiptoken` from `positional_or_keyword` to `keyword_only` + ## 8.0.0b1 (2025-08-15) ### Breaking Changes diff --git a/sdk/resources/azure-mgmt-msi/MANIFEST.in b/sdk/resources/azure-mgmt-msi/MANIFEST.in index dfa1c1bd2271..7450fc869bfd 100644 --- a/sdk/resources/azure-mgmt-msi/MANIFEST.in +++ b/sdk/resources/azure-mgmt-msi/MANIFEST.in @@ -1,7 +1,7 @@ -recursive-include tests *.py *.json -recursive-include samples *.py *.md include *.md -include azure/__init__.py -include azure/mgmt/__init__.py include LICENSE include azure/mgmt/msi/py.typed +recursive-include tests *.py +recursive-include samples *.py *.md +include azure/__init__.py +include azure/mgmt/__init__.py diff --git a/sdk/resources/azure-mgmt-msi/README.md b/sdk/resources/azure-mgmt-msi/README.md index 3626705089ac..ea6c0c67db6d 100644 --- a/sdk/resources/azure-mgmt-msi/README.md +++ b/sdk/resources/azure-mgmt-msi/README.md @@ -1,7 +1,7 @@ # Microsoft Azure SDK for Python This is the Microsoft Azure Msi Management Client Library. -This package has been tested with Python 3.9+. +This package has been tested with Python 3.10+. For a more complete view of Azure libraries, see the [azure sdk python release](https://aka.ms/azsdk/python/all). ## _Disclaimer_ @@ -12,7 +12,7 @@ _Azure SDK Python packages support for Python 2.7 has ended 01 January 2022. For ### Prerequisites -- Python 3.9+ is required to use this package. +- Python 3.10+ is required to use this package. - [Azure subscription](https://azure.microsoft.com/free/) ### Install the package diff --git a/sdk/resources/azure-mgmt-msi/_metadata.json b/sdk/resources/azure-mgmt-msi/_metadata.json index b638c838a1bd..f85dbb69c577 100644 --- a/sdk/resources/azure-mgmt-msi/_metadata.json +++ b/sdk/resources/azure-mgmt-msi/_metadata.json @@ -1,11 +1,10 @@ { - "commit": "aa762a31f1e82f62dcc60784f8ebc89c9a1a39e1", + "apiVersion": "2025-05-31-preview", + "apiVersions": { + "Microsoft.ManagedIdentity": "2025-05-31-preview" + }, + "commit": "7811c124444dfbd4bd149ad88f2648854dd9201b", "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest": "3.10.2", - "use": [ - "@autorest/python@6.38.2", - "@autorest/modelerfour@4.27.0" - ], - "autorest_command": "autorest specification/msi/resource-manager/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --keep-setup-py=True --python --python-sdks-folder=/mnt/vss/_work/1/azure-sdk-for-python/sdk --tag=package-2023-01-31 --use=@autorest/python@6.38.2 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", - "readme": "specification/msi/resource-manager/readme.md" + "typespec_src": "specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity", + "emitterVersion": "0.62.1" } \ No newline at end of file diff --git a/sdk/resources/azure-mgmt-msi/apiview-properties.json b/sdk/resources/azure-mgmt-msi/apiview-properties.json index edd275665f41..609eb95b6af8 100644 --- a/sdk/resources/azure-mgmt-msi/apiview-properties.json +++ b/sdk/resources/azure-mgmt-msi/apiview-properties.json @@ -1,44 +1,50 @@ { - "CrossLanguagePackageId": null, + "CrossLanguagePackageId": "Microsoft.ManagedIdentity", "CrossLanguageDefinitionId": { - "azure.mgmt.msi.models.CloudErrorBody": null, - "azure.mgmt.msi.models.Resource": null, - "azure.mgmt.msi.models.ProxyResource": null, - "azure.mgmt.msi.models.FederatedIdentityCredential": null, - "azure.mgmt.msi.models.FederatedIdentityCredentialsListResult": null, - "azure.mgmt.msi.models.TrackedResource": null, - "azure.mgmt.msi.models.Identity": null, - "azure.mgmt.msi.models.IdentityUpdate": null, - "azure.mgmt.msi.models.Operation": null, - "azure.mgmt.msi.models.OperationDisplay": null, - "azure.mgmt.msi.models.OperationListResult": null, - "azure.mgmt.msi.models.SystemAssignedIdentity": null, - "azure.mgmt.msi.models.SystemData": null, - "azure.mgmt.msi.models.UserAssignedIdentitiesListResult": null, - "azure.mgmt.msi.models.CreatedByType": null, - "azure.mgmt.msi.operations.SystemAssignedIdentitiesOperations.get_by_scope": null, - "azure.mgmt.msi.aio.operations.SystemAssignedIdentitiesOperations.get_by_scope": null, - "azure.mgmt.msi.operations.Operations.list": null, - "azure.mgmt.msi.aio.operations.Operations.list": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.list_by_subscription": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.list_by_subscription": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.list_by_resource_group": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.list_by_resource_group": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.create_or_update": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.create_or_update": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.update": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.update": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.get": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.get": null, - "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.delete": null, - "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.delete": null, - "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.list": null, - "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.list": null, - "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.create_or_update": null, - "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.create_or_update": null, - "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.get": null, - "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.get": null, - "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.delete": null, - "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.delete": null - } + "azure.mgmt.msi.models.AssignmentRestrictions": "Microsoft.ManagedIdentity.AssignmentRestrictions", + "azure.mgmt.msi.models.ClaimsMatchingExpression": "Microsoft.ManagedIdentity.ClaimsMatchingExpression", + "azure.mgmt.msi.models.CloudError": "Microsoft.ManagedIdentity.CloudError", + "azure.mgmt.msi.models.CloudErrorBody": "Microsoft.ManagedIdentity.CloudErrorBody", + "azure.mgmt.msi.models.Resource": "Azure.ResourceManager.CommonTypes.Resource", + "azure.mgmt.msi.models.ExtensionResource": "Azure.ResourceManager.CommonTypes.ExtensionResource", + "azure.mgmt.msi.models.ProxyResource": "Azure.ResourceManager.CommonTypes.ProxyResource", + "azure.mgmt.msi.models.FederatedIdentityCredential": "Microsoft.ManagedIdentity.FederatedIdentityCredential", + "azure.mgmt.msi.models.FederatedIdentityCredentialProperties": "Microsoft.ManagedIdentity.FederatedIdentityCredentialProperties", + "azure.mgmt.msi.models.TrackedResource": "Azure.ResourceManager.CommonTypes.TrackedResource", + "azure.mgmt.msi.models.Identity": "Microsoft.ManagedIdentity.Identity", + "azure.mgmt.msi.models.IdentityUpdate": "Microsoft.ManagedIdentity.IdentityUpdate", + "azure.mgmt.msi.models.Operation": "Microsoft.ManagedIdentity.Operation", + "azure.mgmt.msi.models.OperationDisplay": "Microsoft.ManagedIdentity.OperationDisplay", + "azure.mgmt.msi.models.SystemAssignedIdentity": "Microsoft.ManagedIdentity.SystemAssignedIdentity", + "azure.mgmt.msi.models.SystemAssignedIdentityProperties": "Microsoft.ManagedIdentity.SystemAssignedIdentityProperties", + "azure.mgmt.msi.models.SystemData": "Azure.ResourceManager.CommonTypes.SystemData", + "azure.mgmt.msi.models.UserAssignedIdentityProperties": "Microsoft.ManagedIdentity.UserAssignedIdentityProperties", + "azure.mgmt.msi.models.CreatedByType": "Azure.ResourceManager.CommonTypes.createdByType", + "azure.mgmt.msi.models.IsolationScope": "Microsoft.ManagedIdentity.IsolationScope", + "azure.mgmt.msi.operations.Operations.list": "Azure.ResourceManager.Legacy.Operations.list", + "azure.mgmt.msi.aio.operations.Operations.list": "Azure.ResourceManager.Legacy.Operations.list", + "azure.mgmt.msi.operations.SystemAssignedIdentitiesOperations.get_by_scope": "Microsoft.ManagedIdentity.SystemAssignedIdentities.getByScope", + "azure.mgmt.msi.aio.operations.SystemAssignedIdentitiesOperations.get_by_scope": "Microsoft.ManagedIdentity.SystemAssignedIdentities.getByScope", + "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.get": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.get", + "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.get": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.get", + "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.create_or_update": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.createOrUpdate", + "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.create_or_update": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.createOrUpdate", + "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.delete": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.delete", + "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.delete": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.delete", + "azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations.list": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.list", + "azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations.list": "Microsoft.ManagedIdentity.FederatedIdentityCredentials.list", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.get": "Microsoft.ManagedIdentity.Identities.get", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.get": "Microsoft.ManagedIdentity.Identities.get", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.create_or_update": "Microsoft.ManagedIdentity.Identities.createOrUpdate", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.create_or_update": "Microsoft.ManagedIdentity.Identities.createOrUpdate", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.update": "Microsoft.ManagedIdentity.Identities.update", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.update": "Microsoft.ManagedIdentity.Identities.update", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.delete": "Microsoft.ManagedIdentity.Identities.delete", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.delete": "Microsoft.ManagedIdentity.Identities.delete", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.list_by_resource_group": "Microsoft.ManagedIdentity.Identities.listByResourceGroup", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.list_by_resource_group": "Microsoft.ManagedIdentity.Identities.listByResourceGroup", + "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.list_by_subscription": "Microsoft.ManagedIdentity.Identities.listBySubscription", + "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.list_by_subscription": "Microsoft.ManagedIdentity.Identities.listBySubscription" + }, + "CrossLanguageVersion": "54a26e7e73ed" } \ No newline at end of file diff --git a/sdk/resources/azure-mgmt-msi/azure/__init__.py b/sdk/resources/azure-mgmt-msi/azure/__init__.py index 8db66d3d0f0f..d55ccad1f573 100644 --- a/sdk/resources/azure-mgmt-msi/azure/__init__.py +++ b/sdk/resources/azure-mgmt-msi/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/resources/azure-mgmt-msi/azure/mgmt/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/__init__.py index 8db66d3d0f0f..d55ccad1f573 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/__init__.py +++ b/sdk/resources/azure-mgmt-msi/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/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py index 6bbff1405f98..ca1ad10c23b4 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__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,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._managed_service_identity_client import ManagedServiceIdentityClient # type: ignore +from ._client import ManagedIdentityClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -25,7 +25,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "ManagedServiceIdentityClient", + "ManagedIdentityClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_managed_service_identity_client.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py similarity index 70% rename from sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_managed_service_identity_client.py rename to sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py index 0abaf7c7dce2..4acebf4f79b6 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_managed_service_identity_client.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py @@ -2,13 +2,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 copy import deepcopy +import sys from typing import Any, Optional, TYPE_CHECKING, cast -from typing_extensions import Self from azure.core.pipeline import policies from azure.core.rest import HttpRequest, HttpResponse @@ -17,8 +17,7 @@ from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from . import models as _models -from ._configuration import ManagedServiceIdentityClientConfiguration +from ._configuration import ManagedIdentityClientConfiguration from ._utils.serialization import Deserializer, Serializer from .operations import ( FederatedIdentityCredentialsOperations, @@ -27,44 +26,67 @@ UserAssignedIdentitiesOperations, ) +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self # type: ignore + if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials import TokenCredential -class ManagedServiceIdentityClient: +class ManagedIdentityClient: """The Managed Service Identity Client. + :ivar operations: Operations operations + :vartype operations: azure.mgmt.msi.operations.Operations :ivar system_assigned_identities: SystemAssignedIdentitiesOperations operations :vartype system_assigned_identities: azure.mgmt.msi.operations.SystemAssignedIdentitiesOperations - :ivar operations: Operations operations - :vartype operations: azure.mgmt.msi.operations.Operations - :ivar user_assigned_identities: UserAssignedIdentitiesOperations operations - :vartype user_assigned_identities: azure.mgmt.msi.operations.UserAssignedIdentitiesOperations :ivar federated_identity_credentials: FederatedIdentityCredentialsOperations operations :vartype federated_identity_credentials: azure.mgmt.msi.operations.FederatedIdentityCredentialsOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar user_assigned_identities: UserAssignedIdentitiesOperations operations + :vartype user_assigned_identities: azure.mgmt.msi.operations.UserAssignedIdentitiesOperations + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: The Id of the Subscription to which the identity belongs. 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 "2023-01-31". 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 + "2025-05-31-preview" and None. Default value is None. If not set, the operation's default API + version will be used. Note that overriding this default value may result in unsupported + behavior. :paramtype api_version: str """ 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 = ManagedServiceIdentityClientConfiguration( - credential=credential, subscription_id=subscription_id, credential_scopes=credential_scopes, **kwargs + self._config = ManagedIdentityClientConfiguration( + 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) @@ -85,30 +107,29 @@ 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.operations = Operations(self._client, self._config, self._serialize, self._deserialize) self.system_assigned_identities = SystemAssignedIdentitiesOperations( self._client, self._config, self._serialize, self._deserialize ) - self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.user_assigned_identities = UserAssignedIdentitiesOperations( + self.federated_identity_credentials = FederatedIdentityCredentialsOperations( self._client, self._config, self._serialize, self._deserialize ) - self.federated_identity_credentials = FederatedIdentityCredentialsOperations( + self.user_assigned_identities = UserAssignedIdentitiesOperations( 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 @@ -121,7 +142,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/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py index a2e7754d7c12..c9851865f4db 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py @@ -2,11 +2,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 @@ -14,26 +14,41 @@ from ._version import VERSION if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials import TokenCredential -class ManagedServiceIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long - """Configuration for ManagedServiceIdentityClient. +class ManagedIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for ManagedIdentityClient. 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 Subscription to which the identity belongs. 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 "2023-01-31". 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 + "2025-05-31-preview" and None. Default value is None. If not set, the operation's default API + version will be used. 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: - api_version: str = kwargs.pop("api_version", "2023-01-31") + 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", "2025-05-31-preview") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,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-msi/{}".format(VERSION)) diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_patch.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_patch.py index 17dbc073e01b..ea765788358a 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_patch.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_patch.py @@ -1,32 +1,20 @@ # coding=utf-8 # -------------------------------------------------------------------------- -# # Copyright (c) Microsoft Corporation. All rights reserved. -# -# The MIT License (MIT) -# -# Permission is hereby granted, free of charge, to any person obtaining a copy -# of this software and associated documentation files (the ""Software""), to -# deal in the Software without restriction, including without limitation the -# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or -# sell copies of the Software, and to permit persons to whom the Software is -# furnished to do so, subject to the following conditions: -# -# The above copyright notice and this permission notice shall be included in -# all copies or substantial portions of the Software. -# -# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR -# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, -# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE -# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER -# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING -# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS -# IN THE SOFTWARE. -# +# Licensed under the MIT License. See License.txt in the project root for license information. # -------------------------------------------------------------------------- +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" + +__all__: list[str] = [] # Add all objects you want publicly available to users at this package level -# This file is used for handwritten extensions to the generated code. Example: -# https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md def patch_sdk(): - pass + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/__init__.py index 0af9b28f6607..8026245c2abc 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_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/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/model_base.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/model_base.py new file mode 100644 index 000000000000..d725c55906d3 --- /dev/null +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/model_base.py @@ -0,0 +1,1737 @@ +# 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 +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 + +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self + +_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) + + +# ============================================================================ +# Fast-path scalar deserializer functions for rest_field(deserializer=...) +# These are referenced from rest_field declarations to bypass the generic +# _deserialize -> _deserialize_with_callable chain. +# Only simple/primitive types — no models or container types. +# ============================================================================ + + +def _xml_deser_str(value): + if isinstance(value, ET.Element): + return value.text or "" + return str(value) if value is not None else None + + +def _xml_deser_int(value): + if isinstance(value, ET.Element): + return int(value.text) if value.text else None + return int(value) if value is not None else None + + +def _xml_deser_float(value): + if isinstance(value, ET.Element): + return float(value.text) if value.text else None + return float(value) if value is not None else None + + +def _xml_deser_bool(value): + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + if text in (True, False): + return text + return text.lower() == "true" + + +# pylint: disable=docstring-missing-param +def _xml_deser_bytes(value): + """Deserialize bytes from XML (base64).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_bytes(text) + + +def _xml_deser_bytes_base64url(value): + """Deserialize bytes from XML (base64url).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_bytes_base64(text) + + +def _xml_deser_datetime(value): + """Deserialize a datetime from XML (ISO 8601 / rfc3339).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime(text) + + +def _xml_deser_datetime_rfc7231(value): + """Deserialize a datetime from XML (RFC7231 format).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime_rfc7231(text) + + +def _xml_deser_datetime_unix_timestamp(value): + """Deserialize a datetime from XML (Unix timestamp).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_datetime_unix_timestamp(float(text)) + + +def _xml_deser_date(value): + """Deserialize a date from XML (ISO 8601).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_date(text) + + +def _xml_deser_time(value): + """Deserialize a time from XML (ISO 8601).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_time(text) + + +def _xml_deser_duration(value): + """Deserialize a timedelta from XML (ISO 8601 duration).""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_duration(text) + + +def _xml_deser_decimal(value): + """Deserialize a Decimal from XML.""" + if isinstance(value, ET.Element): + text = value.text + else: + text = value + if text is None: + return None + return _deserialize_decimal(text) + + +def _xml_deser_enum_or_str(enum_cls, value): + """Deserialize a Union[EnumType, str] from XML.""" + text = value.text if isinstance(value, ET.Element) else value + if text is None: + return None + try: + return enum_cls(text) + except ValueError: + return text + + +def _extract_xml_model_type(rf_type): + """Extract the concrete Model class from a resolved rf._type partial chain. + + Unwraps ``Optional[Model]`` and ``_deserialize_model(Model, ...)`` + wrappers. Only handles Model and Optional[Model] — other composite + types (List, Dict, Union, etc.) return None and fall through to the + generic ``_deserialize`` path at runtime. + """ + if rf_type is None: + return None + if isinstance(rf_type, type) and _is_model(rf_type): + return rf_type + if not isinstance(rf_type, functools.partial): + return None + func = rf_type.func + args = rf_type.args + if func is _deserialize_with_optional and args: + return _extract_xml_model_type(args[0]) + if func is _deserialize_model and args: + cls = args[0] + return cls if isinstance(cls, type) and _is_model(cls) else None + return None + + +def _build_xml_field_plan( # pylint: disable=docstring-missing-return, docstring-missing-rtype, unused-variable + cls, attr_to_rest_field: dict +) -> list: + """Build a precomputed XML field plan for fast _init_from_xml iteration. + + Called once per model class in __new__. Returns a list of tuples: + (rest_name, xml_name, kind, deser, rf_type, is_optional, items_name) + + kind: 0=wrapped, 1=attribute, 2=unwrapped, 3=text + + For Model and Optional[Model] fields that lack a scalar + ``_deserializer``, this function precomputes the Model class as the + deserializer so ``_init_from_xml`` can call ``ModelClass(element)`` + directly instead of going through the expensive + ``_get_deserialize_callable_from_annotation`` chain at runtime. + """ + model_meta = getattr(cls, "_xml", {}) + model_ns = model_meta.get("ns") or model_meta.get("namespace") + plan = [] + + for rf in attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + deser = rf._deserializer + + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = _resolve_xml_ns(prop_meta, model_meta) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + is_optional = rf._is_optional + + # For Model / Optional[Model] fields without a scalar deserializer, + # precompute the Model class as the deserializer. + if deser is None and rf._type is not None: + model_cls = _extract_xml_model_type(rf._type) + if model_cls is not None: + deser = model_cls + + if prop_meta.get("attribute", False): + plan.append((rf._rest_name, xml_name, 1, deser, rf._type, is_optional, None)) + elif prop_meta.get("unwrapped", False): + items_name = prop_meta.get("itemsName") + if items_name: + items_ns = prop_meta.get("itemsNs") + if items_ns is not None: + xml_ns = items_ns + if xml_ns: + items_name = "{" + xml_ns + "}" + items_name + else: + items_name = xml_name + plan.append((rf._rest_name, xml_name, 2, deser, rf._type, is_optional, items_name)) + elif prop_meta.get("text", False): + plan.append((rf._rest_name, xml_name, 3, deser, rf._type, is_optional, None)) + else: + plan.append((rf._rest_name, xml_name, 0, deser, rf._type, is_optional, None)) + + return plan + + +# pylint: enable=docstring-missing-param +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: dict[str, typing.Any] = {} + if args: + if isinstance(args[0], ET.Element): + dict_to_pass.update(self._init_from_xml(args[0])) + 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 + } + ) + # Apply client default values for fields the caller didn't set so that + # defaults are part of `_data` and therefore included during serialization. + for rf in self._attr_to_rest_field.values(): + if rf._default is _UNSET: + continue + if rf._rest_name in dict_to_pass: + continue + dict_to_pass[rf._rest_name] = _create_value(rf, rf._default) + super().__init__(dict_to_pass) + + def _init_from_xml( # pylint: disable=too-many-branches, too-many-statements + self, element: ET.Element + ) -> dict[str, typing.Any]: + """Deserialize an XML element into a dict mapping rest field names to values. + + :param ET.Element element: The XML element to deserialize from. + :returns: A dictionary of rest_name to deserialized value pairs. + :rtype: dict + """ + result: dict[str, typing.Any] = {} + existed_attr_keys: list[str] = [] + + field_plan = getattr(self, "_xml_field_plan", None) + if field_plan: + for rest_name, xml_name, kind, deser, rf_type, is_optional, items_name in field_plan: + if kind == 0: # wrapped element (most common) + item = element.find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + if deser: + result[rest_name] = deser(item) + else: + result[rest_name] = _deserialize(rf_type, item) + elif kind == 1: # attribute + attr_val = element.get(xml_name) + if attr_val is not None: + existed_attr_keys.append(xml_name) + if deser: + result[rest_name] = deser(attr_val) + else: + result[rest_name] = attr_val + elif kind == 2: # unwrapped array + items = element.findall(items_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(items_name) + if deser: + result[rest_name] = deser(items) + else: + result[rest_name] = _deserialize(rf_type, items) + elif not is_optional: + existed_attr_keys.append(items_name) + result[rest_name] = [] + elif kind == 3: # text + if element.text is not None: + if deser: + result[rest_name] = deser(element.text) + else: + result[rest_name] = element.text + else: + 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 = _resolve_xml_ns(prop_meta, model_meta) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and element.get(xml_name) is not None: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, element.get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + _items_name = prop_meta.get("itemsName") + if _items_name: + xml_name = _items_name + _items_ns = prop_meta.get("itemsNs") + if _items_ns is not None: + xml_ns = _items_ns + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = element.findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, items) + elif not rf._is_optional: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = [] + continue + + # text element is primitive type + if prop_meta.get("text", False): + if element.text is not None: + result[rf._rest_name] = _deserialize(rf._type, element.text) + continue + + # wrapped element could be normal property or array + item = element.find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + result[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in element: + if e.tag not in existed_attr_keys: + result[e.tag] = _convert_element(e) + + return result + + 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()) + # Build XML field plan for fast _init_from_xml (only for XML models) + if getattr(cls, "_xml", None): + cls._xml_field_plan = _build_xml_field_plan(cls, attr_to_rest_field) + 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 = _resolve_xml_ns(prop_meta, model_meta) + 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, + deserializer: typing.Optional[typing.Callable] = 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 {} + self._deserializer = deserializer + + @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, _UNSET) + if item is _UNSET: + # Field not set by user; return the client default if one exists, otherwise None + return self._default if self._default is not _UNSET else None + 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) + + # Fast path: use _deserializer directly (avoids _serialize/_deserialize chain) + if self._deserializer: + deserialized = self._deserializer(item) + else: + 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, + deserializer: typing.Optional[typing.Callable] = 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, + deserializer=deserializer, + ) + + +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_xml_ns(meta: dict[str, typing.Any]) -> typing.Optional[str]: + """Return the XML namespace from a metadata dict, checking both 'ns' (old-style) and 'namespace' (DPG) keys. + + :param dict meta: The metadata dictionary to extract namespace from. + :returns: The namespace string if 'ns' or 'namespace' key is present, None otherwise. + :rtype: str or None + """ + ns = meta.get("ns") + if ns is None: + ns = meta.get("namespace") + return ns + + +def _resolve_xml_ns( + prop_meta: dict[str, typing.Any], model_meta: typing.Optional[dict[str, typing.Any]] = None +) -> typing.Optional[str]: + """Resolve XML namespace for a property, falling back to model namespace when appropriate. + + Checks the property metadata first; if no namespace is found and the model does not declare + an explicit prefix, falls back to the model-level namespace. + + :param dict prop_meta: The property metadata dictionary. + :param dict model_meta: The model metadata dictionary, used as fallback. + :returns: The resolved namespace string, or None. + :rtype: str or None + """ + ns = _get_xml_ns(prop_meta) + if ns is None and model_meta is not None and not model_meta.get("prefix"): + ns = _get_xml_ns(model_meta) + return ns + + +def _set_xml_attribute(element: ET.Element, name: str, value: typing.Any, prop_meta: dict[str, typing.Any]) -> None: + """Set an XML attribute on an element, handling namespace prefix registration. + + :param ET.Element element: The element to set the attribute on. + :param str name: The default attribute name (wire name). + :param any value: The attribute value. + :param dict prop_meta: The property metadata dictionary. + """ + xml_name = prop_meta.get("name", name) + _attr_ns = _get_xml_ns(prop_meta) + if _attr_ns: + _attr_prefix = prop_meta.get("prefix") + if _attr_prefix: + _safe_register_namespace(_attr_prefix, _attr_ns) + xml_name = "{" + _attr_ns + "}" + xml_name + element.set(xml_name, _get_primitive_type_value(value)) + + +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: + # When serializing as an array item (parent_meta is set), check if the parent has an + # explicit itemsName. This ensures correct element names for unwrapped arrays (where + # the element tag is the property/items name, not the model type name). + _items_name = parent_meta.get("itemsName") if parent_meta is not None else None + element_name = _items_name if _items_name else (model_meta.get("name") or o.__class__.__name__) + _model_ns = _get_xml_ns(model_meta) + wrapped_element = _create_xml_element( + element_name, + model_meta.get("prefix"), + _model_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} + + # Propagate model namespace to properties only for old-style "ns"-keyed models. + # DPG-generated models use the "namespace" key and explicitly declare namespace on + # each property that needs it, so propagation is intentionally skipped for them. + 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): + _set_xml_attribute(wrapped_element, k, v, prop_meta) + 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 = [] + _dict_ns = _get_xml_ns(parent_meta) if parent_meta else None + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": _dict_ns, + "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: + _items_ns = parent_meta.get("itemsNs") + if _items_ns is None: + _items_ns = _get_xml_ns(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": _items_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: + _meta_ns = _get_xml_ns(meta) if meta else None + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, _meta_ns + ) + 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 _safe_register_namespace(prefix: str, ns: str) -> None: + """Register an XML namespace prefix, handling reserved prefix patterns. + + Some prefixes (e.g. 'ns2') match Python's reserved 'ns\\d+' pattern used for + auto-generated prefixes, causing register_namespace to raise ValueError. + Falls back to directly registering in the internal namespace map. + + :param str prefix: The namespace prefix to register. + :param str ns: The namespace URI. + """ + try: + ET.register_namespace(prefix, ns) + except ValueError: + _ns_map = getattr(ET, "_namespace_map", None) + if _ns_map is not None: + _ns_map[ns] = prefix + + +def _create_xml_element( + tag: typing.Any, prefix: typing.Optional[str] = None, ns: typing.Optional[str] = None +) -> ET.Element: + if prefix and ns: + _safe_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 + if _is_model(deserializer): + return deserializer._deserialize(element, []) + 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/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/serialization.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/serialization.py index f5187701d7be..a088671e9c51 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_utils/serialization.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_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: @@ -41,11 +39,15 @@ import xml.etree.ElementTree as ET import isodate # type: ignore -from typing_extensions import Self from azure.core.exceptions import DeserializationError, SerializationError from azure.core.serialization import NULL as CoreNull +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self + _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") JSON = MutableMapping[str, Any] @@ -229,12 +231,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 +313,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 +382,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 +416,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 +530,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 +581,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 +791,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 +825,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 +1193,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 +1395,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 @@ -1396,7 +1405,7 @@ def __init__(self, classes: Optional[Mapping[str, type]] = None) -> None: # Otherwise, result are unexpected self.additional_properties_detection = True - def __call__(self, target_obj, response_data, content_type=None): + def __call__(self, target_obj, response_data, content_type=None): # pylint: disable=too-many-return-statements """Call the deserializer to process a REST response. :param str target_obj: Target data type to deserialize to. @@ -1406,6 +1415,27 @@ def __call__(self, target_obj, response_data, content_type=None): :return: Deserialized object. :rtype: object """ + # Fast path for header deserialization: response_data is a plain str or None + # and target_obj is a simple scalar type. This avoids the expensive + # _unpack_content → _deserialize → _classify_target → deserialize_data chain. + if response_data is None: + return None + if target_obj == "str" and isinstance(response_data, str): + return response_data + if isinstance(response_data, str): + if target_obj == "int": + return int(response_data) + if target_obj == "bool": + if response_data in ("true", "1", "True"): + return True + if response_data in ("false", "0", "False"): + return False + return bool(response_data) + if target_obj == "rfc-1123": + return Deserializer.deserialize_rfc(response_data) + if target_obj == "bytearray": + return Deserializer.deserialize_bytearray(response_data) + data = self._unpack_content(response_data, content_type) return self._deserialize(target_obj, data) @@ -1759,7 +1789,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 +1815,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/resources/azure-mgmt-msi/azure/mgmt/msi/_version.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_version.py index 0d5779dc14de..35a1c7bec106 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_version.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_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 = "8.0.0b1" +VERSION = "8.0.0b2" diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py index 14c69894e88d..afab46398e53 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__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,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._managed_service_identity_client import ManagedServiceIdentityClient # type: ignore +from ._client import ManagedIdentityClient # type: ignore try: from ._patch import __all__ as _patch_all @@ -22,7 +22,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "ManagedServiceIdentityClient", + "ManagedIdentityClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_managed_service_identity_client.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py similarity index 71% rename from sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_managed_service_identity_client.py rename to sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py index 3e3ff16c99be..b35a84f661cc 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_managed_service_identity_client.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py @@ -2,13 +2,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 copy import deepcopy +import sys from typing import Any, Awaitable, Optional, TYPE_CHECKING, cast -from typing_extensions import Self from azure.core.pipeline import policies from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -17,9 +17,8 @@ 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 ManagedServiceIdentityClientConfiguration +from ._configuration import ManagedIdentityClientConfiguration from .operations import ( FederatedIdentityCredentialsOperations, Operations, @@ -27,45 +26,68 @@ UserAssignedIdentitiesOperations, ) +if sys.version_info >= (3, 11): + from typing import Self +else: + from typing_extensions import Self # type: ignore + if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials_async import AsyncTokenCredential -class ManagedServiceIdentityClient: +class ManagedIdentityClient: """The Managed Service Identity Client. + :ivar operations: Operations operations + :vartype operations: azure.mgmt.msi.aio.operations.Operations :ivar system_assigned_identities: SystemAssignedIdentitiesOperations operations :vartype system_assigned_identities: azure.mgmt.msi.aio.operations.SystemAssignedIdentitiesOperations - :ivar operations: Operations operations - :vartype operations: azure.mgmt.msi.aio.operations.Operations - :ivar user_assigned_identities: UserAssignedIdentitiesOperations operations - :vartype user_assigned_identities: - azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations :ivar federated_identity_credentials: FederatedIdentityCredentialsOperations operations :vartype federated_identity_credentials: azure.mgmt.msi.aio.operations.FederatedIdentityCredentialsOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar user_assigned_identities: UserAssignedIdentitiesOperations operations + :vartype user_assigned_identities: + azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations + :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 Subscription to which the identity belongs. 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 "2023-01-31". 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 + "2025-05-31-preview" and None. Default value is None. If not set, the operation's default API + version will be used. Note that overriding this default value may result in unsupported + behavior. :paramtype api_version: str """ 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 = ManagedServiceIdentityClientConfiguration( - credential=credential, subscription_id=subscription_id, credential_scopes=credential_scopes, **kwargs + self._config = ManagedIdentityClientConfiguration( + 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) @@ -87,25 +109,24 @@ 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.operations = Operations(self._client, self._config, self._serialize, self._deserialize) self.system_assigned_identities = SystemAssignedIdentitiesOperations( self._client, self._config, self._serialize, self._deserialize ) - self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) - self.user_assigned_identities = UserAssignedIdentitiesOperations( + self.federated_identity_credentials = FederatedIdentityCredentialsOperations( self._client, self._config, self._serialize, self._deserialize ) - self.federated_identity_credentials = FederatedIdentityCredentialsOperations( + self.user_assigned_identities = UserAssignedIdentitiesOperations( 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. @@ -113,7 +134,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 @@ -126,7 +147,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/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py index bde74845cc58..846202bf526f 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py @@ -2,11 +2,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 @@ -14,26 +14,41 @@ from .._version import VERSION if TYPE_CHECKING: + from azure.core import AzureClouds from azure.core.credentials_async import AsyncTokenCredential -class ManagedServiceIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long - """Configuration for ManagedServiceIdentityClient. +class ManagedIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for ManagedIdentityClient. 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 Subscription to which the identity belongs. 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 "2023-01-31". 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 + "2025-05-31-preview" and None. Default value is None. If not set, the operation's default API + version will be used. 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: - api_version: str = kwargs.pop("api_version", "2023-01-31") + 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", "2025-05-31-preview") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,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-msi/{}".format(VERSION)) diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_patch.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_patch.py index 8bcb627aa475..ea765788358a 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_patch.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_patch.py @@ -7,9 +7,8 @@ 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/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/__init__.py index b50b543ebd92..f8dc23d82292 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/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,20 +12,20 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._system_assigned_identities_operations import SystemAssignedIdentitiesOperations # type: ignore from ._operations import Operations # type: ignore -from ._user_assigned_identities_operations import UserAssignedIdentitiesOperations # type: ignore -from ._federated_identity_credentials_operations import FederatedIdentityCredentialsOperations # type: ignore +from ._operations import SystemAssignedIdentitiesOperations # type: ignore +from ._operations import FederatedIdentityCredentialsOperations # type: ignore +from ._operations import UserAssignedIdentitiesOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ - "SystemAssignedIdentitiesOperations", "Operations", - "UserAssignedIdentitiesOperations", + "SystemAssignedIdentitiesOperations", "FederatedIdentityCredentialsOperations", + "UserAssignedIdentitiesOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_federated_identity_credentials_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_federated_identity_credentials_operations.py deleted file mode 100644 index 31b13390ac1b..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_federated_identity_credentials_operations.py +++ /dev/null @@ -1,431 +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 io import IOBase -from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import AsyncPipelineClient -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -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._federated_identity_credentials_operations import ( - build_create_or_update_request, - build_delete_request, - build_get_request, - build_list_request, -) -from .._configuration import ManagedServiceIdentityClientConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class FederatedIdentityCredentialsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s - :attr:`federated_identity_credentials` 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: ManagedServiceIdentityClientConfiguration = ( - 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 list( - self, - resource_group_name: str, - resource_name: str, - top: Optional[int] = None, - skiptoken: Optional[str] = None, - **kwargs: Any - ) -> AsyncItemPaged["_models.FederatedIdentityCredential"]: - """Lists all the federated identity credentials under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param top: Number of records to return. Default value is None. - :type top: int - :param skiptoken: A skip token is used to continue retrieving items after an operation returns - a partial result. If a previous response contains a nextLink element, the value of the nextLink - element will include a skipToken parameter that specifies a starting point to use for - subsequent calls. Default value is None. - :type skiptoken: str - :return: An iterator like instance of either FederatedIdentityCredential or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.FederatedIdentityCredential] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.FederatedIdentityCredentialsListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - subscription_id=self._config.subscription_id, - top=top, - skiptoken=skiptoken, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("FederatedIdentityCredentialsListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @overload - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: _models.FederatedIdentityCredential, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Required. - :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Required. - :type parameters: 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: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: Union[_models.FederatedIdentityCredential, IO[bytes]], - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Is either - a FederatedIdentityCredential type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential or IO[bytes] - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :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 = 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[_models.FederatedIdentityCredential] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "FederatedIdentityCredential") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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) - - _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, 201]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("FederatedIdentityCredential", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def get( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Gets the federated identity credential. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :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.FederatedIdentityCredential] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("FederatedIdentityCredential", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - **kwargs: Any - ) -> None: - """Deletes the federated identity credential. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :return: None or the result of cls(response) - :rtype: None - :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[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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, 204]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py index 7011fba63c20..e349e5018170 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py @@ -1,12 +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 typing import Any, Callable, Dict, Optional, TypeVar +from io import IOBase +import json +from typing import Any, Callable, IO, Optional, TypeVar, Union, overload import urllib.parse from azure.core import AsyncPipelineClient @@ -17,21 +20,40 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace +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.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ..._utils.serialization import Deserializer, Serializer -from ...operations._operations import build_list_request -from .._configuration import ManagedServiceIdentityClientConfiguration +from ...operations._operations import ( + build_federated_identity_credentials_create_or_update_request, + build_federated_identity_credentials_delete_request, + build_federated_identity_credentials_get_request, + build_federated_identity_credentials_list_request, + build_operations_list_request, + build_system_assigned_identities_get_by_scope_request, + build_user_assigned_identities_create_or_update_request, + build_user_assigned_identities_delete_request, + build_user_assigned_identities_get_request, + build_user_assigned_identities_list_by_resource_group_request, + build_user_assigned_identities_list_by_subscription_request, + build_user_assigned_identities_update_request, +) +from .._configuration import ManagedIdentityClientConfiguration 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]] +JSON = MutableMapping[str, Any] +List = list class Operations: @@ -40,34 +62,29 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s + :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s :attr:`operations` 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: ManagedServiceIdentityClientConfiguration = ( - input_args.pop(0) if input_args else kwargs.pop("config") - ) + self._config: ManagedIdentityClientConfiguration = 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 list(self, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: - """Lists available operations for the Microsoft.ManagedIdentity provider. + """List the operations for the provider. - :return: An iterator like instance of either Operation or the result of cls(response) + :return: An iterator like instance of Operation :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.Operation] :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.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -80,12 +97,17 @@ def list(self, **kwargs: Any) -> AsyncItemPaged["_models.Operation"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_operations_list_request( + 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 @@ -98,18 +120,29 @@ def prepare_request(next_link=None): ) _next_request_params["api-version"] = self._config.api_version _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + 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("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Operation], + 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) @@ -122,7 +155,1244 @@ 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class SystemAssignedIdentitiesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :attr:`system_assigned_identities` 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: ManagedIdentityClientConfiguration = 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_by_scope(self, scope: str, **kwargs: Any) -> _models.SystemAssignedIdentity: + """Gets the systemAssignedIdentity available under the specified RP scope. + + :param scope: The fully qualified Azure Resource manager identifier of the resource. Required. + :type scope: str + :return: SystemAssignedIdentity. The SystemAssignedIdentity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.SystemAssignedIdentity + :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.SystemAssignedIdentity] = kwargs.pop("cls", None) + + _request = build_system_assigned_identities_get_by_scope_request( + scope=scope, + 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.CloudError, + 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.SystemAssignedIdentity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class FederatedIdentityCredentialsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :attr:`federated_identity_credentials` 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: ManagedIdentityClientConfiguration = 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, + resource_name: str, + federated_identity_credential_resource_name: str, + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Gets the federated identity credential. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :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.FederatedIdentityCredential] = kwargs.pop("cls", None) + + _request = build_federated_identity_credentials_get_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_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.CloudError, + 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.FederatedIdentityCredential, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: _models.FederatedIdentityCredential, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: 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: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: Union[_models.FederatedIdentityCredential, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Is one of + the following types: FederatedIdentityCredential, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential or JSON or IO[bytes] + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.FederatedIdentityCredential] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_federated_identity_credentials_create_or_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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, 201]: + 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.CloudError, + 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.FederatedIdentityCredential, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + **kwargs: Any + ) -> None: + """Deletes the federated identity credential. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :return: None + :rtype: None + :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[None] = kwargs.pop("cls", None) + + _request = build_federated_identity_credentials_delete_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list( + self, + resource_group_name: str, + resource_name: str, + *, + top: Optional[int] = None, + skiptoken: Optional[str] = None, + **kwargs: Any + ) -> AsyncItemPaged["_models.FederatedIdentityCredential"]: + """Lists all the federated identity credentials under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :keyword top: Number of records to return. Default value is None. + :paramtype top: int + :keyword skiptoken: A skip token is used to continue retrieving items after an operation + returns a partial result. If a previous response contains a nextLink element, the value of the + nextLink element will include a skipToken parameter that specifies a starting point to use for + subsequent calls. Default value is None. + :paramtype skiptoken: str + :return: An iterator like instance of FederatedIdentityCredential + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.FederatedIdentityCredential] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.FederatedIdentityCredential]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_federated_identity_credentials_list_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + top=top, + skiptoken=skiptoken, + 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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.FederatedIdentityCredential], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class UserAssignedIdentitiesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :attr:`user_assigned_identities` 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: ManagedIdentityClientConfiguration = 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, resource_name: str, **kwargs: Any) -> _models.Identity: + """Gets the identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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.Identity] = kwargs.pop("cls", None) + + _request = build_user_assigned_identities_get_request( + resource_group_name=resource_group_name, + resource_name=resource_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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: _models.Identity, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: ~azure.mgmt.msi.models.Identity + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: 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: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: Union[_models.Identity, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Is one of the following types: + Identity, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.Identity or JSON or IO[bytes] + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Identity] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_user_assigned_identities_create_or_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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, 201]: + 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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + async def update( + self, + resource_group_name: str, + resource_name: str, + parameters: _models.IdentityUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: ~azure.mgmt.msi.models.IdentityUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: 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: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + resource_name: str, + parameters: Union[_models.IdentityUpdate, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Is one of the following types: + IdentityUpdate, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.IdentityUpdate or JSON or IO[bytes] + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Identity] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_user_assigned_identities_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def delete(self, resource_group_name: str, resource_name: str, **kwargs: Any) -> None: + """Deletes the identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :return: None + :rtype: None + :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[None] = kwargs.pop("cls", None) + + _request = build_user_assigned_identities_delete_request( + resource_group_name=resource_group_name, + resource_name=resource_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Identity"]: + """Lists all the userAssignedIdentities available under the specified ResourceGroup. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of Identity + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.Identity] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Identity]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_user_assigned_identities_list_by_resource_group_request( + resource_group_name=resource_group_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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Identity], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.Identity"]: + """Lists all the userAssignedIdentities available under the specified subscription. + + :return: An iterator like instance of Identity + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.Identity] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Identity]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_user_assigned_identities_list_by_subscription_request( + 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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + async def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Identity], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, AsyncList(list_of_elem) + + async def get_next(next_link=None): + _request = prepare_request(next_link) + + _stream = False + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + response = pipeline_response.http_response + + if response.status_code not in [200]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_patch.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_patch.py index 8bcb627aa475..ea765788358a 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_patch.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_patch.py @@ -7,9 +7,8 @@ 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/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_system_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_system_assigned_identities_operations.py deleted file mode 100644 index d27e6e98b079..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_system_assigned_identities_operations.py +++ /dev/null @@ -1,105 +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._system_assigned_identities_operations import build_get_by_scope_request -from .._configuration import ManagedServiceIdentityClientConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class SystemAssignedIdentitiesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s - :attr:`system_assigned_identities` 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: ManagedServiceIdentityClientConfiguration = ( - 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_by_scope(self, scope: str, **kwargs: Any) -> _models.SystemAssignedIdentity: - """Gets the systemAssignedIdentity available under the specified RP scope. - - :param scope: The resource provider scope of the resource. Parent resource being extended by - Managed Identities. Required. - :type scope: str - :return: SystemAssignedIdentity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.SystemAssignedIdentity - :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.SystemAssignedIdentity] = kwargs.pop("cls", None) - - _request = build_get_by_scope_request( - scope=scope, - 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("SystemAssignedIdentity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_user_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_user_assigned_identities_operations.py deleted file mode 100644 index a394dbabdb4e..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_user_assigned_identities_operations.py +++ /dev/null @@ -1,583 +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 io import IOBase -from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import AsyncPipelineClient -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -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._user_assigned_identities_operations import ( - build_create_or_update_request, - build_delete_request, - build_get_request, - build_list_by_resource_group_request, - build_list_by_subscription_request, - build_update_request, -) -from .._configuration import ManagedServiceIdentityClientConfiguration - -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class UserAssignedIdentitiesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s - :attr:`user_assigned_identities` 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: ManagedServiceIdentityClientConfiguration = ( - 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 list_by_subscription(self, **kwargs: Any) -> AsyncItemPaged["_models.Identity"]: - """Lists all the userAssignedIdentities available under the specified subscription. - - :return: An iterator like instance of either Identity or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.Identity] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UserAssignedIdentitiesListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_subscription_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("UserAssignedIdentitiesListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncItemPaged["_models.Identity"]: - """Lists all the userAssignedIdentities available under the specified ResourceGroup. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :return: An iterator like instance of either Identity or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.msi.models.Identity] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UserAssignedIdentitiesListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - async def extract_data(pipeline_response): - deserialized = self._deserialize("UserAssignedIdentitiesListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @overload - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: _models.Identity, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Required. - :type parameters: ~azure.mgmt.msi.models.Identity - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Required. - :type parameters: 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: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: Union[_models.Identity, IO[bytes]], - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Is either a Identity type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.Identity or IO[bytes] - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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 = 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[_models.Identity] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "Identity") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - resource_name=resource_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) - - _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, 201]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - async def update( - self, - resource_group_name: str, - resource_name: str, - parameters: _models.IdentityUpdate, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Required. - :type parameters: ~azure.mgmt.msi.models.IdentityUpdate - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def update( - self, - resource_group_name: str, - resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Required. - :type parameters: 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: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def update( - self, - resource_group_name: str, - resource_name: str, - parameters: Union[_models.IdentityUpdate, IO[bytes]], - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Is either a IdentityUpdate type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.IdentityUpdate or IO[bytes] - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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 = 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[_models.Identity] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "IdentityUpdate") - - _request = build_update_request( - resource_group_name=resource_group_name, - resource_name=resource_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) - - _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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def get(self, resource_group_name: str, resource_name: str, **kwargs: Any) -> _models.Identity: - """Gets the identity. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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.Identity] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - resource_name=resource_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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def delete(self, resource_group_name: str, resource_name: str, **kwargs: Any) -> None: - """Deletes the identity. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :return: None or the result of cls(response) - :rtype: None - :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[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - resource_name=resource_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, 204]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/__init__.py index 24c4535d9b18..5de45163d93b 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/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,46 +13,56 @@ from ._patch import * # pylint: disable=unused-wildcard-import -from ._models_py3 import ( # type: ignore +from ._models import ( # type: ignore + AssignmentRestrictions, + ClaimsMatchingExpression, + CloudError, CloudErrorBody, + ExtensionResource, FederatedIdentityCredential, - FederatedIdentityCredentialsListResult, + FederatedIdentityCredentialProperties, Identity, IdentityUpdate, Operation, OperationDisplay, - OperationListResult, ProxyResource, Resource, SystemAssignedIdentity, + SystemAssignedIdentityProperties, SystemData, TrackedResource, - UserAssignedIdentitiesListResult, + UserAssignedIdentityProperties, ) -from ._managed_service_identity_client_enums import ( # type: ignore +from ._enums import ( # type: ignore CreatedByType, + IsolationScope, ) from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ + "AssignmentRestrictions", + "ClaimsMatchingExpression", + "CloudError", "CloudErrorBody", + "ExtensionResource", "FederatedIdentityCredential", - "FederatedIdentityCredentialsListResult", + "FederatedIdentityCredentialProperties", "Identity", "IdentityUpdate", "Operation", "OperationDisplay", - "OperationListResult", "ProxyResource", "Resource", "SystemAssignedIdentity", + "SystemAssignedIdentityProperties", "SystemData", "TrackedResource", - "UserAssignedIdentitiesListResult", + "UserAssignedIdentityProperties", "CreatedByType", + "IsolationScope", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_managed_service_identity_client_enums.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_enums.py similarity index 54% rename from sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_managed_service_identity_client_enums.py rename to sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_enums.py index b3438c1a4334..15b2d9464f3f 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_managed_service_identity_client_enums.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/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. # -------------------------------------------------------------------------- @@ -11,9 +11,22 @@ 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 IsolationScope(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Enum to configure regional restrictions on identity assignment, as necessary.""" + + NONE = "None" + """NONE.""" + REGIONAL = "Regional" + """REGIONAL.""" diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models.py new file mode 100644 index 000000000000..a0079bbe81d9 --- /dev/null +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models.py @@ -0,0 +1,767 @@ +# 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 AssignmentRestrictions(_Model): + """Configuration to restrict identity assignment to specific resource providers or resource types. + + :ivar providers: List of resource providers or resource providers with resource types that this + identity can be assigned to (case-insensitive). Examples: 'Microsoft.Compute', + 'Microsoft.Storage/Accounts', 'Microsoft.Network/VirtualNetworks'. + :vartype providers: list[str] + """ + + providers: Optional[list[str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """List of resource providers or resource providers with resource types that this identity can be + assigned to (case-insensitive). Examples: 'Microsoft.Compute', 'Microsoft.Storage/Accounts', + 'Microsoft.Network/VirtualNetworks'.""" + + @overload + def __init__( + self, + *, + providers: Optional[list[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 ClaimsMatchingExpression(_Model): + """Object for defining the allowed identifiers of external identities. Introduced in + 2025-01-31-preview. + + :ivar value: Wildcard-based expression for matching incoming subject claims. Required. + :vartype value: str + :ivar language_version: Specifies the version of the flexible fic language used in the + expression. Required. + :vartype language_version: int + """ + + value: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Wildcard-based expression for matching incoming subject claims. Required.""" + language_version: int = rest_field( + name="languageVersion", visibility=["read", "create", "update", "delete", "query"] + ) + """Specifies the version of the flexible fic language used in the expression. Required.""" + + @overload + def __init__( + self, + *, + value: str, + language_version: int, + ) -> 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 CloudError(_Model): + """An error response from the ManagedServiceIdentity service. + + :ivar error: A list of additional details about the error. + :vartype error: ~azure.mgmt.msi.models.CloudErrorBody + """ + + error: Optional["_models.CloudErrorBody"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """A list of additional details about the error.""" + + @overload + def __init__( + self, + *, + error: Optional["_models.CloudErrorBody"] = 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 CloudErrorBody(_Model): + """An error response from the ManagedServiceIdentity service. + + :ivar code: An identifier for the error. + :vartype code: str + :ivar message: A message describing the error, intended to be suitable for display in a user + interface. + :vartype message: str + :ivar target: The target of the particular error. For example, the name of the property in + error. + :vartype target: str + :ivar details: A list of additional details about the error. + :vartype details: list[~azure.mgmt.msi.models.CloudErrorBody] + """ + + code: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """An identifier for the error.""" + message: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """A message describing the error, intended to be suitable for display in a user interface.""" + target: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The target of the particular error. For example, the name of the property in error.""" + details: Optional[list["_models.CloudErrorBody"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """A list of additional details about the error.""" + + @overload + def __init__( + self, + *, + code: Optional[str] = None, + message: Optional[str] = None, + target: Optional[str] = None, + details: Optional[list["_models.CloudErrorBody"]] = 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.msi.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 ExtensionResource(Resource): + """The base extension 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.msi.models.SystemData + """ + + +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.msi.models.SystemData + """ + + +class FederatedIdentityCredential(ProxyResource): + """Describes a federated identity credential. + + :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.msi.models.SystemData + :ivar properties: The properties associated with the federated identity credential. + :vartype properties: ~azure.mgmt.msi.models.FederatedIdentityCredentialProperties + """ + + properties: Optional["_models.FederatedIdentityCredentialProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The properties associated with the federated identity credential.""" + + __flattened_items = ["issuer", "subject", "audiences", "claims_matching_expression"] + + @overload + def __init__( + self, + *, + properties: Optional["_models.FederatedIdentityCredentialProperties"] = 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 FederatedIdentityCredentialProperties(_Model): + """Federated identity credential properties. + + :ivar issuer: The URL of the issuer to be trusted. Required. + :vartype issuer: str + :ivar subject: The identifier of the external identity. + :vartype subject: str + :ivar audiences: The list of audiences that can appear in the issued token. Required. + :vartype audiences: list[str] + :ivar claims_matching_expression: Object for defining the allowed identifiers of external + identities. Either 'subject' or 'claimsMatchingExpression' must be defined, but not both. + Introduced in 2025-01-31-preview. + :vartype claims_matching_expression: ~azure.mgmt.msi.models.ClaimsMatchingExpression + """ + + issuer: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The URL of the issuer to be trusted. Required.""" + subject: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The identifier of the external identity.""" + audiences: list[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The list of audiences that can appear in the issued token. Required.""" + claims_matching_expression: Optional["_models.ClaimsMatchingExpression"] = rest_field( + name="claimsMatchingExpression", visibility=["read", "create", "update", "delete", "query"] + ) + """Object for defining the allowed identifiers of external identities. Either 'subject' or + 'claimsMatchingExpression' must be defined, but not both. Introduced in 2025-01-31-preview.""" + + @overload + def __init__( + self, + *, + issuer: str, + audiences: list[str], + subject: Optional[str] = None, + claims_matching_expression: Optional["_models.ClaimsMatchingExpression"] = 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 TrackedResource(Resource): + """Tracked 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.msi.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + """ + + tags: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource tags.""" + location: str = rest_field(visibility=["read", "create"]) + """The geo-location where the resource lives. Required.""" + + @overload + def __init__( + self, + *, + location: str, + tags: 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 Identity(TrackedResource): + """Describes an identity 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.msi.models.SystemData + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar location: The geo-location where the resource lives. Required. + :vartype location: str + :ivar properties: The properties associated with the identity. + :vartype properties: ~azure.mgmt.msi.models.UserAssignedIdentityProperties + """ + + properties: Optional["_models.UserAssignedIdentityProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The properties associated with the identity.""" + + @overload + def __init__( + self, + *, + location: str, + tags: Optional[dict[str, str]] = None, + properties: Optional["_models.UserAssignedIdentityProperties"] = 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 IdentityUpdate(Resource): + """Describes an identity 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.msi.models.SystemData + :ivar location: The geo-location where the resource lives. + :vartype location: str + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar properties: The properties associated with the identity. + :vartype properties: ~azure.mgmt.msi.models.UserAssignedIdentityProperties + """ + + location: Optional[str] = rest_field(visibility=["read", "create"]) + """The geo-location where the resource lives.""" + tags: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update"]) + """Resource tags.""" + properties: Optional["_models.UserAssignedIdentityProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The properties associated with the identity.""" + + __flattened_items = ["tenant_id", "principal_id", "client_id", "isolation_scope", "assignment_restrictions"] + + @overload + def __init__( + self, + *, + location: Optional[str] = None, + tags: Optional[dict[str, str]] = None, + properties: Optional["_models.UserAssignedIdentityProperties"] = 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 Operation(_Model): + """Microsoft.ManagedIdentity Operation. + + :ivar name: Operation Name. + :vartype name: str + :ivar display: Operation Display. + :vartype display: ~azure.mgmt.msi.models.OperationDisplay + """ + + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Operation Name.""" + display: Optional["_models.OperationDisplay"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Operation Display.""" + + @overload + def __init__( + self, + *, + name: Optional[str] = None, + display: Optional["_models.OperationDisplay"] = 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 OperationDisplay(_Model): + """Operation Display. + + :ivar provider: Resource Provider Name. + :vartype provider: str + :ivar operation: Operation Type. + :vartype operation: str + :ivar resource: Resource Type. + :vartype resource: str + :ivar description: Operation description. + :vartype description: str + """ + + provider: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource Provider Name.""" + operation: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Operation Type.""" + resource: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Resource Type.""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Operation description.""" + + @overload + def __init__( + self, + *, + provider: Optional[str] = None, + operation: Optional[str] = None, + resource: Optional[str] = None, + description: 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 SystemAssignedIdentity(ExtensionResource): + """Describes a system assigned identity 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.msi.models.SystemData + :ivar properties: The properties associated with the identity. + :vartype properties: ~azure.mgmt.msi.models.SystemAssignedIdentityProperties + :ivar location: Required. + :vartype location: str + :ivar tags: + :vartype tags: dict[str, str] + """ + + properties: Optional["_models.SystemAssignedIdentityProperties"] = rest_field(visibility=["read"]) + """The properties associated with the identity.""" + location: str = rest_field(visibility=["read", "create"]) + """Required.""" + tags: Optional[dict[str, str]] = rest_field(visibility=["read", "create", "update"]) + + @overload + def __init__( + self, + *, + location: str, + tags: 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 SystemAssignedIdentityProperties(_Model): + """System Assigned Identity properties. + + :ivar tenant_id: The id of the tenant which the identity belongs to. + :vartype tenant_id: str + :ivar principal_id: The id of the service principal object associated with the created + identity. + :vartype principal_id: str + :ivar client_id: The id of the app associated with the identity. This is a random generated + UUID by MSI. + :vartype client_id: str + :ivar client_secret_url: The ManagedServiceIdentity DataPlane URL that can be queried to obtain + the identity credentials. + :vartype client_secret_url: str + """ + + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The id of the tenant which the identity belongs to.""" + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The id of the service principal object associated with the created identity.""" + client_id: Optional[str] = rest_field(name="clientId", visibility=["read"]) + """The id of the app associated with the identity. This is a random generated UUID by MSI.""" + client_secret_url: Optional[str] = rest_field(name="clientSecretUrl", visibility=["read"]) + """The ManagedServiceIdentity DataPlane URL that can be queried to obtain the identity + credentials.""" + + +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.msi.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.msi.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) + + +class UserAssignedIdentityProperties(_Model): + """User Assigned Identity properties. + + :ivar tenant_id: The id of the tenant which the identity belongs to. + :vartype tenant_id: str + :ivar principal_id: The id of the service principal object associated with the created + identity. + :vartype principal_id: str + :ivar client_id: The id of the app associated with the identity. This is a random generated + UUID by MSI. + :vartype client_id: str + :ivar isolation_scope: Enum to configure regional restrictions on identity assignment, as + necessary. Known values are: "None" and "Regional". + :vartype isolation_scope: str or ~azure.mgmt.msi.models.IsolationScope + :ivar assignment_restrictions: Restrictions on which resource providers this identity can be + assigned to. + :vartype assignment_restrictions: ~azure.mgmt.msi.models.AssignmentRestrictions + """ + + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The id of the tenant which the identity belongs to.""" + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The id of the service principal object associated with the created identity.""" + client_id: Optional[str] = rest_field(name="clientId", visibility=["read"]) + """The id of the app associated with the identity. This is a random generated UUID by MSI.""" + isolation_scope: Optional[Union[str, "_models.IsolationScope"]] = rest_field( + name="isolationScope", visibility=["read", "create", "update"] + ) + """Enum to configure regional restrictions on identity assignment, as necessary. Known values are: + \"None\" and \"Regional\".""" + assignment_restrictions: Optional["_models.AssignmentRestrictions"] = rest_field( + name="assignmentRestrictions", visibility=["read", "create", "update"] + ) + """Restrictions on which resource providers this identity can be assigned to.""" + + @overload + def __init__( + self, + *, + isolation_scope: Optional[Union[str, "_models.IsolationScope"]] = None, + assignment_restrictions: Optional["_models.AssignmentRestrictions"] = 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/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models_py3.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models_py3.py deleted file mode 100644 index 99e46fa5591a..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_models_py3.py +++ /dev/null @@ -1,685 +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, Optional, TYPE_CHECKING, Union - -from .._utils import serialization as _serialization - -if TYPE_CHECKING: - from .. import models as _models - - -class CloudErrorBody(_serialization.Model): - """An error response from the ManagedServiceIdentity service. - - :ivar code: An identifier for the error. - :vartype code: str - :ivar message: A message describing the error, intended to be suitable for display in a user - interface. - :vartype message: str - :ivar target: The target of the particular error. For example, the name of the property in - error. - :vartype target: str - :ivar details: A list of additional details about the error. - :vartype details: list[~azure.mgmt.msi.models.CloudErrorBody] - """ - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "message": {"key": "message", "type": "str"}, - "target": {"key": "target", "type": "str"}, - "details": {"key": "details", "type": "[CloudErrorBody]"}, - } - - def __init__( - self, - *, - code: Optional[str] = None, - message: Optional[str] = None, - target: Optional[str] = None, - details: Optional[List["_models.CloudErrorBody"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword code: An identifier for the error. - :paramtype code: str - :keyword message: A message describing the error, intended to be suitable for display in a user - interface. - :paramtype message: str - :keyword target: The target of the particular error. For example, the name of the property in - error. - :paramtype target: str - :keyword details: A list of additional details about the error. - :paramtype details: list[~azure.mgmt.msi.models.CloudErrorBody] - """ - super().__init__(**kwargs) - self.code = code - self.message = message - self.target = target - self.details = details - - -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. E.g. - "/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.msi.models.SystemData - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.id: Optional[str] = None - self.name: Optional[str] = None - self.type: Optional[str] = None - self.system_data: Optional["_models.SystemData"] = 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. E.g. - "/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.msi.models.SystemData - """ - - -class FederatedIdentityCredential(ProxyResource): - """Describes a federated identity credential. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Fully qualified resource ID for the resource. E.g. - "/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.msi.models.SystemData - :ivar issuer: The URL of the issuer to be trusted. - :vartype issuer: str - :ivar subject: The identifier of the external identity. - :vartype subject: str - :ivar audiences: The list of audiences that can appear in the issued token. - :vartype audiences: list[str] - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "issuer": {"key": "properties.issuer", "type": "str"}, - "subject": {"key": "properties.subject", "type": "str"}, - "audiences": {"key": "properties.audiences", "type": "[str]"}, - } - - def __init__( - self, - *, - issuer: Optional[str] = None, - subject: Optional[str] = None, - audiences: Optional[List[str]] = None, - **kwargs: Any - ) -> None: - """ - :keyword issuer: The URL of the issuer to be trusted. - :paramtype issuer: str - :keyword subject: The identifier of the external identity. - :paramtype subject: str - :keyword audiences: The list of audiences that can appear in the issued token. - :paramtype audiences: list[str] - """ - super().__init__(**kwargs) - self.issuer = issuer - self.subject = subject - self.audiences = audiences - - -class FederatedIdentityCredentialsListResult(_serialization.Model): - """Values returned by the List operation for federated identity credentials. - - :ivar value: The collection of federated identity credentials returned by the listing - operation. - :vartype value: list[~azure.mgmt.msi.models.FederatedIdentityCredential] - :ivar next_link: The url to get the next page of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[FederatedIdentityCredential]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, - *, - value: Optional[List["_models.FederatedIdentityCredential"]] = None, - next_link: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword value: The collection of federated identity credentials returned by the listing - operation. - :paramtype value: list[~azure.mgmt.msi.models.FederatedIdentityCredential] - :keyword next_link: The url to get the next page of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class TrackedResource(Resource): - """The resource model definition for an Azure Resource Manager tracked top level resource which - has 'tags' and a 'location'. - - 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 resource ID for the resource. E.g. - "/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.msi.models.SystemData - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar location: The geo-location where the resource lives. Required. - :vartype location: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "location": {"required": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "tags": {"key": "tags", "type": "{str}"}, - "location": {"key": "location", "type": "str"}, - } - - def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - :keyword location: The geo-location where the resource lives. Required. - :paramtype location: str - """ - super().__init__(**kwargs) - self.tags = tags - self.location = location - - -class Identity(TrackedResource): - """Describes an identity resource. - - 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 resource ID for the resource. E.g. - "/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.msi.models.SystemData - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar location: The geo-location where the resource lives. Required. - :vartype location: str - :ivar tenant_id: The id of the tenant which the identity belongs to. - :vartype tenant_id: str - :ivar principal_id: The id of the service principal object associated with the created - identity. - :vartype principal_id: str - :ivar client_id: The id of the app associated with the identity. This is a random generated - UUID by MSI. - :vartype client_id: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "location": {"required": True}, - "tenant_id": {"readonly": True}, - "principal_id": {"readonly": True}, - "client_id": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "tags": {"key": "tags", "type": "{str}"}, - "location": {"key": "location", "type": "str"}, - "tenant_id": {"key": "properties.tenantId", "type": "str"}, - "principal_id": {"key": "properties.principalId", "type": "str"}, - "client_id": {"key": "properties.clientId", "type": "str"}, - } - - def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - :keyword location: The geo-location where the resource lives. Required. - :paramtype location: str - """ - super().__init__(tags=tags, location=location, **kwargs) - self.tenant_id: Optional[str] = None - self.principal_id: Optional[str] = None - self.client_id: Optional[str] = None - - -class IdentityUpdate(Resource): - """Describes an identity resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Fully qualified resource ID for the resource. E.g. - "/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.msi.models.SystemData - :ivar location: The geo-location where the resource lives. - :vartype location: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar tenant_id: The id of the tenant which the identity belongs to. - :vartype tenant_id: str - :ivar principal_id: The id of the service principal object associated with the created - identity. - :vartype principal_id: str - :ivar client_id: The id of the app associated with the identity. This is a random generated - UUID by MSI. - :vartype client_id: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "tenant_id": {"readonly": True}, - "principal_id": {"readonly": True}, - "client_id": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "location": {"key": "location", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - "tenant_id": {"key": "properties.tenantId", "type": "str"}, - "principal_id": {"key": "properties.principalId", "type": "str"}, - "client_id": {"key": "properties.clientId", "type": "str"}, - } - - def __init__(self, *, location: Optional[str] = None, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword location: The geo-location where the resource lives. - :paramtype location: str - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(**kwargs) - self.location = location - self.tags = tags - self.tenant_id: Optional[str] = None - self.principal_id: Optional[str] = None - self.client_id: Optional[str] = None - - -class Operation(_serialization.Model): - """Operation supported by the Microsoft.ManagedIdentity REST API. - - :ivar name: The name of the REST Operation. This is of the format - {provider}/{resource}/{operation}. - :vartype name: str - :ivar display: The object that describes the operation. - :vartype display: ~azure.mgmt.msi.models.OperationDisplay - """ - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "display": {"key": "display", "type": "OperationDisplay"}, - } - - def __init__( - self, *, name: Optional[str] = None, display: Optional["_models.OperationDisplay"] = None, **kwargs: Any - ) -> None: - """ - :keyword name: The name of the REST Operation. This is of the format - {provider}/{resource}/{operation}. - :paramtype name: str - :keyword display: The object that describes the operation. - :paramtype display: ~azure.mgmt.msi.models.OperationDisplay - """ - super().__init__(**kwargs) - self.name = name - self.display = display - - -class OperationDisplay(_serialization.Model): - """The object that describes the operation. - - :ivar provider: Friendly name of the resource provider. - :vartype provider: str - :ivar operation: The type of operation. For example: read, write, delete. - :vartype operation: str - :ivar resource: The resource type on which the operation is performed. - :vartype resource: str - :ivar description: A description of the operation. - :vartype description: str - """ - - _attribute_map = { - "provider": {"key": "provider", "type": "str"}, - "operation": {"key": "operation", "type": "str"}, - "resource": {"key": "resource", "type": "str"}, - "description": {"key": "description", "type": "str"}, - } - - def __init__( - self, - *, - provider: Optional[str] = None, - operation: Optional[str] = None, - resource: Optional[str] = None, - description: Optional[str] = None, - **kwargs: Any - ) -> None: - """ - :keyword provider: Friendly name of the resource provider. - :paramtype provider: str - :keyword operation: The type of operation. For example: read, write, delete. - :paramtype operation: str - :keyword resource: The resource type on which the operation is performed. - :paramtype resource: str - :keyword description: A description of the operation. - :paramtype description: str - """ - super().__init__(**kwargs) - self.provider = provider - self.operation = operation - self.resource = resource - self.description = description - - -class OperationListResult(_serialization.Model): - """A list of operations supported by Microsoft.ManagedIdentity Resource Provider. - - :ivar value: A list of operations supported by Microsoft.ManagedIdentity Resource Provider. - :vartype value: list[~azure.mgmt.msi.models.Operation] - :ivar next_link: The url to get the next page of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[Operation]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[List["_models.Operation"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: A list of operations supported by Microsoft.ManagedIdentity Resource Provider. - :paramtype value: list[~azure.mgmt.msi.models.Operation] - :keyword next_link: The url to get the next page of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link - - -class SystemAssignedIdentity(ProxyResource): - """Describes a system assigned identity resource. - - 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 resource ID for the resource. E.g. - "/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.msi.models.SystemData - :ivar location: The geo-location where the resource lives. Required. - :vartype location: str - :ivar tags: Resource tags. - :vartype tags: dict[str, str] - :ivar tenant_id: The id of the tenant which the identity belongs to. - :vartype tenant_id: str - :ivar principal_id: The id of the service principal object associated with the created - identity. - :vartype principal_id: str - :ivar client_id: The id of the app associated with the identity. This is a random generated - UUID by MSI. - :vartype client_id: str - :ivar client_secret_url: The ManagedServiceIdentity DataPlane URL that can be queried to obtain - the identity credentials. - :vartype client_secret_url: str - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "location": {"required": True}, - "tenant_id": {"readonly": True}, - "principal_id": {"readonly": True}, - "client_id": {"readonly": True}, - "client_secret_url": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - "name": {"key": "name", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "system_data": {"key": "systemData", "type": "SystemData"}, - "location": {"key": "location", "type": "str"}, - "tags": {"key": "tags", "type": "{str}"}, - "tenant_id": {"key": "properties.tenantId", "type": "str"}, - "principal_id": {"key": "properties.principalId", "type": "str"}, - "client_id": {"key": "properties.clientId", "type": "str"}, - "client_secret_url": {"key": "properties.clientSecretUrl", "type": "str"}, - } - - def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kwargs: Any) -> None: - """ - :keyword location: The geo-location where the resource lives. Required. - :paramtype location: str - :keyword tags: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(**kwargs) - self.location = location - self.tags = tags - self.tenant_id: Optional[str] = None - self.principal_id: Optional[str] = None - self.client_id: Optional[str] = None - self.client_secret_url: Optional[str] = None - - -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.msi.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.msi.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.msi.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.msi.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 - - -class UserAssignedIdentitiesListResult(_serialization.Model): - """Values returned by the List operation. - - :ivar value: The collection of userAssignedIdentities returned by the listing operation. - :vartype value: list[~azure.mgmt.msi.models.Identity] - :ivar next_link: The url to get the next page of results, if any. - :vartype next_link: str - """ - - _attribute_map = { - "value": {"key": "value", "type": "[Identity]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__( - self, *, value: Optional[List["_models.Identity"]] = None, next_link: Optional[str] = None, **kwargs: Any - ) -> None: - """ - :keyword value: The collection of userAssignedIdentities returned by the listing operation. - :paramtype value: list[~azure.mgmt.msi.models.Identity] - :keyword next_link: The url to get the next page of results, if any. - :paramtype next_link: str - """ - super().__init__(**kwargs) - self.value = value - self.next_link = next_link diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_patch.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_patch.py index 8bcb627aa475..ea765788358a 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_patch.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/models/_patch.py @@ -7,9 +7,8 @@ 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/resources/azure-mgmt-msi/azure/mgmt/msi/operations/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/__init__.py index b50b543ebd92..f8dc23d82292 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/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,20 +12,20 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._system_assigned_identities_operations import SystemAssignedIdentitiesOperations # type: ignore from ._operations import Operations # type: ignore -from ._user_assigned_identities_operations import UserAssignedIdentitiesOperations # type: ignore -from ._federated_identity_credentials_operations import FederatedIdentityCredentialsOperations # type: ignore +from ._operations import SystemAssignedIdentitiesOperations # type: ignore +from ._operations import FederatedIdentityCredentialsOperations # type: ignore +from ._operations import UserAssignedIdentitiesOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk __all__ = [ - "SystemAssignedIdentitiesOperations", "Operations", - "UserAssignedIdentitiesOperations", + "SystemAssignedIdentitiesOperations", "FederatedIdentityCredentialsOperations", + "UserAssignedIdentitiesOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_federated_identity_credentials_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_federated_identity_credentials_operations.py deleted file mode 100644 index 2b9a2b70e7de..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_federated_identity_credentials_operations.py +++ /dev/null @@ -1,602 +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 io import IOBase -from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -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 ManagedServiceIdentityClientConfiguration -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_list_request( - resource_group_name: str, - resource_name: str, - subscription_id: str, - *, - top: Optional[int] = None, - skiptoken: Optional[str] = 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - if top is not None: - _params["$top"] = _SERIALIZER.query("top", top, "int", minimum=1) - if skiptoken is not None: - _params["$skiptoken"] = _SERIALIZER.query("skiptoken", skiptoken, "str") - _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_create_or_update_request( - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: 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", "2023-01-31")) - 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/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), - "federatedIdentityCredentialResourceName": _SERIALIZER.url( - "federated_identity_credential_resource_name", - federated_identity_credential_resource_name, - "str", - pattern=r"^[a-zA-Z0-9]{1}[a-zA-Z0-9-_]{2,119}$", - ), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - _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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), - "federatedIdentityCredentialResourceName": _SERIALIZER.url( - "federated_identity_credential_resource_name", - federated_identity_credential_resource_name, - "str", - pattern=r"^[a-zA-Z0-9]{1}[a-zA-Z0-9-_]{2,119}$", - ), - } - - _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_delete_request( - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), - "federatedIdentityCredentialResourceName": _SERIALIZER.url( - "federated_identity_credential_resource_name", - federated_identity_credential_resource_name, - "str", - pattern=r"^[a-zA-Z0-9]{1}[a-zA-Z0-9-_]{2,119}$", - ), - } - - _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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -class FederatedIdentityCredentialsOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s - :attr:`federated_identity_credentials` attribute. - """ - - models = _models - - 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: ManagedServiceIdentityClientConfiguration = ( - 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 list( - self, - resource_group_name: str, - resource_name: str, - top: Optional[int] = None, - skiptoken: Optional[str] = None, - **kwargs: Any - ) -> ItemPaged["_models.FederatedIdentityCredential"]: - """Lists all the federated identity credentials under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param top: Number of records to return. Default value is None. - :type top: int - :param skiptoken: A skip token is used to continue retrieving items after an operation returns - a partial result. If a previous response contains a nextLink element, the value of the nextLink - element will include a skipToken parameter that specifies a starting point to use for - subsequent calls. Default value is None. - :type skiptoken: str - :return: An iterator like instance of either FederatedIdentityCredential or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.FederatedIdentityCredential] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.FederatedIdentityCredentialsListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - subscription_id=self._config.subscription_id, - top=top, - skiptoken=skiptoken, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("FederatedIdentityCredentialsListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: _models.FederatedIdentityCredential, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Required. - :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Required. - :type parameters: 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: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - parameters: Union[_models.FederatedIdentityCredential, IO[bytes]], - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Create or update a federated identity credential under the specified user assigned identity. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :param parameters: Parameters to create or update the federated identity credential. Is either - a FederatedIdentityCredential type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential or IO[bytes] - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :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 = 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[_models.FederatedIdentityCredential] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "FederatedIdentityCredential") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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) - - _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, 201]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("FederatedIdentityCredential", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def get( - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - **kwargs: Any - ) -> _models.FederatedIdentityCredential: - """Gets the federated identity credential. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :return: FederatedIdentityCredential or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential - :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.FederatedIdentityCredential] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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 = 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("FederatedIdentityCredential", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, - resource_group_name: str, - resource_name: str, - federated_identity_credential_resource_name: str, - **kwargs: Any - ) -> None: - """Deletes the federated identity credential. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param federated_identity_credential_resource_name: The name of the federated identity - credential resource. Required. - :type federated_identity_credential_resource_name: str - :return: None or the result of cls(response) - :rtype: None - :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[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - resource_name=resource_name, - federated_identity_credential_resource_name=federated_identity_credential_resource_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 = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 204]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py index 632a6d25a1f1..2d42ba65f04c 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py @@ -1,12 +1,15 @@ +# 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) 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 typing import Any, Callable, Dict, Optional, TypeVar +from io import IOBase +import json +from typing import Any, Callable, IO, Optional, TypeVar, Union, overload import urllib.parse from azure.core import PipelineClient @@ -16,6 +19,8 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.paging import ItemPaged @@ -26,25 +31,359 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models -from .._configuration import ManagedServiceIdentityClientConfiguration +from .._configuration import ManagedIdentityClientConfiguration +from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._utils.serialization import Deserializer, Serializer 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]] +JSON = MutableMapping[str, Any] +List = list _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_list_request(**kwargs: Any) -> HttpRequest: +def build_operations_list_request(**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", "2023-01-31")) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-05-31-preview")) accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop("template_url", "/providers/Microsoft.ManagedIdentity/operations") + _url = "/providers/Microsoft.ManagedIdentity/operations" + + # 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_system_assigned_identities_get_by_scope_request( # pylint: disable=name-too-long + scope: 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/{scope}/providers/Microsoft.ManagedIdentity/identities/default" + path_format_arguments = { + "scope": _SERIALIZER.url("scope", scope, "str", skip_quote=True), + } + + _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_federated_identity_credentials_get_request( # pylint: disable=name-too-long + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), + "federatedIdentityCredentialResourceName": _SERIALIZER.url( + "federated_identity_credential_resource_name", federated_identity_credential_resource_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_federated_identity_credentials_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), + "federatedIdentityCredentialResourceName": _SERIALIZER.url( + "federated_identity_credential_resource_name", federated_identity_credential_resource_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 + 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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_federated_identity_credentials_delete_request( # pylint: disable=name-too-long + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + subscription_id: str, + **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-05-31-preview")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials/{federatedIdentityCredentialResourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), + "federatedIdentityCredentialResourceName": _SERIALIZER.url( + "federated_identity_credential_resource_name", federated_identity_credential_resource_name, "str" + ), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_federated_identity_credentials_list_request( # pylint: disable=name-too-long + resource_group_name: str, + resource_name: str, + subscription_id: str, + *, + top: Optional[int] = None, + skiptoken: Optional[str] = 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}/federatedIdentityCredentials" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if top is not None: + _params["$top"] = _SERIALIZER.query("top", top, "int") + if skiptoken is not None: + _params["$skiptoken"] = _SERIALIZER.query("skiptoken", skiptoken, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_user_assigned_identities_get_request( # pylint: disable=name-too-long + resource_group_name: str, resource_name: 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_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_user_assigned_identities_create_or_update_request( # pylint: disable=name-too-long + resource_group_name: str, resource_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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_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 + 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="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_user_assigned_identities_update_request( # pylint: disable=name-too-long + resource_group_name: str, resource_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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_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 + 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) + + +def build_user_assigned_identities_delete_request( # pylint: disable=name-too-long + resource_group_name: str, resource_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2025-05-31-preview")) + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "resourceName": _SERIALIZER.url("resource_name", resource_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_user_assigned_identities_list_by_resource_group_request( # pylint: disable=name-too-long + resource_group_name: 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_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_user_assigned_identities_list_by_subscription_request( # pylint: disable=name-too-long + 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", "2025-05-31-preview")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.ManagedIdentity/userAssignedIdentities" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore # Construct parameters _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") @@ -61,34 +400,29 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s + :class:`~azure.mgmt.msi.ManagedIdentityClient`'s :attr:`operations` attribute. """ - models = _models - 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: ManagedServiceIdentityClientConfiguration = ( - input_args.pop(0) if input_args else kwargs.pop("config") - ) + self._config: ManagedIdentityClientConfiguration = 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 list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: - """Lists available operations for the Microsoft.ManagedIdentity provider. + """List the operations for the provider. - :return: An iterator like instance of either Operation or the result of cls(response) + :return: An iterator like instance of Operation :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.Operation] :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.OperationListResult] = kwargs.pop("cls", None) + cls: ClsType[List[_models.Operation]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -101,12 +435,17 @@ def list(self, **kwargs: Any) -> ItemPaged["_models.Operation"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_operations_list_request( + 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 @@ -119,18 +458,29 @@ def prepare_request(next_link=None): ) _next_request_params["api-version"] = self._config.api_version _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + 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("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Operation], + 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) @@ -143,7 +493,1245 @@ 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class SystemAssignedIdentitiesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :attr:`system_assigned_identities` 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: ManagedIdentityClientConfiguration = 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_by_scope(self, scope: str, **kwargs: Any) -> _models.SystemAssignedIdentity: + """Gets the systemAssignedIdentity available under the specified RP scope. + + :param scope: The fully qualified Azure Resource manager identifier of the resource. Required. + :type scope: str + :return: SystemAssignedIdentity. The SystemAssignedIdentity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.SystemAssignedIdentity + :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.SystemAssignedIdentity] = kwargs.pop("cls", None) + + _request = build_system_assigned_identities_get_by_scope_request( + scope=scope, + 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.CloudError, + 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.SystemAssignedIdentity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + +class FederatedIdentityCredentialsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :attr:`federated_identity_credentials` 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: ManagedIdentityClientConfiguration = 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, + resource_name: str, + federated_identity_credential_resource_name: str, + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Gets the federated identity credential. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :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.FederatedIdentityCredential] = kwargs.pop("cls", None) + + _request = build_federated_identity_credentials_get_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_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.CloudError, + 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.FederatedIdentityCredential, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: _models.FederatedIdentityCredential, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Required. + :type parameters: 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: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + parameters: Union[_models.FederatedIdentityCredential, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.FederatedIdentityCredential: + """Create or update a federated identity credential under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :param parameters: Parameters to create or update the federated identity credential. Is one of + the following types: FederatedIdentityCredential, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.FederatedIdentityCredential or JSON or IO[bytes] + :return: FederatedIdentityCredential. The FederatedIdentityCredential is compatible with + MutableMapping + :rtype: ~azure.mgmt.msi.models.FederatedIdentityCredential + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.FederatedIdentityCredential] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_federated_identity_credentials_create_or_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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, 201]: + 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.CloudError, + 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.FederatedIdentityCredential, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, + resource_group_name: str, + resource_name: str, + federated_identity_credential_resource_name: str, + **kwargs: Any + ) -> None: + """Deletes the federated identity credential. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param federated_identity_credential_resource_name: The name of the federated identity + credential resource. Required. + :type federated_identity_credential_resource_name: str + :return: None + :rtype: None + :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[None] = kwargs.pop("cls", None) + + _request = build_federated_identity_credentials_delete_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + federated_identity_credential_resource_name=federated_identity_credential_resource_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list( + self, + resource_group_name: str, + resource_name: str, + *, + top: Optional[int] = None, + skiptoken: Optional[str] = None, + **kwargs: Any + ) -> ItemPaged["_models.FederatedIdentityCredential"]: + """Lists all the federated identity credentials under the specified user assigned identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :keyword top: Number of records to return. Default value is None. + :paramtype top: int + :keyword skiptoken: A skip token is used to continue retrieving items after an operation + returns a partial result. If a previous response contains a nextLink element, the value of the + nextLink element will include a skipToken parameter that specifies a starting point to use for + subsequent calls. Default value is None. + :paramtype skiptoken: str + :return: An iterator like instance of FederatedIdentityCredential + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.FederatedIdentityCredential] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.FederatedIdentityCredential]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_federated_identity_credentials_list_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + top=top, + skiptoken=skiptoken, + 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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.FederatedIdentityCredential], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _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 = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class UserAssignedIdentitiesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :attr:`user_assigned_identities` 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: ManagedIdentityClientConfiguration = 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, resource_name: str, **kwargs: Any) -> _models.Identity: + """Gets the identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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.Identity] = kwargs.pop("cls", None) + + _request = build_user_assigned_identities_get_request( + resource_group_name=resource_group_name, + resource_name=resource_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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: _models.Identity, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: ~azure.mgmt.msi.models.Identity + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Required. + :type parameters: 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: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + resource_name: str, + parameters: Union[_models.Identity, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.Identity: + """Create or update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to create or update the identity. Is one of the following types: + Identity, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.Identity or JSON or IO[bytes] + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Identity] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_user_assigned_identities_create_or_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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, 201]: + 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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @overload + def update( + self, + resource_group_name: str, + resource_name: str, + parameters: _models.IdentityUpdate, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: ~azure.mgmt.msi.models.IdentityUpdate + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + resource_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: JSON + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + resource_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Required. + :type parameters: 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: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + resource_name: str, + parameters: Union[_models.IdentityUpdate, JSON, IO[bytes]], + **kwargs: Any + ) -> _models.Identity: + """Update an identity in the specified subscription and resource group. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :param parameters: Parameters to update the identity. Is one of the following types: + IdentityUpdate, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.msi.models.IdentityUpdate or JSON or IO[bytes] + :return: Identity. The Identity is compatible with MutableMapping + :rtype: ~azure.mgmt.msi.models.Identity + :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 = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Identity] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_user_assigned_identities_update_request( + resource_group_name=resource_group_name, + resource_name=resource_name, + subscription_id=self._config.subscription_id, + content_type=content_type, + api_version=self._config.api_version, + content=_content, + 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.CloudError, + 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.Identity, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, resource_name: str, **kwargs: Any + ) -> None: + """Deletes the identity. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param resource_name: The name of the identity resource. Required. + :type resource_name: str + :return: None + :rtype: None + :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[None] = kwargs.pop("cls", None) + + _request = build_user_assigned_identities_delete_request( + resource_group_name=resource_group_name, + resource_name=resource_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) + + _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, 204]: + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Identity"]: + """Lists all the userAssignedIdentities available under the specified ResourceGroup. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :return: An iterator like instance of Identity + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.Identity] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Identity]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_user_assigned_identities_list_by_resource_group_request( + resource_group_name=resource_group_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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Identity], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _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 = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.Identity"]: + """Lists all the userAssignedIdentities available under the specified subscription. + + :return: An iterator like instance of Identity + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.Identity] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.Identity]] = kwargs.pop("cls", None) + + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + def prepare_request(next_link=None): + if not next_link: + + _request = build_user_assigned_identities_list_by_subscription_request( + 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) + + else: + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + _request = HttpRequest( + "GET", + urllib.parse.urljoin(next_link, _parsed_next_link.path), + headers=_headers, + params=_next_request_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) + + return _request + + def extract_data(pipeline_response): + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize( + List[_models.Identity], + deserialized.get("value", []), + ) + if cls: + list_of_elem = cls(list_of_elem) # type: ignore + return deserialized.get("nextLink") or None, iter(list_of_elem) + + def get_next(next_link=None): + _request = prepare_request(next_link) + + _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 = _failsafe_deserialize( + _models.CloudError, + response, + ) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_patch.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_patch.py index 8bcb627aa475..ea765788358a 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_patch.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_patch.py @@ -7,9 +7,8 @@ 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/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_system_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_system_assigned_identities_operations.py deleted file mode 100644 index 90d252314c6e..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_system_assigned_identities_operations.py +++ /dev/null @@ -1,131 +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 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 ManagedServiceIdentityClientConfiguration -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_by_scope_request(scope: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/{scope}/providers/Microsoft.ManagedIdentity/identities/default") - path_format_arguments = { - "scope": _SERIALIZER.url("scope", scope, "str", skip_quote=True), - } - - _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 SystemAssignedIdentitiesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s - :attr:`system_assigned_identities` attribute. - """ - - models = _models - - 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: ManagedServiceIdentityClientConfiguration = ( - 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_by_scope(self, scope: str, **kwargs: Any) -> _models.SystemAssignedIdentity: - """Gets the systemAssignedIdentity available under the specified RP scope. - - :param scope: The resource provider scope of the resource. Parent resource being extended by - Managed Identities. Required. - :type scope: str - :return: SystemAssignedIdentity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.SystemAssignedIdentity - :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.SystemAssignedIdentity] = kwargs.pop("cls", None) - - _request = build_get_by_scope_request( - scope=scope, - 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("SystemAssignedIdentity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_user_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_user_assigned_identities_operations.py deleted file mode 100644 index b3c6b695f6b7..000000000000 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_user_assigned_identities_operations.py +++ /dev/null @@ -1,762 +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 io import IOBase -from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -import urllib.parse - -from azure.core import PipelineClient -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.paging import ItemPaged -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 ManagedServiceIdentityClientConfiguration -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_list_by_subscription_request(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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.ManagedIdentity/userAssignedIdentities" - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_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) - - -def build_list_by_resource_group_request(resource_group_name: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_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_create_or_update_request( - resource_group_name: str, resource_name: 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", "2023-01-31")) - 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/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), - "resourceName": _SERIALIZER.url("resource_name", resource_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 - 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="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_update_request( - resource_group_name: str, resource_name: 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", "2023-01-31")) - 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/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), - "resourceName": _SERIALIZER.url("resource_name", resource_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 - 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) - - -def build_get_request(resource_group_name: str, resource_name: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), - "resourceName": _SERIALIZER.url("resource_name", resource_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_delete_request( - resource_group_name: str, resource_name: 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", "2023-01-31")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{resourceName}", - ) - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), - "resourceName": _SERIALIZER.url("resource_name", resource_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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -class UserAssignedIdentitiesOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s - :attr:`user_assigned_identities` attribute. - """ - - models = _models - - 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: ManagedServiceIdentityClientConfiguration = ( - 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 list_by_subscription(self, **kwargs: Any) -> ItemPaged["_models.Identity"]: - """Lists all the userAssignedIdentities available under the specified subscription. - - :return: An iterator like instance of either Identity or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.Identity] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UserAssignedIdentitiesListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_subscription_request( - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("UserAssignedIdentitiesListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> ItemPaged["_models.Identity"]: - """Lists all the userAssignedIdentities available under the specified ResourceGroup. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :return: An iterator like instance of either Identity or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.msi.models.Identity] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.UserAssignedIdentitiesListResult] = kwargs.pop("cls", None) - - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - def prepare_request(next_link=None): - if not next_link: - - _request = build_list_by_resource_group_request( - resource_group_name=resource_group_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - else: - # make call to next link with the client's api-version - _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) - _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" - return _request - - def extract_data(pipeline_response): - deserialized = self._deserialize("UserAssignedIdentitiesListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, iter(list_of_elem) - - def get_next(next_link=None): - _request = prepare_request(next_link) - - _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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: _models.Identity, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Required. - :type parameters: ~azure.mgmt.msi.models.Identity - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Required. - :type parameters: 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: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def create_or_update( - self, - resource_group_name: str, - resource_name: str, - parameters: Union[_models.Identity, IO[bytes]], - **kwargs: Any - ) -> _models.Identity: - """Create or update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to create or update the identity. Is either a Identity type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.Identity or IO[bytes] - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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 = 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[_models.Identity] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "Identity") - - _request = build_create_or_update_request( - resource_group_name=resource_group_name, - resource_name=resource_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) - - _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, 201]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @overload - def update( - self, - resource_group_name: str, - resource_name: str, - parameters: _models.IdentityUpdate, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Required. - :type parameters: ~azure.mgmt.msi.models.IdentityUpdate - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def update( - self, - resource_group_name: str, - resource_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Required. - :type parameters: 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: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def update( - self, - resource_group_name: str, - resource_name: str, - parameters: Union[_models.IdentityUpdate, IO[bytes]], - **kwargs: Any - ) -> _models.Identity: - """Update an identity in the specified subscription and resource group. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :param parameters: Parameters to update the identity. Is either a IdentityUpdate type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.msi.models.IdentityUpdate or IO[bytes] - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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 = 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[_models.Identity] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "IdentityUpdate") - - _request = build_update_request( - resource_group_name=resource_group_name, - resource_name=resource_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) - - _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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def get(self, resource_group_name: str, resource_name: str, **kwargs: Any) -> _models.Identity: - """Gets the identity. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :return: Identity or the result of cls(response) - :rtype: ~azure.mgmt.msi.models.Identity - :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.Identity] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - resource_name=resource_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 = 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) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - deserialized = self._deserialize("Identity", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def delete( # pylint: disable=inconsistent-return-statements - self, resource_group_name: str, resource_name: str, **kwargs: Any - ) -> None: - """Deletes the identity. - - :param resource_group_name: The name of the Resource Group to which the identity belongs. - Required. - :type resource_group_name: str - :param resource_name: The name of the identity resource. Required. - :type resource_name: str - :return: None or the result of cls(response) - :rtype: None - :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[None] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - resource_name=resource_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 = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 204]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response, error_format=ARMErrorFormat) - - if cls: - return cls(pipeline_response, None, {}) # type: ignore diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py index 77e834ae0f4f..123c76388fda 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="c267c0e7-0a73-4789-9e17-d26aeb0904e5", + subscription_id="SUBSCRIPTION_ID", ) response = client.federated_identity_credentials.create_or_update( @@ -46,6 +45,6 @@ def main(): print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/FederatedIdentityCredentialCreate.json +# x-ms-original-file: 2025-05-31-preview/FederatedIdentityCredentialCreate.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py index e4661a27238e..241d1188b0ac 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="c267c0e7-0a73-4789-9e17-d26aeb0904e5", + subscription_id="SUBSCRIPTION_ID", ) client.federated_identity_credentials.delete( @@ -38,6 +37,6 @@ def main(): ) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/FederatedIdentityCredentialDelete.json +# x-ms-original-file: 2025-05-31-preview/FederatedIdentityCredentialDelete.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py index f2c893ac9b37..b1ae8f5cfb65 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="c267c0e7-0a73-4789-9e17-d26aeb0904e5", + subscription_id="SUBSCRIPTION_ID", ) response = client.federated_identity_credentials.get( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/FederatedIdentityCredentialGet.json +# x-ms-original-file: 2025-05-31-preview/FederatedIdentityCredentialGet.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py index c8cb9ef6c68e..4d02f87efebd 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="c267c0e7-0a73-4789-9e17-d26aeb0904e5", + subscription_id="SUBSCRIPTION_ID", ) response = client.federated_identity_credentials.list( @@ -39,6 +38,6 @@ def main(): print(item) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/FederatedIdentityCredentialList.json +# x-ms-original-file: 2025-05-31-preview/FederatedIdentityCredentialList.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py new file mode 100644 index 000000000000..31cbd208ffd9 --- /dev/null +++ b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py @@ -0,0 +1,53 @@ +# 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. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.msi import ManagedIdentityClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-msi +# USAGE + python flexible_federated_identity_credential_create.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = ManagedIdentityClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.federated_identity_credentials.create_or_update( + resource_group_name="rgName", + resource_name="resourceName", + federated_identity_credential_resource_name="ficResourceName", + parameters={ + "properties": { + "audiences": ["api://AzureADTokenExchange"], + "claimsMatchingExpression": { + "languageVersion": 1, + "value": "claims['sub'] matches system:serviceaccount:ns:*", + }, + "issuer": "https://oidc.prod-aks.azure.com/TenantGUID/IssuerGUID", + } + }, + ) + print(response) + + +# x-ms-original-file: 2025-05-31-preview/FlexibleFederatedIdentityCredentialCreate.json +if __name__ == "__main__": + main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py new file mode 100644 index 000000000000..fdb5dd839d5c --- /dev/null +++ b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py @@ -0,0 +1,43 @@ +# 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. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from azure.mgmt.msi import ManagedIdentityClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-msi +# USAGE + python flexible_federated_identity_credential_get.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = ManagedIdentityClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.federated_identity_credentials.get( + resource_group_name="rgName", + resource_name="resourceName", + federated_identity_credential_resource_name="ficResourceName", + ) + print(response) + + +# x-ms-original-file: 2025-05-31-preview/FlexibleFederatedIdentityCredentialGet.json +if __name__ == "__main__": + main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py index 76424005058e..691639e1af1b 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) response = client.user_assigned_identities.create_or_update( @@ -39,6 +38,6 @@ def main(): print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityCreate.json +# x-ms-original-file: 2025-05-31-preview/IdentityCreate.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py index d8b78de75886..54a3729bc049 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) client.user_assigned_identities.delete( @@ -37,6 +36,6 @@ def main(): ) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityDelete.json +# x-ms-original-file: 2025-05-31-preview/IdentityDelete.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py index 6cad409dd19c..b16d5da9b55b 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) response = client.user_assigned_identities.get( @@ -38,6 +37,6 @@ def main(): print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityGet.json +# x-ms-original-file: 2025-05-31-preview/IdentityGet.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py index ad380f691628..b7ad9c2a95cb 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) response = client.user_assigned_identities.list_by_resource_group( @@ -38,6 +37,6 @@ def main(): print(item) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityListByResourceGroup.json +# x-ms-original-file: 2025-05-31-preview/IdentityListByResourceGroup.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py index 7eeb13622b1c..f55070ae77b6 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,9 +25,9 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) response = client.user_assigned_identities.list_by_subscription() @@ -36,6 +35,6 @@ def main(): print(item) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityListBySubscription.json +# x-ms-original-file: 2025-05-31-preview/IdentityListBySubscription.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py index b3914ce7ca81..d972a87b5997 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,19 +25,26 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), - subscription_id="subid", + subscription_id="SUBSCRIPTION_ID", ) response = client.user_assigned_identities.update( resource_group_name="rgName", resource_name="resourceName", - parameters={"location": "eastus", "tags": {"key1": "value1", "key2": "value2"}}, + parameters={ + "location": "eastus", + "properties": { + "assignmentRestrictions": {"providers": ["Microsoft.Compute", "Microsoft.Storage/Accounts"]}, + "isolationScope": "Regional", + }, + "tags": {"key1": "value1", "key2": "value2"}, + }, ) print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/IdentityUpdate.json +# x-ms-original-file: 2025-05-31-preview/IdentityUpdate.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py b/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py index e97cd374d1fa..07f0871c1e51 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py @@ -1,15 +1,14 @@ -# 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. # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,7 +25,7 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) @@ -36,6 +35,6 @@ def main(): print(item) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/MsiOperationsList.json +# x-ms-original-file: 2025-05-31-preview/MsiOperationsList.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py index 0723e603cf06..15963c81e63f 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py @@ -3,13 +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 azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient """ # PREREQUISITES @@ -26,17 +26,17 @@ def main(): - client = ManagedServiceIdentityClient( + client = ManagedIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) response = client.system_assigned_identities.get_by_scope( - scope="scope", + scope="subscriptions/subId/resourceGroups/resourceGroupName/providers/Resource.Provider/resourceType/resourceName/identities/default", ) print(response) -# x-ms-original-file: specification/msi/resource-manager/Microsoft.ManagedIdentity/stable/2023-01-31/examples/SystemAssignedIdentityGet.json +# x-ms-original-file: 2025-05-31-preview/SystemAssignedIdentityGet.json if __name__ == "__main__": main() diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py b/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py index ee793b8ffac3..43fd9f113397 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py +++ b/sdk/resources/azure-mgmt-msi/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 @@ -21,20 +21,14 @@ # For security, please avoid record sensitive identity information in recordings @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): - managedserviceidentity_subscription_id = os.environ.get( - "AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - managedserviceidentity_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - managedserviceidentity_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - managedserviceidentity_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer( - regex=managedserviceidentity_subscription_id, value="00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=managedserviceidentity_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=managedserviceidentity_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer( - regex=managedserviceidentity_client_secret, value="00000000-0000-0000-0000-000000000000" - ) + managedidentity_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + managedidentity_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + managedidentity_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + managedidentity_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=managedidentity_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=managedidentity_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=managedidentity_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=managedidentity_client_secret, value="00000000-0000-0000-0000-000000000000") add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") add_header_regex_sanitizer(key="Cookie", value="cookie;") diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py similarity index 84% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py index c75cf38dc1ba..33aefe81b67e 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,19 +14,19 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityFederatedIdentityCredentialsOperations(AzureMgmtRecordedTestCase): +class TestManagedIdentityFederatedIdentityCredentialsOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_federated_identity_credentials_list(self, resource_group): - response = self.client.federated_identity_credentials.list( + def test_federated_identity_credentials_get(self, resource_group): + response = self.client.federated_identity_credentials.get( resource_group_name=resource_group.name, resource_name="str", - api_version="2023-01-31", + federated_identity_credential_resource_name="str", ) - result = [r for r in response] + # please add some check logic here by yourself # ... @@ -38,11 +38,14 @@ def test_federated_identity_credentials_create_or_update(self, resource_group): resource_name="str", federated_identity_credential_resource_name="str", parameters={ - "audiences": ["str"], "id": "str", - "issuer": "str", "name": "str", - "subject": "str", + "properties": { + "audiences": ["str"], + "issuer": "str", + "claimsMatchingExpression": {"languageVersion": 0, "value": "str"}, + "subject": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -53,7 +56,6 @@ def test_federated_identity_credentials_create_or_update(self, resource_group): }, "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -61,12 +63,11 @@ def test_federated_identity_credentials_create_or_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_federated_identity_credentials_get(self, resource_group): - response = self.client.federated_identity_credentials.get( + def test_federated_identity_credentials_delete(self, resource_group): + response = self.client.federated_identity_credentials.delete( resource_group_name=resource_group.name, resource_name="str", federated_identity_credential_resource_name="str", - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -74,13 +75,11 @@ def test_federated_identity_credentials_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_federated_identity_credentials_delete(self, resource_group): - response = self.client.federated_identity_credentials.delete( + def test_federated_identity_credentials_list(self, resource_group): + response = self.client.federated_identity_credentials.list( resource_group_name=resource_group.name, resource_name="str", - federated_identity_credential_resource_name="str", - api_version="2023-01-31", ) - + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py similarity index 84% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py index 3bc5c871f3ca..539ffb69ac20 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,19 +15,19 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityFederatedIdentityCredentialsOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedIdentityFederatedIdentityCredentialsOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_federated_identity_credentials_list(self, resource_group): - response = self.client.federated_identity_credentials.list( + async def test_federated_identity_credentials_get(self, resource_group): + response = await self.client.federated_identity_credentials.get( resource_group_name=resource_group.name, resource_name="str", - api_version="2023-01-31", + federated_identity_credential_resource_name="str", ) - result = [r async for r in response] + # please add some check logic here by yourself # ... @@ -39,11 +39,14 @@ async def test_federated_identity_credentials_create_or_update(self, resource_gr resource_name="str", federated_identity_credential_resource_name="str", parameters={ - "audiences": ["str"], "id": "str", - "issuer": "str", "name": "str", - "subject": "str", + "properties": { + "audiences": ["str"], + "issuer": "str", + "claimsMatchingExpression": {"languageVersion": 0, "value": "str"}, + "subject": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -54,7 +57,6 @@ async def test_federated_identity_credentials_create_or_update(self, resource_gr }, "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -62,12 +64,11 @@ async def test_federated_identity_credentials_create_or_update(self, resource_gr @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_federated_identity_credentials_get(self, resource_group): - response = await self.client.federated_identity_credentials.get( + async def test_federated_identity_credentials_delete(self, resource_group): + response = await self.client.federated_identity_credentials.delete( resource_group_name=resource_group.name, resource_name="str", federated_identity_credential_resource_name="str", - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -75,13 +76,11 @@ async def test_federated_identity_credentials_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_federated_identity_credentials_delete(self, resource_group): - response = await self.client.federated_identity_credentials.delete( + async def test_federated_identity_credentials_list(self, resource_group): + response = self.client.federated_identity_credentials.list( resource_group_name=resource_group.name, resource_name="str", - federated_identity_credential_resource_name="str", - api_version="2023-01-31", ) - + result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py similarity index 72% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py index 73f3d2ee91b2..469e25b6c5a0 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityOperations(AzureMgmtRecordedTestCase): +class TestManagedIdentityOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_operations_list(self, resource_group): - response = self.client.operations.list( - api_version="2023-01-31", - ) + response = self.client.operations.list() result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py similarity index 72% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py index c4284e6a8392..b38d583b7c14 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,14 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedIdentityOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async async def test_operations_list(self, resource_group): - response = self.client.operations.list( - api_version="2023-01-31", - ) + response = self.client.operations.list() result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py similarity index 75% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py index 23ec0f33f91b..c5818792f2b7 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,16 +14,15 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentitySystemAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): +class TestManagedIdentitySystemAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_system_assigned_identities_get_by_scope(self, resource_group): response = self.client.system_assigned_identities.get_by_scope( scope="str", - api_version="2023-01-31", ) # please add some check logic here by yourself diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py similarity index 75% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py index 69bdbc9a8b75..0058a041bd2b 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,16 +15,15 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentitySystemAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedIdentitySystemAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async async def test_system_assigned_identities_get_by_scope(self, resource_group): response = await self.client.system_assigned_identities.get_by_scope( scope="str", - api_version="2023-01-31", ) # please add some check logic here by yourself diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py similarity index 82% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py index 469d24e01d74..405a1917a8ea 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,28 +14,18 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): +class TestManagedIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_user_assigned_identities_list_by_subscription(self, resource_group): - response = self.client.user_assigned_identities.list_by_subscription( - api_version="2023-01-31", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_user_assigned_identities_list_by_resource_group(self, resource_group): - response = self.client.user_assigned_identities.list_by_resource_group( + def test_user_assigned_identities_get(self, resource_group): + response = self.client.user_assigned_identities.get( resource_group_name=resource_group.name, - api_version="2023-01-31", + resource_name="str", ) - result = [r for r in response] + # please add some check logic here by yourself # ... @@ -47,10 +37,15 @@ def test_user_assigned_identities_create_or_update(self, resource_group): resource_name="str", parameters={ "location": "str", - "clientId": "str", "id": "str", "name": "str", - "principalId": "str", + "properties": { + "assignmentRestrictions": {"providers": ["str"]}, + "clientId": "str", + "isolationScope": "str", + "principalId": "str", + "tenantId": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -60,10 +55,8 @@ def test_user_assigned_identities_create_or_update(self, resource_group): "lastModifiedByType": "str", }, "tags": {"str": "str"}, - "tenantId": "str", "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -76,11 +69,16 @@ def test_user_assigned_identities_update(self, resource_group): resource_group_name=resource_group.name, resource_name="str", parameters={ - "clientId": "str", "id": "str", "location": "str", "name": "str", - "principalId": "str", + "properties": { + "assignmentRestrictions": {"providers": ["str"]}, + "clientId": "str", + "isolationScope": "str", + "principalId": "str", + "tenantId": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -90,10 +88,8 @@ def test_user_assigned_identities_update(self, resource_group): "lastModifiedByType": "str", }, "tags": {"str": "str"}, - "tenantId": "str", "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -101,11 +97,10 @@ def test_user_assigned_identities_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_user_assigned_identities_get(self, resource_group): - response = self.client.user_assigned_identities.get( + def test_user_assigned_identities_delete(self, resource_group): + response = self.client.user_assigned_identities.delete( resource_group_name=resource_group.name, resource_name="str", - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -113,12 +108,18 @@ def test_user_assigned_identities_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_user_assigned_identities_delete(self, resource_group): - response = self.client.user_assigned_identities.delete( + def test_user_assigned_identities_list_by_resource_group(self, resource_group): + response = self.client.user_assigned_identities.list_by_resource_group( resource_group_name=resource_group.name, - resource_name="str", - api_version="2023-01-31", ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_user_assigned_identities_list_by_subscription(self, resource_group): + response = self.client.user_assigned_identities.list_by_subscription() + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py similarity index 82% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py index 889667740f71..b6ba8ec86b00 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py @@ -2,11 +2,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. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,28 +15,18 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedServiceIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_user_assigned_identities_list_by_subscription(self, resource_group): - response = self.client.user_assigned_identities.list_by_subscription( - api_version="2023-01-31", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_user_assigned_identities_list_by_resource_group(self, resource_group): - response = self.client.user_assigned_identities.list_by_resource_group( + async def test_user_assigned_identities_get(self, resource_group): + response = await self.client.user_assigned_identities.get( resource_group_name=resource_group.name, - api_version="2023-01-31", + resource_name="str", ) - result = [r async for r in response] + # please add some check logic here by yourself # ... @@ -48,10 +38,15 @@ async def test_user_assigned_identities_create_or_update(self, resource_group): resource_name="str", parameters={ "location": "str", - "clientId": "str", "id": "str", "name": "str", - "principalId": "str", + "properties": { + "assignmentRestrictions": {"providers": ["str"]}, + "clientId": "str", + "isolationScope": "str", + "principalId": "str", + "tenantId": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -61,10 +56,8 @@ async def test_user_assigned_identities_create_or_update(self, resource_group): "lastModifiedByType": "str", }, "tags": {"str": "str"}, - "tenantId": "str", "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -77,11 +70,16 @@ async def test_user_assigned_identities_update(self, resource_group): resource_group_name=resource_group.name, resource_name="str", parameters={ - "clientId": "str", "id": "str", "location": "str", "name": "str", - "principalId": "str", + "properties": { + "assignmentRestrictions": {"providers": ["str"]}, + "clientId": "str", + "isolationScope": "str", + "principalId": "str", + "tenantId": "str", + }, "systemData": { "createdAt": "2020-02-20 00:00:00", "createdBy": "str", @@ -91,10 +89,8 @@ async def test_user_assigned_identities_update(self, resource_group): "lastModifiedByType": "str", }, "tags": {"str": "str"}, - "tenantId": "str", "type": "str", }, - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -102,11 +98,10 @@ async def test_user_assigned_identities_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_user_assigned_identities_get(self, resource_group): - response = await self.client.user_assigned_identities.get( + async def test_user_assigned_identities_delete(self, resource_group): + response = await self.client.user_assigned_identities.delete( resource_group_name=resource_group.name, resource_name="str", - api_version="2023-01-31", ) # please add some check logic here by yourself @@ -114,12 +109,18 @@ async def test_user_assigned_identities_get(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_user_assigned_identities_delete(self, resource_group): - response = await self.client.user_assigned_identities.delete( + async def test_user_assigned_identities_list_by_resource_group(self, resource_group): + response = self.client.user_assigned_identities.list_by_resource_group( resource_group_name=resource_group.name, - resource_name="str", - api_version="2023-01-31", ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_user_assigned_identities_list_by_subscription(self, resource_group): + response = self.client.user_assigned_identities.list_by_subscription() + result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/resources/azure-mgmt-msi/pyproject.toml b/sdk/resources/azure-mgmt-msi/pyproject.toml index f855ce57093c..d192f2fb7a50 100644 --- a/sdk/resources/azure-mgmt-msi/pyproject.toml +++ b/sdk/resources/azure-mgmt-msi/pyproject.toml @@ -1,3 +1,71 @@ +[build-system] +requires = [ + "setuptools>=77.0.3", + "wheel", +] +build-backend = "setuptools.build_meta" + +[project] +name = "azure-mgmt-msi" +authors = [ + { name = "Microsoft Corporation", email = "azpysdkhelp@microsoft.com" }, +] +description = "Microsoft Azure Msi 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.10", + "Programming Language :: Python :: 3.11", + "Programming Language :: Python :: 3.12", + "Programming Language :: Python :: 3.13", +] +requires-python = ">=3.10" +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.msi._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", +] + +[tool.setuptools.package-data] +pytyped = [ + "py.typed", +] + [tool.azure-sdk-build] breaking = false mypy = false @@ -15,3 +83,5 @@ is_arm = true need_msrestazure = false need_azuremgmtcore = true title = "ManagedServiceIdentityClient" +sample_link = "" +exclude_folders = "" diff --git a/sdk/resources/azure-mgmt-msi/setup.py b/sdk/resources/azure-mgmt-msi/setup.py deleted file mode 100644 index e01101889e9a..000000000000 --- a/sdk/resources/azure-mgmt-msi/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-msi" -PACKAGE_PPRINT_NAME = "Msi 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 :: 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", - "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", - ] - ), - 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.6.0", - ], - python_requires=">=3.9", -) diff --git a/sdk/resources/azure-mgmt-msi/tsp-location.yaml b/sdk/resources/azure-mgmt-msi/tsp-location.yaml new file mode 100644 index 000000000000..dba1344a4d3e --- /dev/null +++ b/sdk/resources/azure-mgmt-msi/tsp-location.yaml @@ -0,0 +1,4 @@ +directory: specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity +commit: 7811c124444dfbd4bd149ad88f2648854dd9201b +repo: Azure/azure-rest-api-specs +additionalDirectories: From 58886db2bef79fe1335611c04e543057dc9eb17c Mon Sep 17 00:00:00 2001 From: Mohamed Yilmaz Ibrahim Date: Fri, 22 May 2026 11:26:08 -0700 Subject: [PATCH 2/6] Rename ManagedServiceIdentityClient -> ManagedIdentityClient in tests, README, and pyproject title --- sdk/resources/azure-mgmt-msi/README.md | 4 ++-- sdk/resources/azure-mgmt-msi/pyproject.toml | 2 +- .../test_managed_service_identity_operations_async_test.py | 4 ++-- .../tests/test_managed_service_identity_operations_test.py | 4 ++-- ...identity_user_assigned_identities_operations_async_test.py | 4 ++-- ...rvice_identity_user_assigned_identities_operations_test.py | 4 ++-- 6 files changed, 11 insertions(+), 11 deletions(-) diff --git a/sdk/resources/azure-mgmt-msi/README.md b/sdk/resources/azure-mgmt-msi/README.md index ea6c0c67db6d..e22dfc68003b 100644 --- a/sdk/resources/azure-mgmt-msi/README.md +++ b/sdk/resources/azure-mgmt-msi/README.md @@ -36,11 +36,11 @@ With above configuration, client can be authenticated by following code: ```python from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient import os sub_id = os.getenv("AZURE_SUBSCRIPTION_ID") -client = ManagedServiceIdentityClient(credential=DefaultAzureCredential(), subscription_id=sub_id) +client = ManagedIdentityClient(credential=DefaultAzureCredential(), subscription_id=sub_id) ``` ## Examples diff --git a/sdk/resources/azure-mgmt-msi/pyproject.toml b/sdk/resources/azure-mgmt-msi/pyproject.toml index d192f2fb7a50..fc608f53cd54 100644 --- a/sdk/resources/azure-mgmt-msi/pyproject.toml +++ b/sdk/resources/azure-mgmt-msi/pyproject.toml @@ -82,6 +82,6 @@ is_stable = false is_arm = true need_msrestazure = false need_azuremgmtcore = true -title = "ManagedServiceIdentityClient" +title = "ManagedIdentityClient" sample_link = "" exclude_folders = "" diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py index f0b5b239d3b0..7dc9bddb0aa7 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -17,7 +17,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py index f1711df3f4eb..79d42ee47045 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -16,7 +16,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py index fcf1086808ff..671454f122e5 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedServiceIdentityClient +from azure.mgmt.msi.aio import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -17,7 +17,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py index 36230145a31f..4fee7ce3f8ab 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedServiceIdentityClient +from azure.mgmt.msi import ManagedIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -16,7 +16,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedServiceIdentityClient) + self.client = self.create_mgmt_client(ManagedIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy From 984f1a56de28265d29143785bcdb193aa1126c4d Mon Sep 17 00:00:00 2001 From: "Chenxi Jiang (WICRESOFT NORTH AMERICA LTD)" Date: Mon, 25 May 2026 14:31:24 +0800 Subject: [PATCH 3/6] update --- sdk/resources/azure-mgmt-msi/CHANGELOG.md | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/sdk/resources/azure-mgmt-msi/CHANGELOG.md b/sdk/resources/azure-mgmt-msi/CHANGELOG.md index b7e0c8d65234..98c370aadd00 100644 --- a/sdk/resources/azure-mgmt-msi/CHANGELOG.md +++ b/sdk/resources/azure-mgmt-msi/CHANGELOG.md @@ -4,24 +4,19 @@ ### Features Added - - Model `Identity` added property `properties` - - Model `IdentityUpdate` added property `properties` - Added model `AssignmentRestrictions` - Added model `ClaimsMatchingExpression` - Added model `CloudError` - Added model `ExtensionResource` - Added enum `IsolationScope` - - Added model `UserAssignedIdentityProperties` ### 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. - Deleted or renamed client `ManagedServiceIdentityClient` - - Model `Identity` deleted or renamed its instance variable `tenant_id` - - Model `Identity` deleted or renamed its instance variable `principal_id` - - Model `Identity` deleted or renamed its instance variable `client_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `tenant_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `principal_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `client_id` + - Model `Identity` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` + - Model `IdentityUpdate` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` - Method `FederatedIdentityCredentialsOperations.list` changed its parameter `skiptoken` from `positional_or_keyword` to `keyword_only` ## 8.0.0b1 (2025-08-15) From fcdbc614a48652adc17978db53960fbbaeb5ed73 Mon Sep 17 00:00:00 2001 From: Yuchao Yan Date: Mon, 25 May 2026 16:24:36 +0800 Subject: [PATCH 4/6] Update pyproject.toml --- sdk/resources/azure-mgmt-msi/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/resources/azure-mgmt-msi/pyproject.toml b/sdk/resources/azure-mgmt-msi/pyproject.toml index fc608f53cd54..d192f2fb7a50 100644 --- a/sdk/resources/azure-mgmt-msi/pyproject.toml +++ b/sdk/resources/azure-mgmt-msi/pyproject.toml @@ -82,6 +82,6 @@ is_stable = false is_arm = true need_msrestazure = false need_azuremgmtcore = true -title = "ManagedIdentityClient" +title = "ManagedServiceIdentityClient" sample_link = "" exclude_folders = "" From de00a2a00eaf795b2e790f99d59c7071888937cf Mon Sep 17 00:00:00 2001 From: azure-sdk Date: Mon, 25 May 2026 08:37:23 +0000 Subject: [PATCH 5/6] Configurations: 'specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity/tspconfig.yaml', API Version: 2025-05-31-preview, SDK Release Type: beta, and CommitSHA: 'd309d869cbd0360d18e1787c1d931c68f7752b37' in SpecRepo: 'https://github.com/Azure/azure-rest-api-specs' Pipeline run: https://dev.azure.com/azure-sdk/internal/_build/results?buildId=6346556 Refer to https://eng.ms/docs/products/azure-developer-experience/develop/sdk-release/sdk-release-prerequisites to prepare for SDK release. --- sdk/resources/azure-mgmt-msi/CHANGELOG.md | 18 ++++++++----- sdk/resources/azure-mgmt-msi/README.md | 4 +-- sdk/resources/azure-mgmt-msi/_metadata.json | 2 +- .../azure-mgmt-msi/apiview-properties.json | 2 +- .../azure-mgmt-msi/azure/mgmt/msi/__init__.py | 4 +-- .../azure-mgmt-msi/azure/mgmt/msi/_client.py | 6 ++--- .../azure/mgmt/msi/_configuration.py | 4 +-- .../azure/mgmt/msi/aio/__init__.py | 4 +-- .../azure/mgmt/msi/aio/_client.py | 6 ++--- .../azure/mgmt/msi/aio/_configuration.py | 4 +-- .../mgmt/msi/aio/operations/_operations.py | 26 ++++++++++++------- .../azure/mgmt/msi/operations/_operations.py | 26 ++++++++++++------- .../federated_identity_credential_create.py | 4 +-- .../federated_identity_credential_delete.py | 4 +-- .../federated_identity_credential_get.py | 4 +-- .../federated_identity_credential_list.py | 4 +-- ...le_federated_identity_credential_create.py | 4 +-- ...xible_federated_identity_credential_get.py | 4 +-- .../generated_samples/identity_create.py | 4 +-- .../generated_samples/identity_delete.py | 4 +-- .../generated_samples/identity_get.py | 4 +-- .../identity_list_by_resource_group.py | 4 +-- .../identity_list_by_subscription.py | 4 +-- .../generated_samples/identity_update.py | 4 +-- .../generated_samples/msi_operations_list.py | 4 +-- .../system_assigned_identity_get.py | 4 +-- .../generated_tests/conftest.py | 22 ++++++++++------ ...erated_identity_credentials_operations.py} | 6 ++--- ..._identity_credentials_operations_async.py} | 6 ++--- ...st_managed_service_identity_operations.py} | 6 ++--- ...aged_service_identity_operations_async.py} | 6 ++--- ..._system_assigned_identities_operations.py} | 6 ++--- ...m_assigned_identities_operations_async.py} | 6 ++--- ...ty_user_assigned_identities_operations.py} | 6 ++--- ...r_assigned_identities_operations_async.py} | 6 ++--- .../azure-mgmt-msi/tsp-location.yaml | 2 +- 36 files changed, 131 insertions(+), 103 deletions(-) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_federated_identity_credentials_operations.py => test_managed_service_identity_federated_identity_credentials_operations.py} (93%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_federated_identity_credentials_operations_async.py => test_managed_service_identity_federated_identity_credentials_operations_async.py} (93%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_operations.py => test_managed_service_identity_operations.py} (83%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_operations_async.py => test_managed_service_identity_operations_async.py} (82%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_system_assigned_identities_operations.py => test_managed_service_identity_system_assigned_identities_operations.py} (82%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_system_assigned_identities_operations_async.py => test_managed_service_identity_system_assigned_identities_operations_async.py} (81%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_user_assigned_identities_operations.py => test_managed_service_identity_user_assigned_identities_operations.py} (95%) rename sdk/resources/azure-mgmt-msi/generated_tests/{test_managed_identity_user_assigned_identities_operations_async.py => test_managed_service_identity_user_assigned_identities_operations_async.py} (95%) diff --git a/sdk/resources/azure-mgmt-msi/CHANGELOG.md b/sdk/resources/azure-mgmt-msi/CHANGELOG.md index 98c370aadd00..7523f6390439 100644 --- a/sdk/resources/azure-mgmt-msi/CHANGELOG.md +++ b/sdk/resources/azure-mgmt-msi/CHANGELOG.md @@ -1,22 +1,28 @@ # Release History -## 8.0.0b2 (2026-05-22) +## 8.0.0b2 (2026-05-25) ### Features Added + - Client `ManagedServiceIdentityClient` added parameter `cloud_setting` in method `__init__` + - Client `ManagedServiceIdentityClient` added method `send_request` + - Model `Identity` added property `properties` + - Model `IdentityUpdate` added property `properties` - Added model `AssignmentRestrictions` - Added model `ClaimsMatchingExpression` - Added model `CloudError` - Added model `ExtensionResource` - Added enum `IsolationScope` + - Added model `UserAssignedIdentityProperties` ### 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. - - Deleted or renamed client `ManagedServiceIdentityClient` - - Model `Identity` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` - - Model `IdentityUpdate` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` + - Model `Identity` deleted or renamed its instance variable `tenant_id` + - Model `Identity` deleted or renamed its instance variable `principal_id` + - Model `Identity` deleted or renamed its instance variable `client_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `tenant_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `principal_id` + - Model `IdentityUpdate` deleted or renamed its instance variable `client_id` - Method `FederatedIdentityCredentialsOperations.list` changed its parameter `skiptoken` from `positional_or_keyword` to `keyword_only` ## 8.0.0b1 (2025-08-15) diff --git a/sdk/resources/azure-mgmt-msi/README.md b/sdk/resources/azure-mgmt-msi/README.md index e22dfc68003b..ea6c0c67db6d 100644 --- a/sdk/resources/azure-mgmt-msi/README.md +++ b/sdk/resources/azure-mgmt-msi/README.md @@ -36,11 +36,11 @@ With above configuration, client can be authenticated by following code: ```python from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient import os sub_id = os.getenv("AZURE_SUBSCRIPTION_ID") -client = ManagedIdentityClient(credential=DefaultAzureCredential(), subscription_id=sub_id) +client = ManagedServiceIdentityClient(credential=DefaultAzureCredential(), subscription_id=sub_id) ``` ## Examples diff --git a/sdk/resources/azure-mgmt-msi/_metadata.json b/sdk/resources/azure-mgmt-msi/_metadata.json index f85dbb69c577..cd60b1114622 100644 --- a/sdk/resources/azure-mgmt-msi/_metadata.json +++ b/sdk/resources/azure-mgmt-msi/_metadata.json @@ -3,7 +3,7 @@ "apiVersions": { "Microsoft.ManagedIdentity": "2025-05-31-preview" }, - "commit": "7811c124444dfbd4bd149ad88f2648854dd9201b", + "commit": "d309d869cbd0360d18e1787c1d931c68f7752b37", "repository_url": "https://github.com/Azure/azure-rest-api-specs", "typespec_src": "specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity", "emitterVersion": "0.62.1" diff --git a/sdk/resources/azure-mgmt-msi/apiview-properties.json b/sdk/resources/azure-mgmt-msi/apiview-properties.json index 609eb95b6af8..3914faee637c 100644 --- a/sdk/resources/azure-mgmt-msi/apiview-properties.json +++ b/sdk/resources/azure-mgmt-msi/apiview-properties.json @@ -46,5 +46,5 @@ "azure.mgmt.msi.operations.UserAssignedIdentitiesOperations.list_by_subscription": "Microsoft.ManagedIdentity.Identities.listBySubscription", "azure.mgmt.msi.aio.operations.UserAssignedIdentitiesOperations.list_by_subscription": "Microsoft.ManagedIdentity.Identities.listBySubscription" }, - "CrossLanguageVersion": "54a26e7e73ed" + "CrossLanguageVersion": "75700dd7a51c" } \ No newline at end of file diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py index ca1ad10c23b4..f6797c20a230 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/__init__.py @@ -12,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._client import ManagedIdentityClient # type: ignore +from ._client import ManagedServiceIdentityClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -25,7 +25,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "ManagedIdentityClient", + "ManagedServiceIdentityClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py index 4acebf4f79b6..cd9dc23f1361 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_client.py @@ -17,7 +17,7 @@ from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy from azure.mgmt.core.tools import get_arm_endpoints -from ._configuration import ManagedIdentityClientConfiguration +from ._configuration import ManagedServiceIdentityClientConfiguration from ._utils.serialization import Deserializer, Serializer from .operations import ( FederatedIdentityCredentialsOperations, @@ -36,7 +36,7 @@ from azure.core.credentials import TokenCredential -class ManagedIdentityClient: +class ManagedServiceIdentityClient: """The Managed Service Identity Client. :ivar operations: Operations operations @@ -80,7 +80,7 @@ def __init__( if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) - self._config = ManagedIdentityClientConfiguration( + self._config = ManagedServiceIdentityClientConfiguration( credential=credential, subscription_id=subscription_id, base_url=cast(str, base_url), diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py index c9851865f4db..d26c2f6cf738 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/_configuration.py @@ -18,8 +18,8 @@ from azure.core.credentials import TokenCredential -class ManagedIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for ManagedIdentityClient. +class ManagedServiceIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + """Configuration for ManagedServiceIdentityClient. Note that all parameters used to create this instance are saved as instance attributes. diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py index afab46398e53..cb5991f17203 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/__init__.py @@ -12,7 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import -from ._client import ManagedIdentityClient # type: ignore +from ._client import ManagedServiceIdentityClient # type: ignore try: from ._patch import __all__ as _patch_all @@ -22,7 +22,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "ManagedIdentityClient", + "ManagedServiceIdentityClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py index b35a84f661cc..64f417d32caf 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_client.py @@ -18,7 +18,7 @@ from azure.mgmt.core.tools import get_arm_endpoints from .._utils.serialization import Deserializer, Serializer -from ._configuration import ManagedIdentityClientConfiguration +from ._configuration import ManagedServiceIdentityClientConfiguration from .operations import ( FederatedIdentityCredentialsOperations, Operations, @@ -36,7 +36,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ManagedIdentityClient: +class ManagedServiceIdentityClient: """The Managed Service Identity Client. :ivar operations: Operations operations @@ -81,7 +81,7 @@ def __init__( if not base_url: base_url = _endpoints["resource_manager"] credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"]) - self._config = ManagedIdentityClientConfiguration( + self._config = ManagedServiceIdentityClientConfiguration( credential=credential, subscription_id=subscription_id, base_url=cast(str, base_url), diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py index 846202bf526f..33709ae3dca5 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/_configuration.py @@ -18,8 +18,8 @@ from azure.core.credentials_async import AsyncTokenCredential -class ManagedIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for ManagedIdentityClient. +class ManagedServiceIdentityClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + """Configuration for ManagedServiceIdentityClient. Note that all parameters used to create this instance are saved as instance attributes. diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py index e349e5018170..215bac782217 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/aio/operations/_operations.py @@ -48,7 +48,7 @@ build_user_assigned_identities_list_by_subscription_request, build_user_assigned_identities_update_request, ) -from .._configuration import ManagedIdentityClientConfiguration +from .._configuration import ManagedServiceIdentityClientConfiguration T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, dict[str, Any]], Any]] @@ -62,14 +62,16 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s :attr:`operations` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -172,14 +174,16 @@ class SystemAssignedIdentitiesOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s :attr:`system_assigned_identities` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -255,14 +259,16 @@ class FederatedIdentityCredentialsOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s :attr:`federated_identity_credentials` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -730,14 +736,16 @@ class UserAssignedIdentitiesOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.aio.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.aio.ManagedServiceIdentityClient`'s :attr:`user_assigned_identities` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") diff --git a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py index 2d42ba65f04c..fd5220262831 100644 --- a/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py +++ b/sdk/resources/azure-mgmt-msi/azure/mgmt/msi/operations/_operations.py @@ -31,7 +31,7 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models -from .._configuration import ManagedIdentityClientConfiguration +from .._configuration import ManagedServiceIdentityClientConfiguration from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._utils.serialization import Deserializer, Serializer @@ -400,14 +400,16 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s :attr:`operations` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -510,14 +512,16 @@ class SystemAssignedIdentitiesOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s :attr:`system_assigned_identities` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -593,14 +597,16 @@ class FederatedIdentityCredentialsOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s :attr:`federated_identity_credentials` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") @@ -1067,14 +1073,16 @@ class UserAssignedIdentitiesOperations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.msi.ManagedIdentityClient`'s + :class:`~azure.mgmt.msi.ManagedServiceIdentityClient`'s :attr:`user_assigned_identities` 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: ManagedIdentityClientConfiguration = input_args.pop(0) if input_args else kwargs.pop("config") + self._config: ManagedServiceIdentityClientConfiguration = ( + 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") diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py index 123c76388fda..f4e52940a583 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_create.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py index 241d1188b0ac..e29e9362c82a 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_delete.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py index b1ae8f5cfb65..1524b351f350 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_get.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py index 4d02f87efebd..fbbe2e936fa5 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/federated_identity_credential_list.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py index 31cbd208ffd9..25dc4082345f 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_create.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py index fdb5dd839d5c..ee79c070c1cf 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/flexible_federated_identity_credential_get.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py index 691639e1af1b..d9904e60fdfa 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_create.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py index 54a3729bc049..a0c95b7010c8 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_delete.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py index b16d5da9b55b..398797fd30a5 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_get.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py index b7ad9c2a95cb..6ba6c46d55bd 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_resource_group.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py index f55070ae77b6..f07d394970f3 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_list_by_subscription.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py b/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py index d972a87b5997..4f8d78b77ceb 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/identity_update.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py b/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py index 07f0871c1e51..59f9905244b9 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/msi_operations_list.py @@ -8,7 +8,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -25,7 +25,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py b/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py index 15963c81e63f..b09f62e209de 100644 --- a/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py +++ b/sdk/resources/azure-mgmt-msi/generated_samples/system_assigned_identity_get.py @@ -9,7 +9,7 @@ from azure.identity import DefaultAzureCredential -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient """ # PREREQUISITES @@ -26,7 +26,7 @@ def main(): - client = ManagedIdentityClient( + client = ManagedServiceIdentityClient( credential=DefaultAzureCredential(), subscription_id="SUBSCRIPTION_ID", ) diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py b/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py index 43fd9f113397..7dc1a4a0fb5a 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/conftest.py @@ -21,14 +21,20 @@ # For security, please avoid record sensitive identity information in recordings @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): - managedidentity_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - managedidentity_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - managedidentity_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - managedidentity_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=managedidentity_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=managedidentity_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=managedidentity_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=managedidentity_client_secret, value="00000000-0000-0000-0000-000000000000") + managedserviceidentity_subscription_id = os.environ.get( + "AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + managedserviceidentity_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + managedserviceidentity_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + managedserviceidentity_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer( + regex=managedserviceidentity_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=managedserviceidentity_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=managedserviceidentity_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer( + regex=managedserviceidentity_client_secret, value="00000000-0000-0000-0000-000000000000" + ) add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") add_header_regex_sanitizer(key="Cookie", value="cookie;") diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py similarity index 93% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py index 33aefe81b67e..5604028e5755 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,9 +14,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityFederatedIdentityCredentialsOperations(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityFederatedIdentityCredentialsOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py similarity index 93% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py index 539ffb69ac20..873be066c6d9 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_federated_identity_credentials_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_federated_identity_credentials_operations_async.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,9 +15,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityFederatedIdentityCredentialsOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityFederatedIdentityCredentialsOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py similarity index 83% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py index 469e25b6c5a0..458db0bdc80d 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,9 +14,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityOperations(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py similarity index 82% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py index b38d583b7c14..81468b5c987c 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_operations_async.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,9 +15,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py similarity index 82% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py index c5818792f2b7..c3088962ecaa 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,9 +14,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentitySystemAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentitySystemAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py similarity index 81% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py index 0058a041bd2b..45b2ef20704e 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_system_assigned_identities_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_system_assigned_identities_operations_async.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,9 +15,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentitySystemAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentitySystemAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py similarity index 95% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py index 405a1917a8ea..661213ee0880 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,9 +14,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py similarity index 95% rename from sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py rename to sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py index b6ba8ec86b00..9d2a981cb9f3 100644 --- a/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_identity_user_assigned_identities_operations_async.py +++ b/sdk/resources/azure-mgmt-msi/generated_tests/test_managed_service_identity_user_assigned_identities_operations_async.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,9 +15,9 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestManagedIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): +class TestManagedServiceIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/tsp-location.yaml b/sdk/resources/azure-mgmt-msi/tsp-location.yaml index dba1344a4d3e..d7fcca9507af 100644 --- a/sdk/resources/azure-mgmt-msi/tsp-location.yaml +++ b/sdk/resources/azure-mgmt-msi/tsp-location.yaml @@ -1,4 +1,4 @@ directory: specification/msi/resource-manager/Microsoft.ManagedIdentity/ManagedIdentity -commit: 7811c124444dfbd4bd149ad88f2648854dd9201b +commit: d309d869cbd0360d18e1787c1d931c68f7752b37 repo: Azure/azure-rest-api-specs additionalDirectories: From 12fdd96319fe36b53ca36aac34a7631b99855a7a Mon Sep 17 00:00:00 2001 From: "Chenxi Jiang (WICRESOFT NORTH AMERICA LTD)" Date: Mon, 25 May 2026 17:25:01 +0800 Subject: [PATCH 6/6] Fix client name in tests and optimize CHANGELOG - Fix ManagedIdentityClient -> ManagedServiceIdentityClient in test files - Optimize CHANGELOG: consolidate moved properties, add migration notes Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --- sdk/resources/azure-mgmt-msi/CHANGELOG.md | 13 ++++--------- ...anaged_service_identity_operations_async_test.py | 4 ++-- ...test_managed_service_identity_operations_test.py | 4 ++-- ...ser_assigned_identities_operations_async_test.py | 4 ++-- ...tity_user_assigned_identities_operations_test.py | 4 ++-- 5 files changed, 12 insertions(+), 17 deletions(-) diff --git a/sdk/resources/azure-mgmt-msi/CHANGELOG.md b/sdk/resources/azure-mgmt-msi/CHANGELOG.md index 7523f6390439..b6df5b1a5b57 100644 --- a/sdk/resources/azure-mgmt-msi/CHANGELOG.md +++ b/sdk/resources/azure-mgmt-msi/CHANGELOG.md @@ -6,23 +6,18 @@ - Client `ManagedServiceIdentityClient` added parameter `cloud_setting` in method `__init__` - Client `ManagedServiceIdentityClient` added method `send_request` - - Model `Identity` added property `properties` - - Model `IdentityUpdate` added property `properties` - Added model `AssignmentRestrictions` - Added model `ClaimsMatchingExpression` - Added model `CloudError` - Added model `ExtensionResource` - Added enum `IsolationScope` - - Added model `UserAssignedIdentityProperties` ### Breaking Changes - - Model `Identity` deleted or renamed its instance variable `tenant_id` - - Model `Identity` deleted or renamed its instance variable `principal_id` - - Model `Identity` deleted or renamed its instance variable `client_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `tenant_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `principal_id` - - Model `IdentityUpdate` deleted or renamed its instance variable `client_id` + - 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 `Identity` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` + - Model `IdentityUpdate` moved instance variable `tenant_id`, `principal_id` and `client_id` under property `properties` whose type is `UserAssignedIdentityProperties` - Method `FederatedIdentityCredentialsOperations.list` changed its parameter `skiptoken` from `positional_or_keyword` to `keyword_only` ## 8.0.0b1 (2025-08-15) diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py index 7dc9bddb0aa7..f0b5b239d3b0 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_async_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -17,7 +17,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py index 79d42ee47045..f1711df3f4eb 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_operations_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -16,7 +16,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py index 671454f122e5..fcf1086808ff 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_async_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi.aio import ManagedIdentityClient +from azure.mgmt.msi.aio import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -17,7 +17,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityUserAssignedIdentitiesOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient, is_async=True) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient, is_async=True) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async diff --git a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py index 4fee7ce3f8ab..36230145a31f 100644 --- a/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py +++ b/sdk/resources/azure-mgmt-msi/tests/test_managed_service_identity_user_assigned_identities_operations_test.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import pytest -from azure.mgmt.msi import ManagedIdentityClient +from azure.mgmt.msi import ManagedServiceIdentityClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -16,7 +16,7 @@ @pytest.mark.live_test_only class TestManagedServiceIdentityUserAssignedIdentitiesOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(ManagedIdentityClient) + self.client = self.create_mgmt_client(ManagedServiceIdentityClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy