From 94c1d95a31572089b3c1b500ef936042cd86650e Mon Sep 17 00:00:00 2001 From: SDKAuto Date: Fri, 9 May 2025 08:08:11 +0000 Subject: [PATCH] CodeGen from PR 34150 in Azure/azure-rest-api-specs Merge 3b8d68d4f2ef8370b4aa0cd9b3798fcb33cec69d into 761d8b58e465b26eddb8ad70dda3cde7b4228bdf --- .../azure-mgmt-storageactions/CHANGELOG.md | 4 + .../azure-mgmt-storageactions/README.md | 4 +- .../azure-mgmt-storageactions/_meta.json | 11 +- .../apiview-properties.json | 65 + .../azure/mgmt/storageactions/__init__.py | 6 +- ...rage_actions_mgmt_client.py => _client.py} | 80 +- .../mgmt/storageactions/_configuration.py | 23 +- .../azure/mgmt/storageactions/_patch.py | 9 +- .../mgmt/storageactions/_utils/__init__.py | 6 + .../mgmt/storageactions/_utils/model_base.py | 1232 +++++++++++++ .../serialization.py} | 178 +- .../azure/mgmt/storageactions/_version.py | 4 +- .../azure/mgmt/storageactions/aio/__init__.py | 6 +- ...rage_actions_mgmt_client.py => _client.py} | 82 +- .../mgmt/storageactions/aio/_configuration.py | 23 +- .../azure/mgmt/storageactions/aio/_patch.py | 9 +- .../storageactions/aio/operations/__init__.py | 10 +- .../aio/operations/_operations.py | 1310 +++++++++++++- .../storageactions/aio/operations/_patch.py | 9 +- .../_storage_task_assignment_operations.py | 147 -- .../operations/_storage_tasks_operations.py | 930 ---------- .../_storage_tasks_report_operations.py | 156 -- .../mgmt/storageactions/models/__init__.py | 20 +- ...actions_mgmt_client_enums.py => _enums.py} | 34 +- .../mgmt/storageactions/models/_models.py | 1297 ++++++++++++++ .../mgmt/storageactions/models/_models_py3.py | 1368 -------------- .../mgmt/storageactions/models/_patch.py | 9 +- .../storageactions/operations/__init__.py | 10 +- .../storageactions/operations/_operations.py | 1572 ++++++++++++++++- .../mgmt/storageactions/operations/_patch.py | 9 +- .../_storage_task_assignment_operations.py | 191 -- .../operations/_storage_tasks_operations.py | 1152 ------------ .../_storage_tasks_report_operations.py | 204 --- .../generated_samples/misc/operations_list.py | 40 - .../perform_storage_task_actions_preview.py | 87 - .../storage_tasks_crud/delete_storage_task.py | 41 - .../storage_tasks_crud/get_storage_task.py | 42 - .../storage_tasks_crud/patch_storage_task.py | 68 - .../storage_tasks_crud/put_storage_task.py | 64 - .../list_storage_task_assignment_ids.py | 43 - .../list_storage_tasks_by_resource_group.py | 42 - .../list_storage_tasks_by_subscription.py | 40 - .../list_storage_tasks_run_report_summary.py | 43 - .../generated_tests/conftest.py | 18 +- ...mgmt_storage_task_assignment_operations.py | 31 - ...torage_task_assignment_operations_async.py | 32 - ...ns_mgmt_storage_tasks_report_operations.py | 31 - ...t_storage_tasks_report_operations_async.py | 32 - ....py => test_storage_actions_operations.py} | 12 +- ... test_storage_actions_operations_async.py} | 12 +- ...torage_tasks_operation_group_operations.py | 44 + ..._tasks_operation_group_operations_async.py | 45 + ...orage_actions_storage_tasks_operations.py} | 129 +- ...actions_storage_tasks_operations_async.py} | 121 +- .../sdk_packaging.toml | 2 +- .../azure-mgmt-storageactions/setup.py | 7 +- .../tsp-location.yaml | 4 + 57 files changed, 5891 insertions(+), 5309 deletions(-) create mode 100644 sdk/storageactions/azure-mgmt-storageactions/apiview-properties.json rename sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/{_storage_actions_mgmt_client.py => _client.py} (63%) create mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/__init__.py create mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/model_base.py rename sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/{_serialization.py => _utils/serialization.py} (94%) rename sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/{_storage_actions_mgmt_client.py => _client.py} (65%) delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_task_assignment_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_report_operations.py rename sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/{_storage_actions_mgmt_client_enums.py => _enums.py} (70%) create mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models_py3.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_task_assignment_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_report_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/operations_list.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/perform_storage_task_actions_preview.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/delete_storage_task.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/get_storage_task.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/patch_storage_task.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/put_storage_task.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_task_assignment_ids.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_resource_group.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_subscription.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_run_report_summary.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations_async.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations.py delete mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations_async.py rename sdk/storageactions/azure-mgmt-storageactions/generated_tests/{test_storage_actions_mgmt_operations.py => test_storage_actions_operations.py} (72%) rename sdk/storageactions/azure-mgmt-storageactions/generated_tests/{test_storage_actions_mgmt_operations_async.py => test_storage_actions_operations_async.py} (72%) create mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations.py create mode 100644 sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations_async.py rename sdk/storageactions/azure-mgmt-storageactions/generated_tests/{test_storage_actions_mgmt_storage_tasks_operations.py => test_storage_actions_storage_tasks_operations.py} (68%) rename sdk/storageactions/azure-mgmt-storageactions/generated_tests/{test_storage_actions_mgmt_storage_tasks_operations_async.py => test_storage_actions_storage_tasks_operations_async.py} (78%) create mode 100644 sdk/storageactions/azure-mgmt-storageactions/tsp-location.yaml diff --git a/sdk/storageactions/azure-mgmt-storageactions/CHANGELOG.md b/sdk/storageactions/azure-mgmt-storageactions/CHANGELOG.md index 3ae444fc617b..73156bcaad06 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/CHANGELOG.md +++ b/sdk/storageactions/azure-mgmt-storageactions/CHANGELOG.md @@ -1,5 +1,9 @@ # Release History +## 1.0.0 (2025-05-09) + +change log generation failed!!! + ## 1.0.0b2 (2025-04-20) ### Features Added diff --git a/sdk/storageactions/azure-mgmt-storageactions/README.md b/sdk/storageactions/azure-mgmt-storageactions/README.md index d78ac66e3045..29b3be628a60 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/README.md +++ b/sdk/storageactions/azure-mgmt-storageactions/README.md @@ -1,7 +1,7 @@ # Microsoft Azure SDK for Python This is the Microsoft Azure Storageactions Management Client Library. -This package has been tested with Python 3.8+. +This package has been tested with Python 3.9+. 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.8+ is required to use this package. +- Python 3.9+ is required to use this package. - [Azure subscription](https://azure.microsoft.com/free/) ### Install the package diff --git a/sdk/storageactions/azure-mgmt-storageactions/_meta.json b/sdk/storageactions/azure-mgmt-storageactions/_meta.json index 976ec339a2e0..946b7cfb1a2b 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/_meta.json +++ b/sdk/storageactions/azure-mgmt-storageactions/_meta.json @@ -1,11 +1,6 @@ { - "commit": "27046dbff974e3901970aa53b29cec6d8ec1342a", + "commit": "b3cda8e13ff3d36e63ea936afa4b6d7fa2cd731f", "repository_url": "https://github.com/Azure/azure-rest-api-specs", - "autorest": "3.10.2", - "use": [ - "@autorest/python@6.27.4", - "@autorest/modelerfour@4.27.0" - ], - "autorest_command": "autorest specification/storageactions/resource-manager/readme.md --generate-sample=True --generate-test=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.27.4 --use=@autorest/modelerfour@4.27.0 --version=3.10.2 --version-tolerant=False", - "readme": "specification/storageactions/resource-manager/readme.md" + "typespec_src": "specification/storageactions/StorageAction.Management", + "@azure-tools/typespec-python": "0.44.2" } \ No newline at end of file diff --git a/sdk/storageactions/azure-mgmt-storageactions/apiview-properties.json b/sdk/storageactions/azure-mgmt-storageactions/apiview-properties.json new file mode 100644 index 000000000000..ee932a4b048b --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/apiview-properties.json @@ -0,0 +1,65 @@ +{ + "CrossLanguagePackageId": "Microsoft.StorageActions", + "CrossLanguageDefinitionId": { + "azure.mgmt.storageactions.models.ElseCondition": "Microsoft.StorageActions.ElseCondition", + "azure.mgmt.storageactions.models.ErrorAdditionalInfo": "Azure.ResourceManager.CommonTypes.ErrorAdditionalInfo", + "azure.mgmt.storageactions.models.ErrorDetail": "Azure.ResourceManager.CommonTypes.ErrorDetail", + "azure.mgmt.storageactions.models.ErrorResponse": "Azure.ResourceManager.CommonTypes.ErrorResponse", + "azure.mgmt.storageactions.models.IfCondition": "Microsoft.StorageActions.IfCondition", + "azure.mgmt.storageactions.models.ManagedServiceIdentity": "Azure.ResourceManager.CommonTypes.ManagedServiceIdentity", + "azure.mgmt.storageactions.models.Operation": "Azure.ResourceManager.CommonTypes.Operation", + "azure.mgmt.storageactions.models.OperationDisplay": "Azure.ResourceManager.CommonTypes.OperationDisplay", + "azure.mgmt.storageactions.models.Resource": "Azure.ResourceManager.CommonTypes.Resource", + "azure.mgmt.storageactions.models.ProxyResource": "Azure.ResourceManager.CommonTypes.ProxyResource", + "azure.mgmt.storageactions.models.TrackedResource": "Azure.ResourceManager.CommonTypes.TrackedResource", + "azure.mgmt.storageactions.models.StorageTask": "Microsoft.StorageActions.StorageTask", + "azure.mgmt.storageactions.models.StorageTaskAction": "Microsoft.StorageActions.StorageTaskAction", + "azure.mgmt.storageactions.models.StorageTaskAssignment": "Microsoft.StorageActions.StorageTaskAssignment", + "azure.mgmt.storageactions.models.StorageTaskOperation": "Microsoft.StorageActions.StorageTaskOperation", + "azure.mgmt.storageactions.models.StorageTaskPreviewAction": "Microsoft.StorageActions.StorageTaskPreviewAction", + "azure.mgmt.storageactions.models.StorageTaskPreviewActionCondition": "Microsoft.StorageActions.StorageTaskPreviewActionCondition", + "azure.mgmt.storageactions.models.StorageTaskPreviewActionIfCondition": "Microsoft.StorageActions.StorageTaskPreviewActionIfCondition", + "azure.mgmt.storageactions.models.StorageTaskPreviewActionProperties": "Microsoft.StorageActions.StorageTaskPreviewActionProperties", + "azure.mgmt.storageactions.models.StorageTaskPreviewBlobProperties": "Microsoft.StorageActions.StorageTaskPreviewBlobProperties", + "azure.mgmt.storageactions.models.StorageTaskPreviewContainerProperties": "Microsoft.StorageActions.StorageTaskPreviewContainerProperties", + "azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties": "Microsoft.StorageActions.StorageTaskPreviewKeyValueProperties", + "azure.mgmt.storageactions.models.StorageTaskProperties": "Microsoft.StorageActions.StorageTaskProperties", + "azure.mgmt.storageactions.models.StorageTaskReportInstance": "Microsoft.StorageActions.StorageTaskReportInstance", + "azure.mgmt.storageactions.models.StorageTaskReportProperties": "Microsoft.StorageActions.StorageTaskReportProperties", + "azure.mgmt.storageactions.models.StorageTaskUpdateParameters": "Microsoft.StorageActions.StorageTaskUpdateParameters", + "azure.mgmt.storageactions.models.StorageTaskUpdateProperties": "Microsoft.StorageActions.StorageTaskUpdateProperties", + "azure.mgmt.storageactions.models.SystemData": "Azure.ResourceManager.CommonTypes.SystemData", + "azure.mgmt.storageactions.models.UserAssignedIdentity": "Azure.ResourceManager.CommonTypes.UserAssignedIdentity", + "azure.mgmt.storageactions.models.Origin": "Azure.ResourceManager.CommonTypes.Origin", + "azure.mgmt.storageactions.models.ActionType": "Azure.ResourceManager.CommonTypes.ActionType", + "azure.mgmt.storageactions.models.CreatedByType": "Azure.ResourceManager.CommonTypes.createdByType", + "azure.mgmt.storageactions.models.ManagedServiceIdentityType": "Azure.ResourceManager.CommonTypes.ManagedServiceIdentityType", + "azure.mgmt.storageactions.models.StorageTaskOperationName": "Microsoft.StorageActions.StorageTaskOperationName", + "azure.mgmt.storageactions.models.OnSuccess": "Microsoft.StorageActions.OnSuccess", + "azure.mgmt.storageactions.models.OnFailure": "Microsoft.StorageActions.OnFailure", + "azure.mgmt.storageactions.models.ProvisioningState": "Microsoft.StorageActions.ProvisioningState", + "azure.mgmt.storageactions.models.RunStatusEnum": "Microsoft.StorageActions.RunStatusEnum", + "azure.mgmt.storageactions.models.RunResult": "Microsoft.StorageActions.RunResult", + "azure.mgmt.storageactions.models.MatchedBlockName": "Microsoft.StorageActions.MatchedBlockName", + "azure.mgmt.storageactions.operations.Operations.list": "Azure.ResourceManager.Operations.list", + "azure.mgmt.storageactions.aio.operations.Operations.list": "Azure.ResourceManager.Operations.list", + "azure.mgmt.storageactions.operations.StorageTasksOperations.get": "Microsoft.StorageActions.StorageTasks.get", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.get": "Microsoft.StorageActions.StorageTasks.get", + "azure.mgmt.storageactions.operations.StorageTasksOperations.begin_create": "Microsoft.StorageActions.StorageTasks.create", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.begin_create": "Microsoft.StorageActions.StorageTasks.create", + "azure.mgmt.storageactions.operations.StorageTasksOperations.begin_update": "Microsoft.StorageActions.StorageTasks.update", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.begin_update": "Microsoft.StorageActions.StorageTasks.update", + "azure.mgmt.storageactions.operations.StorageTasksOperations.begin_delete": "Microsoft.StorageActions.StorageTasks.delete", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.begin_delete": "Microsoft.StorageActions.StorageTasks.delete", + "azure.mgmt.storageactions.operations.StorageTasksOperations.list_by_resource_group": "Microsoft.StorageActions.StorageTasks.listByResourceGroup", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.list_by_resource_group": "Microsoft.StorageActions.StorageTasks.listByResourceGroup", + "azure.mgmt.storageactions.operations.StorageTasksOperations.list_by_subscription": "Microsoft.StorageActions.StorageTasks.listBySubscription", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.list_by_subscription": "Microsoft.StorageActions.StorageTasks.listBySubscription", + "azure.mgmt.storageactions.operations.StorageTasksOperations.list": "Microsoft.StorageActions.StorageTasks.list", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.list": "Microsoft.StorageActions.StorageTasks.list", + "azure.mgmt.storageactions.operations.StorageTasksOperations.storage_task_assignment_list": "Microsoft.StorageActions.StorageTasks.storageTaskAssignmentList", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperations.storage_task_assignment_list": "Microsoft.StorageActions.StorageTasks.storageTaskAssignmentList", + "azure.mgmt.storageactions.operations.StorageTasksOperationGroupOperations.preview_actions": "Microsoft.StorageActions.StorageTasksOperationGroup.previewActions", + "azure.mgmt.storageactions.aio.operations.StorageTasksOperationGroupOperations.preview_actions": "Microsoft.StorageActions.StorageTasksOperationGroup.previewActions" + } +} \ No newline at end of file diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/__init__.py index c1d157ae9feb..fb42652d84bb 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/__init__.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/__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 ._storage_actions_mgmt_client import StorageActionsMgmtClient # type: ignore +from ._client import StorageActionsClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -25,7 +25,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "StorageActionsMgmtClient", + "StorageActionsClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_storage_actions_mgmt_client.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_client.py similarity index 63% rename from sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_storage_actions_mgmt_client.py rename to sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_client.py index 976330af584f..5bd398c6a24e 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_storage_actions_mgmt_client.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_client.py @@ -2,69 +2,69 @@ # -------------------------------------------------------------------------- # 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 -from typing import Any, TYPE_CHECKING +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 +from azure.core.settings import settings from azure.mgmt.core import ARMPipelineClient from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy +from azure.mgmt.core.tools import get_arm_endpoints -from . import models as _models -from ._configuration import StorageActionsMgmtClientConfiguration -from ._serialization import Deserializer, Serializer -from .operations import ( - Operations, - StorageTaskAssignmentOperations, - StorageTasksOperations, - StorageTasksReportOperations, -) +from ._configuration import StorageActionsClientConfiguration +from ._utils.serialization import Deserializer, Serializer +from .operations import Operations, StorageTasksOperationGroupOperations, StorageTasksOperations if TYPE_CHECKING: from azure.core.credentials import TokenCredential -class StorageActionsMgmtClient: +class StorageActionsClient: """The Azure Storage Actions Management API. :ivar operations: Operations operations :vartype operations: azure.mgmt.storageactions.operations.Operations :ivar storage_tasks: StorageTasksOperations operations :vartype storage_tasks: azure.mgmt.storageactions.operations.StorageTasksOperations - :ivar storage_task_assignment: StorageTaskAssignmentOperations operations - :vartype storage_task_assignment: - azure.mgmt.storageactions.operations.StorageTaskAssignmentOperations - :ivar storage_tasks_report: StorageTasksReportOperations operations - :vartype storage_tasks_report: - azure.mgmt.storageactions.operations.StorageTasksReportOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar storage_tasks_operation_group: StorageTasksOperationGroupOperations operations + :vartype storage_tasks_operation_group: + azure.mgmt.storageactions.operations.StorageTasksOperationGroupOperations + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is "https://management.azure.com". + :param base_url: Service host. Default value is None. :type base_url: str - :keyword api_version: Api Version. Default value is "2023-01-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Default value is "2023-01-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. """ def __init__( - self, - credential: "TokenCredential", - subscription_id: str, - base_url: str = "https://management.azure.com", - **kwargs: Any + self, credential: "TokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any ) -> None: - self._config = StorageActionsMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, **kwargs + _endpoint = "{endpoint}" + _cloud = kwargs.pop("cloud_setting", None) 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 = StorageActionsClientConfiguration( + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + credential_scopes=credential_scopes, + **kwargs ) + _policies = kwargs.pop("policies", None) if _policies is None: _policies = [ @@ -83,28 +83,24 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: ARMPipelineClient = ARMPipelineClient(base_url=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.storage_tasks = StorageTasksOperations(self._client, self._config, self._serialize, self._deserialize) - self.storage_task_assignment = StorageTaskAssignmentOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.storage_tasks_report = StorageTasksReportOperations( + self.storage_tasks_operation_group = StorageTasksOperationGroupOperations( 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 @@ -117,7 +113,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/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_configuration.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_configuration.py index 04c32f360d70..9ebed663e13e 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_configuration.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -17,22 +17,30 @@ from azure.core.credentials import TokenCredential -class StorageActionsMgmtClientConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for StorageActionsMgmtClient. +class StorageActionsClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for StorageActionsClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials.TokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: Api Version. Default value is "2023-01-01". Note that overriding this - default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is "2023-01-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: api_version: str = kwargs.pop("api_version", "2023-01-01") if credential is None: @@ -42,6 +50,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-storageactions/{}".format(VERSION)) diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_patch.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_patch.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/__init__.py new file mode 100644 index 000000000000..8026245c2abc --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/__init__.py @@ -0,0 +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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/model_base.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/model_base.py new file mode 100644 index 000000000000..49d5c7259389 --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/model_base.py @@ -0,0 +1,1232 @@ +# 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) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +# pylint: disable=protected-access, broad-except + +import copy +import calendar +import decimal +import functools +import sys +import logging +import base64 +import re +import typing +import enum +import email.utils +from datetime import datetime, date, time, timedelta, timezone +from json import JSONEncoder +import xml.etree.ElementTree as ET +from collections.abc import MutableMapping +from typing_extensions import Self +import isodate +from azure.core.exceptions import DeserializationError +from azure.core import CaseInsensitiveEnumMeta +from azure.core.pipeline import PipelineResponse +from azure.core.serialization import _Null + +_LOGGER = logging.getLogger(__name__) + +__all__ = ["SdkJSONEncoder", "Model", "rest_field", "rest_discriminator"] + +TZ_UTC = timezone.utc +_T = typing.TypeVar("_T") + + +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" +) + + +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 + + +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) + + +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 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: typing.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: + return self._data.__getitem__(key) + + def __setitem__(self, key: str, value: typing.Any) -> None: + 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) -> typing.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: + 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): + 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: typing.Dict[str, "_RestField"], rest_name: str +) -> typing.Optional["_RestField"]: + try: + return next(rf for rf in attr_to_rest_field.values() if rf._rest_name == rest_name) + except StopIteration: + return None + + +def _create_value(rf: typing.Optional["_RestField"], value: typing.Any) -> typing.Any: + if not rf: + return _serialize(value, None) + if rf._is_multipart_file_input: + return value + if rf._is_model: + return _deserialize(rf._type, value) + if isinstance(value, ET.Element): + value = _deserialize(rf._type, value) + return _serialize(value, rf._format) + + +class Model(_MyMutableMapping): + _is_model = True + # label whether current class's _attr_to_rest_field has been calculated + # could not see _attr_to_rest_field directly because subclass inherits it from parent class + _calculated: typing.Set[str] = set() + + def __init__(self, *args: typing.Any, **kwargs: typing.Any) -> None: + class_name = self.__class__.__name__ + if len(args) > 1: + raise TypeError(f"{class_name}.__init__() takes 2 positional arguments but {len(args) + 1} were given") + dict_to_pass = { + rest_field._rest_name: rest_field._default + for rest_field in self._attr_to_rest_field.values() + if rest_field._default is not _UNSET + } + if args: # pylint: disable=too-many-nested-blocks + if isinstance(args[0], ET.Element): + existed_attr_keys = [] + model_meta = getattr(self, "_xml", {}) + + for rf in self._attr_to_rest_field.values(): + prop_meta = getattr(rf, "_xml", {}) + xml_name = prop_meta.get("name", rf._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + # attribute + if prop_meta.get("attribute", False) and args[0].get(xml_name) is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].get(xml_name)) + continue + + # unwrapped element is array + if prop_meta.get("unwrapped", False): + # unwrapped array could either use prop items meta/prop meta + if prop_meta.get("itemsName"): + xml_name = prop_meta.get("itemsName") + xml_ns = prop_meta.get("itemNs") + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + items = args[0].findall(xml_name) # pyright: ignore + if len(items) > 0: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, items) + continue + + # text element is primitive type + if prop_meta.get("text", False): + if args[0].text is not None: + dict_to_pass[rf._rest_name] = _deserialize(rf._type, args[0].text) + continue + + # wrapped element could be normal property or array, it should only have one element + item = args[0].find(xml_name) + if item is not None: + existed_attr_keys.append(xml_name) + dict_to_pass[rf._rest_name] = _deserialize(rf._type, item) + + # rest thing is additional properties + for e in args[0]: + if e.tag not in existed_attr_keys: + dict_to_pass[e.tag] = _convert_element(e) + else: + dict_to_pass.update( + {k: _create_value(_get_rest_field(self._attr_to_rest_field, k), v) for k, v in args[0].items()} + ) + else: + non_attr_kwargs = [k for k in kwargs if k not in self._attr_to_rest_field] + if non_attr_kwargs: + # actual type errors only throw the first wrong keyword arg they see, so following that. + raise TypeError(f"{class_name}.__init__() got an unexpected keyword argument '{non_attr_kwargs[0]}'") + dict_to_pass.update( + { + self._attr_to_rest_field[k]._rest_name: _create_value(self._attr_to_rest_field[k], v) + for k, v in kwargs.items() + if v is not None + } + ) + super().__init__(dict_to_pass) + + def copy(self) -> "Model": + return Model(self.__dict__) + + def __new__(cls, *args: typing.Any, **kwargs: typing.Any) -> Self: + if f"{cls.__module__}.{cls.__qualname__}" not in cls._calculated: + # we know the last nine classes in mro are going to be 'Model', '_MyMutableMapping', 'MutableMapping', + # 'Mapping', 'Collection', 'Sized', 'Iterable', 'Container' and 'object' + mros = cls.__mro__[:-9][::-1] # ignore parents, and reverse the mro order + attr_to_rest_field: typing.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: typing.Dict[str, _RestField] = dict(attr_to_rest_field.items()) + cls._calculated.add(f"{cls.__module__}.{cls.__qualname__}") + + return super().__new__(cls) + + def __init_subclass__(cls, discriminator: typing.Optional[str] = None) -> None: + for base in cls.__bases__: + if hasattr(base, "__mapping__"): + base.__mapping__[discriminator or cls.__name__] = cls # type: ignore + + @classmethod + def _get_discriminator(cls, exist_discriminators) -> typing.Optional["_RestField"]: + for v in cls.__dict__.values(): + if isinstance(v, _RestField) and v._is_discriminator and v._rest_name not in exist_discriminators: + return v + return None + + @classmethod + def _deserialize(cls, data, exist_discriminators): + if not hasattr(cls, "__mapping__"): + return cls(data) + discriminator = cls._get_discriminator(exist_discriminators) + if discriminator is None: + return cls(data) + exist_discriminators.append(discriminator._rest_name) + if isinstance(data, ET.Element): + model_meta = getattr(cls, "_xml", {}) + prop_meta = getattr(discriminator, "_xml", {}) + xml_name = prop_meta.get("name", discriminator._rest_name) + xml_ns = prop_meta.get("ns", model_meta.get("ns", None)) + if xml_ns: + xml_name = "{" + xml_ns + "}" + xml_name + + if data.get(xml_name) is not None: + discriminator_value = data.get(xml_name) + else: + discriminator_value = data.find(xml_name).text # pyright: ignore + else: + discriminator_value = data.get(discriminator._rest_name) + mapped_cls = cls.__mapping__.get(discriminator_value, cls) # pyright: ignore # pylint: disable=no-member + return mapped_cls._deserialize(data, exist_discriminators) + + def as_dict(self, *, exclude_readonly: bool = False) -> typing.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: typing.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: typing.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 _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) + return type(obj)(_deserialize(deserializer, entry, module) for entry in obj) + + +def _sorted_annotations(types: typing.List[typing.Any]) -> typing.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-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 for a in annotation.__args__ if a == type(None)): # pyright: ignore + if len(annotation.__args__) <= 2: # pyright: ignore + if_obj_deserializer = _get_deserialize_callable_from_annotation( + next(a for a in annotation.__args__ if a != type(None)), 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 != type(None)] # 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: + if annotation._name == "Dict": # pyright: ignore + 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: + if annotation._name in ["List", "Set", "Tuple", "Sequence"]: # pyright: ignore + 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 is None: + return value + if deserializer in [int, float, bool]: + return deserializer(value) + if isinstance(deserializer, CaseInsensitiveEnumMeta): + try: + return deserializer(value) + except ValueError: + # for unknown value, return raw value + return 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, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _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, + value: typing.Any, +) -> typing.Any: + try: + return _deserialize_xml(deserializer, value) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + +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[typing.List[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, + ): + self._type = type + self._rest_name_input = name + self._module: typing.Optional[str] = None + self._is_discriminator = is_discriminator + self._visibility = visibility + self._is_model = False + self._default = default + self._format = format + self._is_multipart_file_input = is_multipart_file_input + self._xml = xml if xml is not None else {} + + @property + def _class_type(self) -> typing.Any: + return getattr(self._type, "args", [None])[0] + + @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 + item = obj.get(self._rest_name) + if item is None: + return item + if self._is_model: + return item + return _deserialize(self._type, _serialize(item, self._format), rf=self) + + def __set__(self, obj: Model, value) -> None: + 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[typing.List[str]] = None, + default: typing.Any = _UNSET, + format: typing.Optional[str] = None, + is_multipart_file_input: bool = False, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField( + name=name, + type=type, + visibility=visibility, + default=default, + format=format, + is_multipart_file_input=is_multipart_file_input, + xml=xml, + ) + + +def rest_discriminator( + *, + name: typing.Optional[str] = None, + type: typing.Optional[typing.Callable] = None, # pylint: disable=redefined-builtin + visibility: typing.Optional[typing.List[str]] = None, + xml: typing.Optional[typing.Dict[str, typing.Any]] = None, +) -> typing.Any: + return _RestField(name=name, type=type, is_discriminator=True, visibility=visibility, xml=xml) + + +def serialize_xml(model: Model, exclude_readonly: bool = False) -> str: + """Serialize a model to XML. + + :param Model model: The model to serialize. + :param bool exclude_readonly: Whether to exclude readonly properties. + :returns: The XML representation of the model. + :rtype: str + """ + return ET.tostring(_get_element(model, exclude_readonly), encoding="unicode") # type: ignore + + +def _get_element( + o: typing.Any, + exclude_readonly: bool = False, + parent_meta: typing.Optional[typing.Dict[str, typing.Any]] = None, + wrapped_element: typing.Optional[ET.Element] = None, +) -> typing.Union[ET.Element, typing.List[ET.Element]]: + if _is_model(o): + model_meta = getattr(o, "_xml", {}) + + # if prop is a model, then use the prop element directly, else generate a wrapper of model + if wrapped_element is None: + wrapped_element = _create_xml_element( + model_meta.get("name", o.__class__.__name__), + model_meta.get("prefix"), + model_meta.get("ns"), + ) + + readonly_props = [] + if exclude_readonly: + readonly_props = [p._rest_name for p in o._attr_to_rest_field.values() if _is_readonly(p)] + + for k, v in o.items(): + # do not serialize readonly properties + if exclude_readonly and k in readonly_props: + continue + + prop_rest_field = _get_rest_field(o._attr_to_rest_field, k) + if prop_rest_field: + prop_meta = getattr(prop_rest_field, "_xml").copy() + # use the wire name as xml name if no specific name is set + if prop_meta.get("name") is None: + prop_meta["name"] = k + else: + # additional properties will not have rest field, use the wire name as xml name + prop_meta = {"name": k} + + # if no ns for prop, use model's + if prop_meta.get("ns") is None and model_meta.get("ns"): + prop_meta["ns"] = model_meta.get("ns") + prop_meta["prefix"] = model_meta.get("prefix") + + if prop_meta.get("unwrapped", False): + # unwrapped could only set on array + wrapped_element.extend(_get_element(v, exclude_readonly, prop_meta)) + elif prop_meta.get("text", False): + # text could only set on primitive type + wrapped_element.text = _get_primitive_type_value(v) + elif prop_meta.get("attribute", False): + xml_name = prop_meta.get("name", k) + if prop_meta.get("ns"): + ET.register_namespace(prop_meta.get("prefix"), prop_meta.get("ns")) # pyright: ignore + xml_name = "{" + prop_meta.get("ns") + "}" + xml_name # pyright: ignore + # attribute should be primitive type + wrapped_element.set(xml_name, _get_primitive_type_value(v)) + else: + # other wrapped prop element + wrapped_element.append(_get_wrapped_element(v, exclude_readonly, prop_meta)) + return wrapped_element + if isinstance(o, list): + return [_get_element(x, exclude_readonly, parent_meta) for x in o] # type: ignore + if isinstance(o, dict): + result = [] + for k, v in o.items(): + result.append( + _get_wrapped_element( + v, + exclude_readonly, + { + "name": k, + "ns": parent_meta.get("ns") if parent_meta else None, + "prefix": parent_meta.get("prefix") if parent_meta else None, + }, + ) + ) + return result + + # primitive case need to create element based on parent_meta + if parent_meta: + return _get_wrapped_element( + o, + exclude_readonly, + { + "name": parent_meta.get("itemsName", parent_meta.get("name")), + "prefix": parent_meta.get("itemsPrefix", parent_meta.get("prefix")), + "ns": parent_meta.get("itemsNs", parent_meta.get("ns")), + }, + ) + + raise ValueError("Could not serialize value into xml: " + o) + + +def _get_wrapped_element( + v: typing.Any, + exclude_readonly: bool, + meta: typing.Optional[typing.Dict[str, typing.Any]], +) -> ET.Element: + wrapped_element = _create_xml_element( + meta.get("name") if meta else None, meta.get("prefix") if meta else None, meta.get("ns") if meta else None + ) + if isinstance(v, (dict, list)): + wrapped_element.extend(_get_element(v, exclude_readonly, meta)) + elif _is_model(v): + _get_element(v, exclude_readonly, meta, wrapped_element) + else: + wrapped_element.text = _get_primitive_type_value(v) + return wrapped_element + + +def _get_primitive_type_value(v) -> str: + if v is True: + return "true" + if v is False: + return "false" + if isinstance(v, _Null): + return "" + return str(v) + + +def _create_xml_element(tag, prefix=None, ns=None): + if prefix and ns: + ET.register_namespace(prefix, ns) + if ns: + return ET.Element("{" + ns + "}" + tag) + return ET.Element(tag) + + +def _deserialize_xml( + deserializer: typing.Any, + value: str, +) -> typing.Any: + element = ET.fromstring(value) # nosec + return _deserialize(deserializer, element) + + +def _convert_element(e: ET.Element): + # dict case + if len(e.attrib) > 0 or len({child.tag for child in e}) > 1: + dict_result: typing.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: typing.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/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_serialization.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/serialization.py similarity index 94% rename from sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_serialization.py rename to sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/serialization.py index b24ab2885450..eb86ea23c965 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_serialization.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_utils/serialization.py @@ -1,28 +1,10 @@ -# pylint: disable=too-many-lines +# pylint: disable=line-too-long,useless-suppression,too-many-lines +# 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. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- # pyright: reportUnnecessaryTypeIgnoreComment=false @@ -48,9 +30,7 @@ IO, Mapping, Callable, - TypeVar, MutableMapping, - Type, List, ) @@ -61,13 +41,13 @@ 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 _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") -ModelType = TypeVar("ModelType", bound="Model") JSON = MutableMapping[str, Any] @@ -185,73 +165,7 @@ def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], except NameError: _long_type = int - -class UTC(datetime.tzinfo): - """Time Zone info for handling UTC""" - - def utcoffset(self, dt): - """UTF offset for UTC is 0. - - :param datetime.datetime dt: The datetime - :returns: The offset - :rtype: datetime.timedelta - """ - return datetime.timedelta(0) - - def tzname(self, dt): - """Timestamp representation. - - :param datetime.datetime dt: The datetime - :returns: The timestamp representation - :rtype: str - """ - return "Z" - - def dst(self, dt): - """No daylight saving for UTC. - - :param datetime.datetime dt: The datetime - :returns: The daylight saving time - :rtype: datetime.timedelta - """ - return datetime.timedelta(hours=1) - - -try: - from datetime import timezone as _FixedOffset # type: ignore -except ImportError: # Python 2.7 - - class _FixedOffset(datetime.tzinfo): # type: ignore - """Fixed offset in minutes east from UTC. - Copy/pasted from Python doc - :param datetime.timedelta offset: offset in timedelta format - """ - - def __init__(self, offset) -> None: - self.__offset = offset - - def utcoffset(self, dt): - return self.__offset - - def tzname(self, dt): - return str(self.__offset.total_seconds() / 3600) - - def __repr__(self): - return "".format(self.tzname(None)) - - def dst(self, dt): - return datetime.timedelta(0) - - def __getinitargs__(self): - return (self.__offset,) - - -try: - from datetime import timezone - - TZ_UTC = timezone.utc -except ImportError: - TZ_UTC = UTC() # type: ignore +TZ_UTC = datetime.timezone.utc _FLATTEN = re.compile(r"(? ModelType: + def deserialize(cls, data: Any, content_type: Optional[str] = None) -> Self: """Parse a str using the RestAPI syntax and return a model. :param str data: A str using RestAPI structure. JSON by default. :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model - :raises: DeserializationError if something went wrong - :rtype: ModelType + :raises DeserializationError: if something went wrong + :rtype: Self """ deserializer = Deserializer(cls._infer_class_models()) return deserializer(cls.__name__, data, content_type=content_type) # type: ignore @classmethod def from_dict( - cls: Type[ModelType], + cls, data: Any, key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, content_type: Optional[str] = None, - ) -> ModelType: + ) -> Self: """Parse a dict using given key extractor return a model. By default consider key @@ -479,8 +393,8 @@ def from_dict( :param function key_extractors: A key extractor function. :param str content_type: JSON by default, set application/xml if XML. :returns: An instance of this model - :raises: DeserializationError if something went wrong - :rtype: ModelType + :raises DeserializationError: if something went wrong + :rtype: Self """ deserializer = Deserializer(cls._infer_class_models()) deserializer.key_extractors = ( # type: ignore @@ -626,7 +540,7 @@ def _serialize( # pylint: disable=too-many-nested-blocks, too-many-branches, to :param object target_obj: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str, dict - :raises: SerializationError if serialization fails. + :raises SerializationError: if serialization fails. :returns: The serialized data. """ key_transformer = kwargs.get("key_transformer", self.key_transformer) @@ -736,8 +650,8 @@ def body(self, data, data_type, **kwargs): :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: dict - :raises: SerializationError if serialization fails. - :raises: ValueError if data is None + :raises SerializationError: if serialization fails. + :raises ValueError: if data is None :returns: The serialized request body """ @@ -781,8 +695,8 @@ def url(self, name, data, data_type, **kwargs): :param str data_type: The type to be serialized from. :rtype: str :returns: The serialized URL path - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :raises TypeError: if serialization fails. + :raises ValueError: if data is None """ try: output = self.serialize_data(data, data_type, **kwargs) @@ -805,8 +719,8 @@ def query(self, name, data, data_type, **kwargs): :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str, list - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :raises TypeError: if serialization fails. + :raises ValueError: if data is None :returns: The serialized query parameter """ try: @@ -835,8 +749,8 @@ def header(self, name, data, data_type, **kwargs): :param object data: The data to be serialized. :param str data_type: The type to be serialized from. :rtype: str - :raises: TypeError if serialization fails. - :raises: ValueError if data is None + :raises TypeError: if serialization fails. + :raises ValueError: if data is None :returns: The serialized header """ try: @@ -855,9 +769,9 @@ def serialize_data(self, data, data_type, **kwargs): :param object data: The data to be serialized. :param str data_type: The type to be serialized from. - :raises: AttributeError if required data is None. - :raises: ValueError if data is None - :raises: SerializationError if serialization fails. + :raises AttributeError: if required data is None. + :raises ValueError: if data is None + :raises SerializationError: if serialization fails. :returns: The serialized data. :rtype: str, int, float, bool, dict, list """ @@ -1192,7 +1106,7 @@ def serialize_rfc(attr, **kwargs): # pylint: disable=unused-argument :param Datetime attr: Object to be serialized. :rtype: str - :raises: TypeError if format invalid. + :raises TypeError: if format invalid. :return: serialized rfc """ try: @@ -1218,7 +1132,7 @@ def serialize_iso(attr, **kwargs): # pylint: disable=unused-argument :param Datetime attr: Object to be serialized. :rtype: str - :raises: SerializationError if format invalid. + :raises SerializationError: if format invalid. :return: serialized iso """ if isinstance(attr, str): @@ -1251,7 +1165,7 @@ def serialize_unix(attr, **kwargs): # pylint: disable=unused-argument :param Datetime attr: Object to be serialized. :rtype: int - :raises: SerializationError if format invalid + :raises SerializationError: if format invalid :return: serialied unix """ if isinstance(attr, int): @@ -1429,7 +1343,7 @@ def xml_key_extractor(attr, attr_desc, data): # pylint: disable=unused-argument # Iter and wrapped, should have found one node only (the wrap one) if len(children) != 1: raise DeserializationError( - "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( # pylint: disable=line-too-long + "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format( xml_name ) ) @@ -1488,7 +1402,7 @@ def __call__(self, target_obj, response_data, content_type=None): :param str target_obj: Target data type to deserialize to. :param requests.Response response_data: REST response object. :param str content_type: Swagger "produces" if available. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. :rtype: object """ @@ -1502,7 +1416,7 @@ def _deserialize(self, target_obj, data): # pylint: disable=inconsistent-return :param str target_obj: Target data type to deserialize to. :param object data: Object to deserialize. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. :rtype: object """ @@ -1717,7 +1631,7 @@ def deserialize_data(self, data, data_type): # pylint: disable=too-many-return- :param str data: The response string to be deserialized. :param str data_type: The type to deserialize to. - :raises: DeserializationError if deserialization fails. + :raises DeserializationError: if deserialization fails. :return: Deserialized object. :rtype: object """ @@ -1799,7 +1713,7 @@ def deserialize_object(self, attr, **kwargs): # pylint: disable=too-many-return :param dict attr: Dictionary to be deserialized. :return: Deserialized object. :rtype: dict - :raises: TypeError if non-builtin datatype encountered. + :raises TypeError: if non-builtin datatype encountered. """ if attr is None: return None @@ -1845,7 +1759,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. """ # If we're here, data is supposed to be a basic type. # If it's still an XML node, take the text @@ -1936,7 +1850,7 @@ def deserialize_bytearray(attr): :param str attr: response string to be deserialized. :return: Deserialized bytearray :rtype: bytearray - :raises: TypeError if string format invalid. + :raises TypeError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1949,7 +1863,7 @@ def deserialize_base64(attr): :param str attr: response string to be deserialized. :return: Deserialized base64 string :rtype: bytearray - :raises: TypeError if string format invalid. + :raises TypeError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1964,7 +1878,7 @@ def deserialize_decimal(attr): :param str attr: response string to be deserialized. :return: Deserialized decimal - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. :rtype: decimal """ if isinstance(attr, ET.Element): @@ -1982,7 +1896,7 @@ def deserialize_long(attr): :param str attr: response string to be deserialized. :return: Deserialized int :rtype: long or int - :raises: ValueError if string format invalid. + :raises ValueError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -1995,7 +1909,7 @@ def deserialize_duration(attr): :param str attr: response string to be deserialized. :return: Deserialized duration :rtype: TimeDelta - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -2013,7 +1927,7 @@ def deserialize_date(attr): :param str attr: response string to be deserialized. :return: Deserialized date :rtype: Date - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -2029,7 +1943,7 @@ def deserialize_time(attr): :param str attr: response string to be deserialized. :return: Deserialized time :rtype: datetime.time - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -2044,14 +1958,14 @@ def deserialize_rfc(attr): :param str attr: response string to be deserialized. :return: Deserialized RFC datetime :rtype: Datetime - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text try: parsed_date = email.utils.parsedate_tz(attr) # type: ignore date_obj = datetime.datetime( - *parsed_date[:6], tzinfo=_FixedOffset(datetime.timedelta(minutes=(parsed_date[9] or 0) / 60)) + *parsed_date[:6], tzinfo=datetime.timezone(datetime.timedelta(minutes=(parsed_date[9] or 0) / 60)) ) if not date_obj.tzinfo: date_obj = date_obj.astimezone(tz=TZ_UTC) @@ -2067,7 +1981,7 @@ def deserialize_iso(attr): :param str attr: response string to be deserialized. :return: Deserialized ISO datetime :rtype: Datetime - :raises: DeserializationError if string format invalid. + :raises DeserializationError: if string format invalid. """ if isinstance(attr, ET.Element): attr = attr.text @@ -2105,7 +2019,7 @@ def deserialize_unix(attr): :param int attr: Object to be serialized. :return: Deserialized datetime :rtype: Datetime - :raises: DeserializationError if format invalid + :raises DeserializationError: if format invalid """ if isinstance(attr, ET.Element): attr = int(attr.text) # type: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_version.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_version.py index dfa6ee022f15..0ec13ea52bbf 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_version.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/_version.py @@ -2,8 +2,8 @@ # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.0.0b2" +VERSION = "1.0.0" diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/__init__.py index cbb899c4466e..5ba51723e2ed 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/__init__.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/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 ._storage_actions_mgmt_client import StorageActionsMgmtClient # type: ignore +from ._client import StorageActionsClient # type: ignore try: from ._patch import __all__ as _patch_all @@ -22,7 +22,7 @@ from ._patch import patch_sdk as _patch_sdk __all__ = [ - "StorageActionsMgmtClient", + "StorageActionsClient", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_storage_actions_mgmt_client.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_client.py similarity index 65% rename from sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_storage_actions_mgmt_client.py rename to sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_client.py index 306aa053741f..9370ac6b7dd1 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_storage_actions_mgmt_client.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_client.py @@ -2,69 +2,69 @@ # -------------------------------------------------------------------------- # 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 -from typing import Any, Awaitable, TYPE_CHECKING +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 +from azure.core.settings import settings from azure.mgmt.core import AsyncARMPipelineClient from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy +from azure.mgmt.core.tools import get_arm_endpoints -from .. import models as _models -from .._serialization import Deserializer, Serializer -from ._configuration import StorageActionsMgmtClientConfiguration -from .operations import ( - Operations, - StorageTaskAssignmentOperations, - StorageTasksOperations, - StorageTasksReportOperations, -) +from .._utils.serialization import Deserializer, Serializer +from ._configuration import StorageActionsClientConfiguration +from .operations import Operations, StorageTasksOperationGroupOperations, StorageTasksOperations if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential -class StorageActionsMgmtClient: +class StorageActionsClient: """The Azure Storage Actions Management API. :ivar operations: Operations operations :vartype operations: azure.mgmt.storageactions.aio.operations.Operations :ivar storage_tasks: StorageTasksOperations operations :vartype storage_tasks: azure.mgmt.storageactions.aio.operations.StorageTasksOperations - :ivar storage_task_assignment: StorageTaskAssignmentOperations operations - :vartype storage_task_assignment: - azure.mgmt.storageactions.aio.operations.StorageTaskAssignmentOperations - :ivar storage_tasks_report: StorageTasksReportOperations operations - :vartype storage_tasks_report: - azure.mgmt.storageactions.aio.operations.StorageTasksReportOperations - :param credential: Credential needed for the client to connect to Azure. Required. + :ivar storage_tasks_operation_group: StorageTasksOperationGroupOperations operations + :vartype storage_tasks_operation_group: + azure.mgmt.storageactions.aio.operations.StorageTasksOperationGroupOperations + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :param base_url: Service URL. Default value is "https://management.azure.com". + :param base_url: Service host. Default value is None. :type base_url: str - :keyword api_version: Api Version. Default value is "2023-01-01". Note that overriding this - default value may result in unsupported behavior. + :keyword api_version: The API version to use for this operation. Default value is "2023-01-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. """ def __init__( - self, - credential: "AsyncTokenCredential", - subscription_id: str, - base_url: str = "https://management.azure.com", - **kwargs: Any + self, credential: "AsyncTokenCredential", subscription_id: str, base_url: Optional[str] = None, **kwargs: Any ) -> None: - self._config = StorageActionsMgmtClientConfiguration( - credential=credential, subscription_id=subscription_id, **kwargs + _endpoint = "{endpoint}" + _cloud = kwargs.pop("cloud_setting", None) 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 = StorageActionsClientConfiguration( + credential=credential, + subscription_id=subscription_id, + base_url=cast(str, base_url), + credential_scopes=credential_scopes, + **kwargs ) + _policies = kwargs.pop("policies", None) if _policies is None: _policies = [ @@ -83,22 +83,20 @@ def __init__( policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, self._config.http_logging_policy, ] - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + self._client: AsyncARMPipelineClient = AsyncARMPipelineClient( + 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.storage_tasks = StorageTasksOperations(self._client, self._config, self._serialize, self._deserialize) - self.storage_task_assignment = StorageTaskAssignmentOperations( - self._client, self._config, self._serialize, self._deserialize - ) - self.storage_tasks_report = StorageTasksReportOperations( + self.storage_tasks_operation_group = StorageTasksOperationGroupOperations( 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. @@ -106,7 +104,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 @@ -119,7 +117,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/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_configuration.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_configuration.py index 64446f660572..8b5af7c61ea6 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_configuration.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_configuration.py @@ -2,7 +2,7 @@ # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. +# Code generated by Microsoft (R) Python Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- @@ -17,22 +17,30 @@ from azure.core.credentials_async import AsyncTokenCredential -class StorageActionsMgmtClientConfiguration: # pylint: disable=too-many-instance-attributes - """Configuration for StorageActionsMgmtClient. +class StorageActionsClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for StorageActionsClient. Note that all parameters used to create this instance are saved as instance attributes. - :param credential: Credential needed for the client to connect to Azure. Required. + :param credential: Credential used to authenticate requests to the service. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential :param subscription_id: The ID of the target subscription. The value must be an UUID. Required. :type subscription_id: str - :keyword api_version: Api Version. Default value is "2023-01-01". Note that overriding this - default value may result in unsupported behavior. + :param base_url: Service host. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: The API version to use for this operation. Default value is "2023-01-01". + Note that overriding this default value may result in unsupported behavior. :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: + def __init__( + self, + credential: "AsyncTokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: api_version: str = kwargs.pop("api_version", "2023-01-01") if credential is None: @@ -42,6 +50,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.credential = credential self.subscription_id = subscription_id + self.base_url = base_url self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-storageactions/{}".format(VERSION)) diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_patch.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_patch.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/__init__.py index 39b5abc1fb68..248b5bd5bf04 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/__init__.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/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 @@ -13,9 +13,8 @@ from ._patch import * # pylint: disable=unused-wildcard-import from ._operations import Operations # type: ignore -from ._storage_tasks_operations import StorageTasksOperations # type: ignore -from ._storage_task_assignment_operations import StorageTaskAssignmentOperations # type: ignore -from ._storage_tasks_report_operations import StorageTasksReportOperations # type: ignore +from ._operations import StorageTasksOperations # type: ignore +from ._operations import StorageTasksOperationGroupOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * @@ -24,8 +23,7 @@ __all__ = [ "Operations", "StorageTasksOperations", - "StorageTaskAssignmentOperations", - "StorageTasksReportOperations", + "StorageTasksOperationGroupOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_operations.py index ebc8172376f6..21bafeaf20d5 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_operations.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_operations.py @@ -1,14 +1,18 @@ +# 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. # -------------------------------------------------------------------------- -import sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar +from collections.abc import MutableMapping +from io import IOBase +import json +from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, List, Optional, TypeVar, Union, cast, overload import urllib.parse +from azure.core import AsyncPipelineClient from azure.core.async_paging import AsyncItemPaged, AsyncList from azure.core.exceptions import ( ClientAuthenticationError, @@ -16,23 +20,39 @@ ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.pipeline import PipelineResponse +from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod 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 azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._operations import build_list_request +from ..._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize +from ..._utils.serialization import Deserializer, Serializer +from ...operations._operations import ( + build_operations_list_request, + build_storage_tasks_create_request, + build_storage_tasks_delete_request, + build_storage_tasks_get_request, + build_storage_tasks_list_by_resource_group_request, + build_storage_tasks_list_by_subscription_request, + build_storage_tasks_list_request, + build_storage_tasks_operation_group_preview_actions_request, + build_storage_tasks_storage_task_assignment_list_request, + build_storage_tasks_update_request, +) +from .._configuration import StorageActionsClientConfiguration -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] +JSON = MutableMapping[str, Any] class Operations: @@ -41,32 +61,807 @@ class Operations: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.aio.StorageActionsMgmtClient`'s + :class:`~azure.mgmt.storageactions.aio.StorageActionsClient`'s :attr:`operations` attribute. """ - models = _models - def __init__(self, *args, **kwargs) -> None: input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + self._client: AsyncPipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: StorageActionsClientConfiguration = 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) -> AsyncIterable["_models.Operation"]: """Lists all of the available Storage Actions Rest API operations. - :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.storageactions.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 {} + + cls: ClsType[List[_models.Operation]] = 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_operations_list_request( + 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), 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.Operation], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class StorageTasksOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.storageactions.aio.StorageActionsClient`'s + :attr:`storage_tasks` 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: StorageActionsClientConfiguration = 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, storage_task_name: str, **kwargs: Any) -> _models.StorageTask: + """Get the storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: StorageTask. The StorageTask is compatible with MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTask + :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.StorageTask] = kwargs.pop("cls", None) + + _request = build_storage_tasks_get_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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 = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.StorageTask, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + async def _create_initial( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTask, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_storage_tasks_create_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: _models.StorageTask, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTask + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. 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: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. 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: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTask, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. Is one of the following types: + StorageTask, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTask or JSON or IO[bytes] + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _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[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._create_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + parameters=parameters, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + async def _update_initial( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTaskUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_storage_tasks_update_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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) + + _stream = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + async def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: _models.StorageTaskUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of AsyncLROPoller that returns StorageTask. The StorageTask is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. 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: An instance of AsyncLROPoller that returns StorageTask. The StorageTask is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. 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: An instance of AsyncLROPoller that returns StorageTask. The StorageTask is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTaskUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> AsyncLROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. Is one of the + following types: StorageTaskUpdateParameters, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters or JSON or + IO[bytes] + :return: An instance of AsyncLROPoller that returns StorageTask. The StorageTask is compatible + with MutableMapping + :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _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.StorageTask] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._update_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + parameters=parameters, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.StorageTask, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[_models.StorageTask].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[_models.StorageTask]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + async def _delete_initial( + self, resource_group_name: str, storage_task_name: str, **kwargs: Any + ) -> AsyncIterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + + _request = build_storage_tasks_delete_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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 = True + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) + if response.status_code not in [202, 204]: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace_async + async def begin_delete( + self, resource_group_name: str, storage_task_name: str, **kwargs: Any + ) -> AsyncLROPoller[None]: + """Delete the storage task resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: An instance of AsyncLROPoller that returns None + :rtype: ~azure.core.polling.AsyncLROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = await self._delete_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: AsyncPollingMethod = cast( + AsyncPollingMethod, AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling + if cont_token: + return AsyncLROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.StorageTask"]: + """Lists all the storage tasks available under the given resource group. + + :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 StorageTask + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.StorageTask]] = kwargs.pop("cls", None) error_map: MutableMapping = { 401: ClientAuthenticationError, @@ -79,12 +874,19 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Operation"]: def prepare_request(next_link=None): if not next_link: - _request = build_list_request( - api_version=api_version, + _request = build_storage_tasks_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, ) - _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 @@ -99,16 +901,21 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("OperationListResult", pipeline_response) - list_of_elem = deserialized.value + deserialized = pipeline_response.http_response.json() + list_of_elem = _deserialize(List[_models.StorageTask], 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) @@ -121,9 +928,456 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) 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) -> AsyncIterable["_models.StorageTask"]: + """Lists all the storage tasks available under the subscription. + + :return: An iterator like instance of StorageTask + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.StorageTask]] = 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_storage_tasks_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), 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.StorageTask], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def list( + self, resource_group_name: str, storage_task_name: str, *, filter: Optional[str] = None, **kwargs: Any + ) -> AsyncIterable["_models.StorageTaskReportInstance"]: + """Fetch the storage tasks run report summary for each assignment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :keyword filter: Optional. When specified, it can be used to query using reporting properties. + Default value is None. + :paramtype filter: str + :return: An iterator like instance of StorageTaskReportInstance + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTaskReportInstance] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + maxpagesize = kwargs.pop("maxpagesize", None) + cls: ClsType[List[_models.StorageTaskReportInstance]] = 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_storage_tasks_list_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + subscription_id=self._config.subscription_id, + maxpagesize=maxpagesize, + filter=filter, + 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), 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.StorageTaskReportInstance], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + @distributed_trace + def storage_task_assignment_list( + self, resource_group_name: str, storage_task_name: str, **kwargs: Any + ) -> AsyncIterable["_models.StorageTaskAssignment"]: + """Lists Resource IDs of the Storage Task Assignments associated with this Storage Task. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: An iterator like instance of StorageTaskAssignment + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTaskAssignment] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + maxpagesize = kwargs.pop("maxpagesize", None) + cls: ClsType[List[_models.StorageTaskAssignment]] = 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_storage_tasks_storage_task_assignment_list_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + subscription_id=self._config.subscription_id, + maxpagesize=maxpagesize, + 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), 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.StorageTaskAssignment], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return AsyncItemPaged(get_next, extract_data) + + +class StorageTasksOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.storageactions.aio.StorageActionsClient`'s + :attr:`storage_tasks_operation_group` 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: StorageActionsClientConfiguration = 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") + + @overload + async def preview_actions( + self, + location: str, + parameters: _models.StorageTaskPreviewAction, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def preview_actions( + self, location: str, parameters: JSON, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def preview_actions( + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def preview_actions( + self, location: str, parameters: Union[_models.StorageTaskPreviewAction, JSON, IO[bytes]], **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. Is one of the following types: + StorageTaskPreviewAction, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction or JSON or + IO[bytes] + :return: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :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.StorageTaskPreviewAction] = 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_storage_tasks_operation_group_preview_actions_request( + location=location, + 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) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + await response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.StorageTaskPreviewAction, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_patch.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_patch.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_task_assignment_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_task_assignment_operations.py deleted file mode 100644 index 3b3bc670c749..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_task_assignment_operations.py +++ /dev/null @@ -1,147 +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. -# -------------------------------------------------------------------------- -import sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._storage_task_assignment_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class StorageTaskAssignmentOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.aio.StorageActionsMgmtClient`'s - :attr:`storage_task_assignment` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, storage_task_name: str, maxpagesize: Optional[int] = None, **kwargs: Any - ) -> AsyncIterable["_models.StorageTaskAssignment"]: - """Lists Resource IDs of the Storage Task Assignments associated with this Storage Task. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param maxpagesize: Optional, specifies the maximum number of Storage Task Assignment Resource - IDs to be included in the list response. Default value is None. - :type maxpagesize: int - :return: An iterator like instance of either StorageTaskAssignment or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTaskAssignment] - :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.StorageTaskAssignmentsListResult] = 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, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - maxpagesize=maxpagesize, - 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("StorageTaskAssignmentsListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_operations.py deleted file mode 100644 index 9d7c5c406d3e..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_operations.py +++ /dev/null @@ -1,930 +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 io import IOBase -import sys -from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod -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 azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling - -from ... import models as _models -from ...operations._storage_tasks_operations import ( - build_create_request, - build_delete_request, - build_get_request, - build_list_by_resource_group_request, - build_list_by_subscription_request, - build_preview_actions_request, - build_update_request, -) - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class StorageTasksOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.aio.StorageActionsMgmtClient`'s - :attr:`storage_tasks` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - async def _create_initial( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTask, IO[bytes]], - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "StorageTask") - - _request = build_create_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 201, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: _models.StorageTask, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTask - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. 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: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTask, IO[bytes]], - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. Is either a StorageTask type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTask or IO[bytes] - :return: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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.StorageTask] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._create_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - parameters=parameters, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.StorageTask].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.StorageTask]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - async def _delete_initial( - self, resource_group_name: str, storage_task_name: str, **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202, 204]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @distributed_trace_async - async def begin_delete( - self, resource_group_name: str, storage_task_name: str, **kwargs: Any - ) -> AsyncLROPoller[None]: - """Delete the storage task resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._delete_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - @distributed_trace_async - async def get(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> _models.StorageTask: - """Get the storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :return: StorageTask or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTask - :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.StorageTask] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - async def _update_initial( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTaskUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> AsyncIterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - - content_type = content_type or "application/json" - _json = None - _content = None - if isinstance(parameters, (IOBase, bytes)): - _content = parameters - else: - _json = self._serialize.body(parameters, "StorageTaskUpdateParameters") - - _request = build_update_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - await response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - async def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: _models.StorageTaskUpdateParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. 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: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTaskUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> AsyncLROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. Is either a - StorageTaskUpdateParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters or IO[bytes] - :return: An instance of AsyncLROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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.StorageTask] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = await self._update_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - parameters=parameters, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: AsyncPollingMethod = cast( - AsyncPollingMethod, AsyncARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling - if cont_token: - return AsyncLROPoller[_models.StorageTask].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return AsyncLROPoller[_models.StorageTask]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.StorageTask"]: - """Lists all the storage tasks available under the subscription. - - :return: An iterator like instance of either StorageTask or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTask] - :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.StorageTasksListResult] = 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("StorageTasksListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @distributed_trace - def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.StorageTask"]: - """Lists all the storage tasks available under the given resource group. - - :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 either StorageTask or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTask] - :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.StorageTasksListResult] = 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("StorageTasksListResult", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) - - @overload - async def preview_actions( - self, - location: str, - parameters: _models.StorageTaskPreviewAction, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - async def preview_actions( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace_async - async def preview_actions( - self, location: str, parameters: Union[_models.StorageTaskPreviewAction, IO[bytes]], **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. Is either a - StorageTaskPreviewAction type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction or IO[bytes] - :return: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :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.StorageTaskPreviewAction] = 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, "StorageTaskPreviewAction") - - _request = build_preview_actions_request( - location=location, - 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("StorageTaskPreviewAction", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_report_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_report_operations.py deleted file mode 100644 index df61abe36b33..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/aio/operations/_storage_tasks_report_operations.py +++ /dev/null @@ -1,156 +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. -# -------------------------------------------------------------------------- -import sys -from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar -import urllib.parse - -from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) -from azure.core.pipeline import PipelineResponse -from azure.core.rest import AsyncHttpResponse, HttpRequest -from azure.core.tracing.decorator import distributed_trace -from azure.core.utils import case_insensitive_dict -from azure.mgmt.core.exceptions import ARMErrorFormat - -from ... import models as _models -from ...operations._storage_tasks_report_operations import build_list_request - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - - -class StorageTasksReportOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.aio.StorageActionsMgmtClient`'s - :attr:`storage_tasks_report` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs) -> None: - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, - resource_group_name: str, - storage_task_name: str, - maxpagesize: Optional[int] = None, - filter: Optional[str] = None, - **kwargs: Any - ) -> AsyncIterable["_models.StorageTaskReportInstance"]: - """Fetch the storage tasks run report summary for each assignment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param maxpagesize: Optional, specifies the maximum number of Storage Task Assignment Resource - IDs to be included in the list response. Default value is None. - :type maxpagesize: int - :param filter: Optional. When specified, it can be used to query using reporting properties. - Default value is None. - :type filter: str - :return: An iterator like instance of either StorageTaskReportInstance or the result of - cls(response) - :rtype: - ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storageactions.models.StorageTaskReportInstance] - :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.StorageTaskReportSummary] = 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, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - maxpagesize=maxpagesize, - filter=filter, - 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("StorageTaskReportSummary", pipeline_response) - list_of_elem = deserialized.value - if cls: - list_of_elem = cls(list_of_elem) # type: ignore - return deserialized.next_link or None, AsyncList(list_of_elem) - - async def get_next(next_link=None): - _request = prepare_request(next_link) - - _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - response = pipeline_response.http_response - - if response.status_code not in [200]: - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return AsyncItemPaged(get_next, extract_data) diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/__init__.py index 2f51856604ed..ce2176452a1e 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/__init__.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/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,7 +13,7 @@ from ._patch import * # pylint: disable=unused-wildcard-import -from ._models_py3 import ( # type: ignore +from ._models import ( # type: ignore ElseCondition, ErrorAdditionalInfo, ErrorDetail, @@ -22,13 +22,11 @@ ManagedServiceIdentity, Operation, OperationDisplay, - OperationListResult, ProxyResource, Resource, StorageTask, StorageTaskAction, StorageTaskAssignment, - StorageTaskAssignmentsListResult, StorageTaskOperation, StorageTaskPreviewAction, StorageTaskPreviewActionCondition, @@ -40,19 +38,20 @@ StorageTaskProperties, StorageTaskReportInstance, StorageTaskReportProperties, - StorageTaskReportSummary, StorageTaskUpdateParameters, - StorageTasksListResult, + StorageTaskUpdateProperties, SystemData, TrackedResource, UserAssignedIdentity, ) -from ._storage_actions_mgmt_client_enums import ( # type: ignore +from ._enums import ( # type: ignore ActionType, CreatedByType, ManagedServiceIdentityType, MatchedBlockName, + OnFailure, + OnSuccess, Origin, ProvisioningState, RunResult, @@ -72,13 +71,11 @@ "ManagedServiceIdentity", "Operation", "OperationDisplay", - "OperationListResult", "ProxyResource", "Resource", "StorageTask", "StorageTaskAction", "StorageTaskAssignment", - "StorageTaskAssignmentsListResult", "StorageTaskOperation", "StorageTaskPreviewAction", "StorageTaskPreviewActionCondition", @@ -90,9 +87,8 @@ "StorageTaskProperties", "StorageTaskReportInstance", "StorageTaskReportProperties", - "StorageTaskReportSummary", "StorageTaskUpdateParameters", - "StorageTasksListResult", + "StorageTaskUpdateProperties", "SystemData", "TrackedResource", "UserAssignedIdentity", @@ -100,6 +96,8 @@ "CreatedByType", "ManagedServiceIdentityType", "MatchedBlockName", + "OnFailure", + "OnSuccess", "Origin", "ProvisioningState", "RunResult", diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_storage_actions_mgmt_client_enums.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_enums.py similarity index 70% rename from sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_storage_actions_mgmt_client_enums.py rename to sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_enums.py index f3b6d70c768f..8c42691275e5 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_storage_actions_mgmt_client_enums.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/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,18 +11,25 @@ class ActionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Enum. Indicates the action type. "Internal" refers to actions that are for internal only APIs.""" + """Extensible enum. Indicates the action type. "Internal" refers to actions that are for internal + only APIs. + """ INTERNAL = "Internal" + """Actions are for internal-only APIs.""" 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 ManagedServiceIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -31,9 +38,13 @@ class ManagedServiceIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """ NONE = "None" + """No managed identity.""" SYSTEM_ASSIGNED = "SystemAssigned" + """System assigned managed identity.""" USER_ASSIGNED = "UserAssigned" + """User assigned managed identity.""" SYSTEM_ASSIGNED_USER_ASSIGNED = "SystemAssigned,UserAssigned" + """System and user assigned managed identity.""" class MatchedBlockName(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -42,8 +53,18 @@ class MatchedBlockName(str, Enum, metaclass=CaseInsensitiveEnumMeta): IF = "If" ELSE = "Else" NONE = "None" - IF_ENUM = "If" - ELSE_ENUM = "Else" + + +class OnFailure(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Action to be taken when the operation fails for a object.""" + + BREAK = "break" + + +class OnSuccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Action to be taken when the operation is successful for a object.""" + + CONTINUE = "continue" class Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -52,8 +73,11 @@ class Origin(str, Enum, metaclass=CaseInsensitiveEnumMeta): """ USER = "user" + """Indicates the operation is initiated by a user.""" SYSTEM = "system" + """Indicates the operation is initiated by a system.""" USER_SYSTEM = "user,system" + """Indicates the operation is initiated by a user or system.""" class ProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models.py new file mode 100644 index 000000000000..73bb1514039a --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models.py @@ -0,0 +1,1297 @@ +# 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=useless-super-delegation + +import datetime +from typing import Any, Dict, List, 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 ElseCondition(_Model): + """The else block of storage task operation. + + :ivar operations: List of operations to execute in the else block. Required. + :vartype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] + """ + + operations: List["_models.StorageTaskOperation"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """List of operations to execute in the else block. Required.""" + + @overload + def __init__( + self, + *, + operations: List["_models.StorageTaskOperation"], + ) -> 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 ErrorAdditionalInfo(_Model): + """The resource management error additional info. + + :ivar type: The additional info type. + :vartype type: str + :ivar info: The additional info. + :vartype info: any + """ + + type: Optional[str] = rest_field(visibility=["read"]) + """The additional info type.""" + info: Optional[Any] = rest_field(visibility=["read"]) + """The additional info.""" + + +class ErrorDetail(_Model): + """The error detail. + + :ivar code: The error code. + :vartype code: str + :ivar message: The error message. + :vartype message: str + :ivar target: The error target. + :vartype target: str + :ivar details: The error details. + :vartype details: list[~azure.mgmt.storageactions.models.ErrorDetail] + :ivar additional_info: The error additional info. + :vartype additional_info: list[~azure.mgmt.storageactions.models.ErrorAdditionalInfo] + """ + + code: Optional[str] = rest_field(visibility=["read"]) + """The error code.""" + message: Optional[str] = rest_field(visibility=["read"]) + """The error message.""" + target: Optional[str] = rest_field(visibility=["read"]) + """The error target.""" + details: Optional[List["_models.ErrorDetail"]] = rest_field(visibility=["read"]) + """The error details.""" + additional_info: Optional[List["_models.ErrorAdditionalInfo"]] = rest_field( + name="additionalInfo", visibility=["read"] + ) + """The error additional info.""" + + +class ErrorResponse(_Model): + """Error response. + + :ivar error: The error object. + :vartype error: ~azure.mgmt.storageactions.models.ErrorDetail + """ + + error: Optional["_models.ErrorDetail"] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The error object.""" + + @overload + def __init__( + self, + *, + error: Optional["_models.ErrorDetail"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class IfCondition(_Model): + """The if block of storage task operation. + + :ivar condition: Condition predicate to evaluate each object. See + `https://aka.ms/storagetaskconditions `_ for valid + properties and operators. Required. + :vartype condition: str + :ivar operations: List of operations to execute when the condition predicate satisfies. + Required. + :vartype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] + """ + + condition: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Condition predicate to evaluate each object. See `https://aka.ms/storagetaskconditions + `_ for valid properties and operators. Required.""" + operations: List["_models.StorageTaskOperation"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """List of operations to execute when the condition predicate satisfies. Required.""" + + @overload + def __init__( + self, + *, + condition: str, + operations: List["_models.StorageTaskOperation"], + ) -> 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 ManagedServiceIdentity(_Model): + """Managed service identity (system assigned and/or user assigned identities). + + :ivar principal_id: The service principal ID of the system assigned identity. This property + will only be provided for a system assigned identity. + :vartype principal_id: str + :ivar tenant_id: The tenant ID of the system assigned identity. This property will only be + provided for a system assigned identity. + :vartype tenant_id: str + :ivar type: The type of managed identity assigned to this resource. Required. Known values are: + "None", "SystemAssigned", "UserAssigned", and "SystemAssigned,UserAssigned". + :vartype type: str or ~azure.mgmt.storageactions.models.ManagedServiceIdentityType + :ivar user_assigned_identities: The identities assigned to this resource by the user. + :vartype user_assigned_identities: dict[str, + ~azure.mgmt.storageactions.models.UserAssignedIdentity] + """ + + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The service principal ID of the system assigned identity. This property will only be provided + for a system assigned identity.""" + tenant_id: Optional[str] = rest_field(name="tenantId", visibility=["read"]) + """The tenant ID of the system assigned identity. This property will only be provided for a system + assigned identity.""" + type: Union[str, "_models.ManagedServiceIdentityType"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The type of managed identity assigned to this resource. Required. Known values are: \"None\", + \"SystemAssigned\", \"UserAssigned\", and \"SystemAssigned,UserAssigned\".""" + user_assigned_identities: Optional[Dict[str, "_models.UserAssignedIdentity"]] = rest_field( + name="userAssignedIdentities", visibility=["read", "create", "update", "delete", "query"] + ) + """The identities assigned to this resource by the user.""" + + @overload + def __init__( + self, + *, + type: Union[str, "_models.ManagedServiceIdentityType"], + user_assigned_identities: Optional[Dict[str, "_models.UserAssignedIdentity"]] = 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 Operation(_Model): + """REST API Operation. + + :ivar name: The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + "Microsoft.Compute/virtualMachines/write", "Microsoft.Compute/virtualMachines/capture/action". + :vartype name: str + :ivar is_data_action: Whether the operation applies to data-plane. This is "true" for + data-plane operations and "false" for Azure Resource Manager/control-plane operations. + :vartype is_data_action: bool + :ivar display: Localized display information for this particular operation. + :vartype display: ~azure.mgmt.storageactions.models.OperationDisplay + :ivar origin: The intended executor of the operation; as in Resource Based Access Control + (RBAC) and audit logs UX. Default value is "user,system". Known values are: "user", "system", + and "user,system". + :vartype origin: str or ~azure.mgmt.storageactions.models.Origin + :ivar action_type: Extensible enum. Indicates the action type. "Internal" refers to actions + that are for internal only APIs. "Internal" + :vartype action_type: str or ~azure.mgmt.storageactions.models.ActionType + """ + + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the operation, as per Resource-Based Access Control (RBAC). Examples: + \"Microsoft.Compute/virtualMachines/write\", + \"Microsoft.Compute/virtualMachines/capture/action\".""" + is_data_action: Optional[bool] = rest_field(name="isDataAction", visibility=["read"]) + """Whether the operation applies to data-plane. This is \"true\" for data-plane operations and + \"false\" for Azure Resource Manager/control-plane operations.""" + display: Optional["_models.OperationDisplay"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Localized display information for this particular operation.""" + origin: Optional[Union[str, "_models.Origin"]] = rest_field(visibility=["read"]) + """The intended executor of the operation; as in Resource Based Access Control (RBAC) and audit + logs UX. Default value is \"user,system\". Known values are: \"user\", \"system\", and + \"user,system\".""" + action_type: Optional[Union[str, "_models.ActionType"]] = rest_field(name="actionType", visibility=["read"]) + """Extensible enum. Indicates the action type. \"Internal\" refers to actions that are for + internal only APIs. \"Internal\"""" + + @overload + def __init__( + self, + *, + 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): + """Localized display information for and operation. + + :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft + Monitoring Insights" or "Microsoft Compute". + :vartype provider: str + :ivar resource: The localized friendly name of the resource type related to this operation. + E.g. "Virtual Machines" or "Job Schedule Collections". + :vartype resource: str + :ivar operation: The concise, localized friendly name for the operation; suitable for + dropdowns. E.g. "Create or Update Virtual Machine", "Restart Virtual Machine". + :vartype operation: str + :ivar description: The short, localized friendly description of the operation; suitable for + tool tips and detailed views. + :vartype description: str + """ + + provider: Optional[str] = rest_field(visibility=["read"]) + """The localized friendly form of the resource provider name, e.g. \"Microsoft Monitoring + Insights\" or \"Microsoft Compute\".""" + resource: Optional[str] = rest_field(visibility=["read"]) + """The localized friendly name of the resource type related to this operation. E.g. \"Virtual + Machines\" or \"Job Schedule Collections\".""" + operation: Optional[str] = rest_field(visibility=["read"]) + """The concise, localized friendly name for the operation; suitable for dropdowns. E.g. \"Create + or Update Virtual Machine\", \"Restart Virtual Machine\".""" + description: Optional[str] = rest_field(visibility=["read"]) + """The short, localized friendly description of the operation; suitable for tool tips and detailed + views.""" + + +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.storageactions.models.SystemData + """ + + id: Optional[str] = rest_field(visibility=["read"]) + """Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}.""" + name: Optional[str] = rest_field(visibility=["read"]) + """The name of the resource.""" + type: Optional[str] = rest_field(visibility=["read"]) + """The type of the resource. E.g. \"Microsoft.Compute/virtualMachines\" or + \"Microsoft.Storage/storageAccounts\".""" + system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"]) + """Azure Resource Manager metadata containing createdBy and modifiedBy information.""" + + +class ProxyResource(Resource): + """Proxy Resource. + + :ivar id: Fully qualified resource ID for the resource. Ex - + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/{resourceProviderNamespace}/{resourceType}/{resourceName}. + :vartype id: str + :ivar name: The name of the resource. + :vartype name: str + :ivar type: The type of the resource. E.g. "Microsoft.Compute/virtualMachines" or + "Microsoft.Storage/storageAccounts". + :vartype type: str + :ivar system_data: Azure Resource Manager metadata containing createdBy and modifiedBy + information. + :vartype system_data: ~azure.mgmt.storageactions.models.SystemData + """ + + +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.storageactions.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 StorageTask(TrackedResource): + """Represents Storage Task. + + :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.storageactions.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 identity: The managed service identity of the resource. Required. + :vartype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity + :ivar properties: Properties of the storage task. Required. + :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskProperties + """ + + identity: "_models.ManagedServiceIdentity" = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The managed service identity of the resource. Required.""" + properties: "_models.StorageTaskProperties" = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Properties of the storage task. Required.""" + + __flattened_items = [ + "task_version", + "enabled", + "description", + "action", + "provisioning_state", + "creation_time_in_utc", + ] + + @overload + def __init__( + self, + *, + location: str, + identity: "_models.ManagedServiceIdentity", + properties: "_models.StorageTaskProperties", + 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: + _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 StorageTaskAction(_Model): + """The storage task action represents conditional statements and operations to be performed on + target objects. + + :ivar if_property: The if block of storage task operation. Required. + :vartype if_property: ~azure.mgmt.storageactions.models.IfCondition + :ivar else_property: The else block of storage task operation. + :vartype else_property: ~azure.mgmt.storageactions.models.ElseCondition + """ + + if_property: "_models.IfCondition" = rest_field( + name="if", visibility=["read", "create", "update", "delete", "query"] + ) + """The if block of storage task operation. Required.""" + else_property: Optional["_models.ElseCondition"] = rest_field( + name="else", visibility=["read", "create", "update", "delete", "query"] + ) + """The else block of storage task operation.""" + + @overload + def __init__( + self, + *, + if_property: "_models.IfCondition", + else_property: Optional["_models.ElseCondition"] = 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 StorageTaskAssignment(_Model): + """Storage Task Assignment associated with this Storage Task. + + :ivar id: Resource ID of the Storage Task Assignment. + :vartype id: str + """ + + id: Optional[str] = rest_field(visibility=["read"]) + """Resource ID of the Storage Task Assignment.""" + + +class StorageTaskOperation(_Model): + """Represents an operation to be performed on the object. + + :ivar name: The operation to be performed on the object. Required. Known values are: + "SetBlobTier", "SetBlobTags", "SetBlobImmutabilityPolicy", "SetBlobLegalHold", "SetBlobExpiry", + "DeleteBlob", and "UndeleteBlob". + :vartype name: str or ~azure.mgmt.storageactions.models.StorageTaskOperationName + :ivar parameters: Key-value parameters for the operation. + :vartype parameters: dict[str, str] + :ivar on_success: Action to be taken when the operation is successful for a object. "continue" + :vartype on_success: str or ~azure.mgmt.storageactions.models.OnSuccess + :ivar on_failure: Action to be taken when the operation fails for a object. "break" + :vartype on_failure: str or ~azure.mgmt.storageactions.models.OnFailure + """ + + name: Union[str, "_models.StorageTaskOperationName"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The operation to be performed on the object. Required. Known values are: \"SetBlobTier\", + \"SetBlobTags\", \"SetBlobImmutabilityPolicy\", \"SetBlobLegalHold\", \"SetBlobExpiry\", + \"DeleteBlob\", and \"UndeleteBlob\".""" + parameters: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Key-value parameters for the operation.""" + on_success: Optional[Union[str, "_models.OnSuccess"]] = rest_field( + name="onSuccess", visibility=["read", "create", "update", "delete", "query"] + ) + """Action to be taken when the operation is successful for a object. \"continue\"""" + on_failure: Optional[Union[str, "_models.OnFailure"]] = rest_field( + name="onFailure", visibility=["read", "create", "update", "delete", "query"] + ) + """Action to be taken when the operation fails for a object. \"break\"""" + + @overload + def __init__( + self, + *, + name: Union[str, "_models.StorageTaskOperationName"], + parameters: Optional[Dict[str, str]] = None, + on_success: Optional[Union[str, "_models.OnSuccess"]] = None, + on_failure: Optional[Union[str, "_models.OnFailure"]] = 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 StorageTaskPreviewAction(_Model): + """Storage Task Preview Action. + + :ivar properties: Properties of the storage task preview. Required. + :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionProperties + """ + + properties: "_models.StorageTaskPreviewActionProperties" = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the storage task preview. Required.""" + + __flattened_items = ["container", "blobs", "action"] + + @overload + def __init__( + self, + *, + properties: "_models.StorageTaskPreviewActionProperties", + ) -> 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 StorageTaskPreviewActionCondition(_Model): + """Represents the storage task conditions to be tested for a match with container and blob + properties. + + :ivar if_property: The condition to be tested for a match with container and blob properties. + Required. + :vartype if_property: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionIfCondition + :ivar else_block_exists: Specify whether the else block is present in the condition. Required. + :vartype else_block_exists: bool + """ + + if_property: "_models.StorageTaskPreviewActionIfCondition" = rest_field( + name="if", visibility=["read", "create", "update", "delete", "query"] + ) + """The condition to be tested for a match with container and blob properties. Required.""" + else_block_exists: bool = rest_field( + name="elseBlockExists", visibility=["read", "create", "update", "delete", "query"] + ) + """Specify whether the else block is present in the condition. Required.""" + + @overload + def __init__( + self, + *, + if_property: "_models.StorageTaskPreviewActionIfCondition", + else_block_exists: bool, + ) -> 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 StorageTaskPreviewActionIfCondition(_Model): + """Represents storage task preview action condition. + + :ivar condition: Storage task condition to bes tested for a match. + :vartype condition: str + """ + + condition: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Storage task condition to bes tested for a match.""" + + @overload + def __init__( + self, + *, + condition: 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 StorageTaskPreviewActionProperties(_Model): + """Storage task preview action properties. + + :ivar container: Properties of a sample container to test for a match with the preview action. + Required. + :vartype container: ~azure.mgmt.storageactions.models.StorageTaskPreviewContainerProperties + :ivar blobs: Properties of some sample blobs in the container to test for matches with the + preview action. Required. + :vartype blobs: list[~azure.mgmt.storageactions.models.StorageTaskPreviewBlobProperties] + :ivar action: Preview action to test. Required. + :vartype action: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionCondition + """ + + container: "_models.StorageTaskPreviewContainerProperties" = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of a sample container to test for a match with the preview action. Required.""" + blobs: List["_models.StorageTaskPreviewBlobProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of some sample blobs in the container to test for matches with the preview action. + Required.""" + action: "_models.StorageTaskPreviewActionCondition" = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Preview action to test. Required.""" + + @overload + def __init__( + self, + *, + container: "_models.StorageTaskPreviewContainerProperties", + blobs: List["_models.StorageTaskPreviewBlobProperties"], + action: "_models.StorageTaskPreviewActionCondition", + ) -> 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 StorageTaskPreviewBlobProperties(_Model): + """Storage task preview container properties. + + :ivar name: Name of test blob. + :vartype name: str + :ivar properties: properties key value pairs to be tested for a match against the provided + condition. + :vartype properties: + list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] + :ivar metadata: metadata key value pairs to be tested for a match against the provided + condition. + :vartype metadata: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] + :ivar tags: tags key value pairs to be tested for a match against the provided condition. + :vartype tags: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] + :ivar matched_block: Represents the condition block name that matched blob properties. Known + values are: "If", "Else", and "None". + :vartype matched_block: str or ~azure.mgmt.storageactions.models.MatchedBlockName + """ + + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Name of test blob.""" + properties: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """properties key value pairs to be tested for a match against the provided condition.""" + metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """metadata key value pairs to be tested for a match against the provided condition.""" + tags: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """tags key value pairs to be tested for a match against the provided condition.""" + matched_block: Optional[Union[str, "_models.MatchedBlockName"]] = rest_field( + name="matchedBlock", visibility=["read"] + ) + """Represents the condition block name that matched blob properties. Known values are: \"If\", + \"Else\", and \"None\".""" + + @overload + def __init__( + self, + *, + name: Optional[str] = None, + properties: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, + metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, + tags: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = 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 StorageTaskPreviewContainerProperties(_Model): + """Storage task preview container properties. + + :ivar name: Name of test container. + :vartype name: str + :ivar metadata: metadata key value pairs to be tested for a match against the provided + condition. + :vartype metadata: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] + """ + + name: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Name of test container.""" + metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """metadata key value pairs to be tested for a match against the provided condition.""" + + @overload + def __init__( + self, + *, + name: Optional[str] = None, + metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = 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 StorageTaskPreviewKeyValueProperties(_Model): + """Storage task preview object key value pair properties. + + :ivar key: Represents the key property of the pair. + :vartype key: str + :ivar value: Represents the value property of the pair. + :vartype value: str + """ + + key: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Represents the key property of the pair.""" + value: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Represents the value property of the pair.""" + + @overload + def __init__( + self, + *, + key: Optional[str] = None, + value: 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 StorageTaskProperties(_Model): + """Properties of the storage task. + + :ivar task_version: Storage task version. + :vartype task_version: int + :ivar enabled: Storage Task is enabled when set to true and disabled when set to false. + Required. + :vartype enabled: bool + :ivar description: Text that describes the purpose of the storage task. Required. + :vartype description: str + :ivar action: The storage task action that is executed. Required. + :vartype action: ~azure.mgmt.storageactions.models.StorageTaskAction + :ivar provisioning_state: Represents the provisioning state of the storage task. Known values + are: "ValidateSubscriptionQuotaBegin", "ValidateSubscriptionQuotaEnd", "Accepted", "Creating", + "Succeeded", "Deleting", "Canceled", and "Failed". + :vartype provisioning_state: str or ~azure.mgmt.storageactions.models.ProvisioningState + :ivar creation_time_in_utc: The creation date and time of the storage task in UTC. + :vartype creation_time_in_utc: ~datetime.datetime + """ + + task_version: Optional[int] = rest_field(name="taskVersion", visibility=["read"]) + """Storage task version.""" + enabled: bool = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Storage Task is enabled when set to true and disabled when set to false. Required.""" + description: str = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Text that describes the purpose of the storage task. Required.""" + action: "_models.StorageTaskAction" = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """The storage task action that is executed. Required.""" + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Represents the provisioning state of the storage task. Known values are: + \"ValidateSubscriptionQuotaBegin\", \"ValidateSubscriptionQuotaEnd\", \"Accepted\", + \"Creating\", \"Succeeded\", \"Deleting\", \"Canceled\", and \"Failed\".""" + creation_time_in_utc: Optional[datetime.datetime] = rest_field( + name="creationTimeInUtc", visibility=["read"], format="rfc3339" + ) + """The creation date and time of the storage task in UTC.""" + + @overload + def __init__( + self, + *, + enabled: bool, + description: str, + action: "_models.StorageTaskAction", + ) -> 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 StorageTaskReportInstance(ProxyResource): + """Storage Tasks run report instance. + + :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.storageactions.models.SystemData + :ivar properties: Storage task execution report for a run instance. + :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskReportProperties + """ + + properties: Optional["_models.StorageTaskReportProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Storage task execution report for a run instance.""" + + @overload + def __init__( + self, + *, + properties: Optional["_models.StorageTaskReportProperties"] = 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 StorageTaskReportProperties(_Model): + """Storage task execution report for a run instance. + + :ivar task_assignment_id: Resource ID of the Storage Task Assignment associated with this + reported run. + :vartype task_assignment_id: str + :ivar storage_account_id: Resource ID of the Storage Account where this reported run executed. + :vartype storage_account_id: str + :ivar start_time: Start time of the run instance. Filter options such as startTime gt + '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for + DateTime properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype start_time: str + :ivar finish_time: End time of the run instance. Filter options such as startTime gt + '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for + DateTime properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype finish_time: str + :ivar objects_targeted_count: Total number of objects that meet the condition as defined in the + storage task assignment execution context. Filter options such as objectsTargetedCount gt 50 + and other comparison operators can be used as described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype objects_targeted_count: str + :ivar objects_operated_on_count: Total number of objects that meet the storage tasks condition + and were operated upon. Filter options such as objectsOperatedOnCount ge 100 and other + comparison operators can be used as described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype objects_operated_on_count: str + :ivar object_failed_count: Total number of objects where task operation failed when was + attempted. Filter options such as objectFailedCount eq 0 and other comparison operators can be + used as described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype object_failed_count: str + :ivar objects_succeeded_count: Total number of objects where task operation succeeded when was + attempted.Filter options such as objectsSucceededCount gt 150 and other comparison operators + can be used as described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_. + :vartype objects_succeeded_count: str + :ivar run_status_error: Well known Azure Storage error code that represents the error + encountered during execution of the run instance. + :vartype run_status_error: str + :ivar run_status_enum: Represents the status of the execution. Known values are: "InProgress" + and "Finished". + :vartype run_status_enum: str or ~azure.mgmt.storageactions.models.RunStatusEnum + :ivar summary_report_path: Full path to the verbose report stored in the reporting container as + specified in the assignment execution context for the storage account. + :vartype summary_report_path: str + :ivar task_id: Resource ID of the Storage Task applied during this run. + :vartype task_id: str + :ivar task_version: Storage Task Version. + :vartype task_version: str + :ivar run_result: Represents the overall result of the execution for the run instance. Known + values are: "Succeeded" and "Failed". + :vartype run_result: str or ~azure.mgmt.storageactions.models.RunResult + """ + + task_assignment_id: Optional[str] = rest_field(name="taskAssignmentId", visibility=["read"]) + """Resource ID of the Storage Task Assignment associated with this reported run.""" + storage_account_id: Optional[str] = rest_field(name="storageAccountId", visibility=["read"]) + """Resource ID of the Storage Account where this reported run executed.""" + start_time: Optional[str] = rest_field(name="startTime", visibility=["read"]) + """Start time of the run instance. Filter options such as startTime gt + '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for + DateTime properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + finish_time: Optional[str] = rest_field(name="finishTime", visibility=["read"]) + """End time of the run instance. Filter options such as startTime gt + '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for + DateTime properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + objects_targeted_count: Optional[str] = rest_field(name="objectsTargetedCount", visibility=["read"]) + """Total number of objects that meet the condition as defined in the storage task assignment + execution context. Filter options such as objectsTargetedCount gt 50 and other comparison + operators can be used as described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + objects_operated_on_count: Optional[str] = rest_field(name="objectsOperatedOnCount", visibility=["read"]) + """Total number of objects that meet the storage tasks condition and were operated upon. Filter + options such as objectsOperatedOnCount ge 100 and other comparison operators can be used as + described for Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + object_failed_count: Optional[str] = rest_field(name="objectFailedCount", visibility=["read"]) + """Total number of objects where task operation failed when was attempted. Filter options such as + objectFailedCount eq 0 and other comparison operators can be used as described for Numerical + properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + objects_succeeded_count: Optional[str] = rest_field(name="objectsSucceededCount", visibility=["read"]) + """Total number of objects where task operation succeeded when was attempted.Filter options such + as objectsSucceededCount gt 150 and other comparison operators can be used as described for + Numerical properties in + `https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators + `_.""" + run_status_error: Optional[str] = rest_field(name="runStatusError", visibility=["read"]) + """Well known Azure Storage error code that represents the error encountered during execution of + the run instance.""" + run_status_enum: Optional[Union[str, "_models.RunStatusEnum"]] = rest_field( + name="runStatusEnum", visibility=["read"] + ) + """Represents the status of the execution. Known values are: \"InProgress\" and \"Finished\".""" + summary_report_path: Optional[str] = rest_field(name="summaryReportPath", visibility=["read"]) + """Full path to the verbose report stored in the reporting container as specified in the + assignment execution context for the storage account.""" + task_id: Optional[str] = rest_field(name="taskId", visibility=["read"]) + """Resource ID of the Storage Task applied during this run.""" + task_version: Optional[str] = rest_field(name="taskVersion", visibility=["read"]) + """Storage Task Version.""" + run_result: Optional[Union[str, "_models.RunResult"]] = rest_field(name="runResult", visibility=["read"]) + """Represents the overall result of the execution for the run instance. Known values are: + \"Succeeded\" and \"Failed\".""" + + +class StorageTaskUpdateParameters(_Model): + """Parameters of the storage task update request. + + :ivar identity: The identity of the resource. + :vartype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity + :ivar tags: Gets or sets a list of key value pairs that describe the resource. These tags can + be used in viewing and grouping this resource (across resource groups). A maximum of 15 tags + can be provided for a resource. Each tag must have a key no greater in length than 128 + characters and a value no greater in length than 256 characters. + :vartype tags: dict[str, str] + :ivar properties: Properties of the storage task. + :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskUpdateProperties + """ + + identity: Optional["_models.ManagedServiceIdentity"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The identity of the resource.""" + tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Gets or sets a list of key value pairs that describe the resource. These tags can be used in + viewing and grouping this resource (across resource groups). A maximum of 15 tags can be + provided for a resource. Each tag must have a key no greater in length than 128 characters and + a value no greater in length than 256 characters.""" + properties: Optional["_models.StorageTaskUpdateProperties"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """Properties of the storage task.""" + + __flattened_items = [ + "task_version", + "enabled", + "description", + "action", + "provisioning_state", + "creation_time_in_utc", + ] + + @overload + def __init__( + self, + *, + identity: Optional["_models.ManagedServiceIdentity"] = None, + tags: Optional[Dict[str, str]] = None, + properties: Optional["_models.StorageTaskUpdateProperties"] = 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 StorageTaskUpdateProperties(_Model): + """Properties of the storage task. + + :ivar task_version: Storage task version. + :vartype task_version: int + :ivar enabled: Storage Task is enabled when set to true and disabled when set to false. + :vartype enabled: bool + :ivar description: Text that describes the purpose of the storage task. + :vartype description: str + :ivar action: The storage task action that is executed. + :vartype action: ~azure.mgmt.storageactions.models.StorageTaskAction + :ivar provisioning_state: Represents the provisioning state of the storage task. Known values + are: "ValidateSubscriptionQuotaBegin", "ValidateSubscriptionQuotaEnd", "Accepted", "Creating", + "Succeeded", "Deleting", "Canceled", and "Failed". + :vartype provisioning_state: str or ~azure.mgmt.storageactions.models.ProvisioningState + :ivar creation_time_in_utc: The creation date and time of the storage task in UTC. + :vartype creation_time_in_utc: ~datetime.datetime + """ + + task_version: Optional[int] = rest_field(name="taskVersion", visibility=["read"]) + """Storage task version.""" + enabled: Optional[bool] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Storage Task is enabled when set to true and disabled when set to false.""" + description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"]) + """Text that describes the purpose of the storage task.""" + action: Optional["_models.StorageTaskAction"] = rest_field( + visibility=["read", "create", "update", "delete", "query"] + ) + """The storage task action that is executed.""" + provisioning_state: Optional[Union[str, "_models.ProvisioningState"]] = rest_field( + name="provisioningState", visibility=["read"] + ) + """Represents the provisioning state of the storage task. Known values are: + \"ValidateSubscriptionQuotaBegin\", \"ValidateSubscriptionQuotaEnd\", \"Accepted\", + \"Creating\", \"Succeeded\", \"Deleting\", \"Canceled\", and \"Failed\".""" + creation_time_in_utc: Optional[datetime.datetime] = rest_field( + name="creationTimeInUtc", visibility=["read"], format="rfc3339" + ) + """The creation date and time of the storage task in UTC.""" + + @overload + def __init__( + self, + *, + enabled: Optional[bool] = None, + description: Optional[str] = None, + action: Optional["_models.StorageTaskAction"] = None, + ) -> None: ... + + @overload + def __init__(self, mapping: Mapping[str, Any]) -> None: + """ + :param mapping: raw JSON to initialize the model. + :type mapping: Mapping[str, Any] + """ + + def __init__(self, *args: Any, **kwargs: Any) -> None: + super().__init__(*args, **kwargs) + + +class SystemData(_Model): + """Metadata pertaining to creation and last modification of the resource. + + :ivar created_by: The identity that created the resource. + :vartype created_by: str + :ivar created_by_type: The type of identity that created the resource. Known values are: + "User", "Application", "ManagedIdentity", and "Key". + :vartype created_by_type: str or ~azure.mgmt.storageactions.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.storageactions.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 UserAssignedIdentity(_Model): + """User assigned identity properties. + + :ivar principal_id: The principal ID of the assigned identity. + :vartype principal_id: str + :ivar client_id: The client ID of the assigned identity. + :vartype client_id: str + """ + + principal_id: Optional[str] = rest_field(name="principalId", visibility=["read"]) + """The principal ID of the assigned identity.""" + client_id: Optional[str] = rest_field(name="clientId", visibility=["read"]) + """The client ID of the assigned identity.""" diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models_py3.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models_py3.py deleted file mode 100644 index 15f27302a88b..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_models_py3.py +++ /dev/null @@ -1,1368 +0,0 @@ -# pylint: disable=too-many-lines -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import datetime -from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union - -from .. import _serialization - -if TYPE_CHECKING: - from .. import models as _models - - -class ElseCondition(_serialization.Model): - """The else block of storage task operation. - - All required parameters must be populated in order to send to server. - - :ivar operations: List of operations to execute in the else block. Required. - :vartype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] - """ - - _validation = { - "operations": {"required": True}, - } - - _attribute_map = { - "operations": {"key": "operations", "type": "[StorageTaskOperation]"}, - } - - def __init__(self, *, operations: List["_models.StorageTaskOperation"], **kwargs: Any) -> None: - """ - :keyword operations: List of operations to execute in the else block. Required. - :paramtype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] - """ - super().__init__(**kwargs) - self.operations = operations - - -class ErrorAdditionalInfo(_serialization.Model): - """The resource management error additional info. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar type: The additional info type. - :vartype type: str - :ivar info: The additional info. - :vartype info: JSON - """ - - _validation = { - "type": {"readonly": True}, - "info": {"readonly": True}, - } - - _attribute_map = { - "type": {"key": "type", "type": "str"}, - "info": {"key": "info", "type": "object"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.type = None - self.info = None - - -class ErrorDetail(_serialization.Model): - """The error detail. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar code: The error code. - :vartype code: str - :ivar message: The error message. - :vartype message: str - :ivar target: The error target. - :vartype target: str - :ivar details: The error details. - :vartype details: list[~azure.mgmt.storageactions.models.ErrorDetail] - :ivar additional_info: The error additional info. - :vartype additional_info: list[~azure.mgmt.storageactions.models.ErrorAdditionalInfo] - """ - - _validation = { - "code": {"readonly": True}, - "message": {"readonly": True}, - "target": {"readonly": True}, - "details": {"readonly": True}, - "additional_info": {"readonly": True}, - } - - _attribute_map = { - "code": {"key": "code", "type": "str"}, - "message": {"key": "message", "type": "str"}, - "target": {"key": "target", "type": "str"}, - "details": {"key": "details", "type": "[ErrorDetail]"}, - "additional_info": {"key": "additionalInfo", "type": "[ErrorAdditionalInfo]"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.code = None - self.message = None - self.target = None - self.details = None - self.additional_info = None - - -class ErrorResponse(_serialization.Model): - """Common error response for all Azure Resource Manager APIs to return error details for failed - operations. (This also follows the OData error response format.). - - :ivar error: The error object. - :vartype error: ~azure.mgmt.storageactions.models.ErrorDetail - """ - - _attribute_map = { - "error": {"key": "error", "type": "ErrorDetail"}, - } - - def __init__(self, *, error: Optional["_models.ErrorDetail"] = None, **kwargs: Any) -> None: - """ - :keyword error: The error object. - :paramtype error: ~azure.mgmt.storageactions.models.ErrorDetail - """ - super().__init__(**kwargs) - self.error = error - - -class IfCondition(_serialization.Model): - """The if block of storage task operation. - - All required parameters must be populated in order to send to server. - - :ivar condition: Condition predicate to evaluate each object. See - https://aka.ms/storagetaskconditions for valid properties and operators. Required. - :vartype condition: str - :ivar operations: List of operations to execute when the condition predicate satisfies. - Required. - :vartype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] - """ - - _validation = { - "condition": {"required": True}, - "operations": {"required": True}, - } - - _attribute_map = { - "condition": {"key": "condition", "type": "str"}, - "operations": {"key": "operations", "type": "[StorageTaskOperation]"}, - } - - def __init__(self, *, condition: str, operations: List["_models.StorageTaskOperation"], **kwargs: Any) -> None: - """ - :keyword condition: Condition predicate to evaluate each object. See - https://aka.ms/storagetaskconditions for valid properties and operators. Required. - :paramtype condition: str - :keyword operations: List of operations to execute when the condition predicate satisfies. - Required. - :paramtype operations: list[~azure.mgmt.storageactions.models.StorageTaskOperation] - """ - super().__init__(**kwargs) - self.condition = condition - self.operations = operations - - -class ManagedServiceIdentity(_serialization.Model): - """Managed service identity (system assigned and/or user assigned identities). - - 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 principal_id: The service principal ID of the system assigned identity. This property - will only be provided for a system assigned identity. - :vartype principal_id: str - :ivar tenant_id: The tenant ID of the system assigned identity. This property will only be - provided for a system assigned identity. - :vartype tenant_id: str - :ivar type: Type of managed service identity (where both SystemAssigned and UserAssigned types - are allowed). Required. Known values are: "None", "SystemAssigned", "UserAssigned", and - "SystemAssigned,UserAssigned". - :vartype type: str or ~azure.mgmt.storageactions.models.ManagedServiceIdentityType - :ivar user_assigned_identities: The set of user assigned identities associated with the - resource. The userAssignedIdentities dictionary keys will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}. # pylint: disable=line-too-long - The dictionary values can be empty objects ({}) in requests. - :vartype user_assigned_identities: dict[str, - ~azure.mgmt.storageactions.models.UserAssignedIdentity] - """ - - _validation = { - "principal_id": {"readonly": True}, - "tenant_id": {"readonly": True}, - "type": {"required": True}, - } - - _attribute_map = { - "principal_id": {"key": "principalId", "type": "str"}, - "tenant_id": {"key": "tenantId", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "user_assigned_identities": {"key": "userAssignedIdentities", "type": "{UserAssignedIdentity}"}, - } - - def __init__( - self, - *, - type: Union[str, "_models.ManagedServiceIdentityType"], - user_assigned_identities: Optional[Dict[str, "_models.UserAssignedIdentity"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword type: Type of managed service identity (where both SystemAssigned and UserAssigned - types are allowed). Required. Known values are: "None", "SystemAssigned", "UserAssigned", and - "SystemAssigned,UserAssigned". - :paramtype type: str or ~azure.mgmt.storageactions.models.ManagedServiceIdentityType - :keyword user_assigned_identities: The set of user assigned identities associated with the - resource. The userAssignedIdentities dictionary keys will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}. # pylint: disable=line-too-long - The dictionary values can be empty objects ({}) in requests. - :paramtype user_assigned_identities: dict[str, - ~azure.mgmt.storageactions.models.UserAssignedIdentity] - """ - super().__init__(**kwargs) - self.principal_id = None - self.tenant_id = None - self.type = type - self.user_assigned_identities = user_assigned_identities - - -class Operation(_serialization.Model): - """Details of a REST API operation, returned from the Resource Provider Operations API. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: The name of the operation, as per Resource-Based Access Control (RBAC). Examples: - "Microsoft.Compute/virtualMachines/write", "Microsoft.Compute/virtualMachines/capture/action". - :vartype name: str - :ivar is_data_action: Whether the operation applies to data-plane. This is "true" for - data-plane operations and "false" for ARM/control-plane operations. - :vartype is_data_action: bool - :ivar display: Localized display information for this particular operation. - :vartype display: ~azure.mgmt.storageactions.models.OperationDisplay - :ivar origin: The intended executor of the operation; as in Resource Based Access Control - (RBAC) and audit logs UX. Default value is "user,system". Known values are: "user", "system", - and "user,system". - :vartype origin: str or ~azure.mgmt.storageactions.models.Origin - :ivar action_type: Enum. Indicates the action type. "Internal" refers to actions that are for - internal only APIs. "Internal" - :vartype action_type: str or ~azure.mgmt.storageactions.models.ActionType - """ - - _validation = { - "name": {"readonly": True}, - "is_data_action": {"readonly": True}, - "origin": {"readonly": True}, - "action_type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "is_data_action": {"key": "isDataAction", "type": "bool"}, - "display": {"key": "display", "type": "OperationDisplay"}, - "origin": {"key": "origin", "type": "str"}, - "action_type": {"key": "actionType", "type": "str"}, - } - - def __init__(self, *, display: Optional["_models.OperationDisplay"] = None, **kwargs: Any) -> None: - """ - :keyword display: Localized display information for this particular operation. - :paramtype display: ~azure.mgmt.storageactions.models.OperationDisplay - """ - super().__init__(**kwargs) - self.name = None - self.is_data_action = None - self.display = display - self.origin = None - self.action_type = None - - -class OperationDisplay(_serialization.Model): - """Localized display information for this particular operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provider: The localized friendly form of the resource provider name, e.g. "Microsoft - Monitoring Insights" or "Microsoft Compute". - :vartype provider: str - :ivar resource: The localized friendly name of the resource type related to this operation. - E.g. "Virtual Machines" or "Job Schedule Collections". - :vartype resource: str - :ivar operation: The concise, localized friendly name for the operation; suitable for - dropdowns. E.g. "Create or Update Virtual Machine", "Restart Virtual Machine". - :vartype operation: str - :ivar description: The short, localized friendly description of the operation; suitable for - tool tips and detailed views. - :vartype description: str - """ - - _validation = { - "provider": {"readonly": True}, - "resource": {"readonly": True}, - "operation": {"readonly": True}, - "description": {"readonly": True}, - } - - _attribute_map = { - "provider": {"key": "provider", "type": "str"}, - "resource": {"key": "resource", "type": "str"}, - "operation": {"key": "operation", "type": "str"}, - "description": {"key": "description", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.provider = None - self.resource = None - self.operation = None - self.description = None - - -class OperationListResult(_serialization.Model): - """A list of REST API operations supported by an Azure Resource Provider. It contains an URL link - to get the next set of results. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of operations supported by the resource provider. - :vartype value: list[~azure.mgmt.storageactions.models.Operation] - :ivar next_link: URL to get the next set of operation list results (if there are any). - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[Operation]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value = None - self.next_link = None - - -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}". # pylint: disable=line-too-long - :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.storageactions.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 = None - self.name = None - self.type = None - self.system_data = 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}". # pylint: disable=line-too-long - :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.storageactions.models.SystemData - """ - - -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}". # pylint: disable=line-too-long - :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.storageactions.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 StorageTask(TrackedResource): - """Represents Storage Task. - - 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}". # pylint: disable=line-too-long - :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.storageactions.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 identity: The managed service identity of the resource. Required. - :vartype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity - :ivar properties: Properties of the storage task. Required. - :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskProperties - """ - - _validation = { - "id": {"readonly": True}, - "name": {"readonly": True}, - "type": {"readonly": True}, - "system_data": {"readonly": True}, - "location": {"required": True}, - "identity": {"required": True}, - "properties": {"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"}, - "identity": {"key": "identity", "type": "ManagedServiceIdentity"}, - "properties": {"key": "properties", "type": "StorageTaskProperties"}, - } - - def __init__( - self, - *, - location: str, - identity: "_models.ManagedServiceIdentity", - properties: "_models.StorageTaskProperties", - 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 - :keyword identity: The managed service identity of the resource. Required. - :paramtype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity - :keyword properties: Properties of the storage task. Required. - :paramtype properties: ~azure.mgmt.storageactions.models.StorageTaskProperties - """ - super().__init__(tags=tags, location=location, **kwargs) - self.identity = identity - self.properties = properties - - -class StorageTaskAction(_serialization.Model): - """The storage task action represents conditional statements and operations to be performed on - target objects. - - All required parameters must be populated in order to send to server. - - :ivar if_property: The if block of storage task operation. Required. - :vartype if_property: ~azure.mgmt.storageactions.models.IfCondition - :ivar else_property: The else block of storage task operation. - :vartype else_property: ~azure.mgmt.storageactions.models.ElseCondition - """ - - _validation = { - "if_property": {"required": True}, - } - - _attribute_map = { - "if_property": {"key": "if", "type": "IfCondition"}, - "else_property": {"key": "else", "type": "ElseCondition"}, - } - - def __init__( - self, - *, - if_property: "_models.IfCondition", - else_property: Optional["_models.ElseCondition"] = None, - **kwargs: Any - ) -> None: - """ - :keyword if_property: The if block of storage task operation. Required. - :paramtype if_property: ~azure.mgmt.storageactions.models.IfCondition - :keyword else_property: The else block of storage task operation. - :paramtype else_property: ~azure.mgmt.storageactions.models.ElseCondition - """ - super().__init__(**kwargs) - self.if_property = if_property - self.else_property = else_property - - -class StorageTaskAssignment(_serialization.Model): - """Storage Task Assignment associated with this Storage Task. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: Resource ID of the Storage Task Assignment. - :vartype id: str - """ - - _validation = { - "id": {"readonly": True}, - } - - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.id = None - - -class StorageTaskAssignmentsListResult(_serialization.Model): - """The response from the List Storage Tasks operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: List of Storage Task Assignment Resource IDs associated with this Storage Task. - :vartype value: list[~azure.mgmt.storageactions.models.StorageTaskAssignment] - :ivar next_link: Request URL that can be used to query next page of Resource IDs. Returned when - total number of requested Resource IDs exceed maximum page size. - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[StorageTaskAssignment]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value = None - self.next_link = None - - -class StorageTaskOperation(_serialization.Model): - """Represents an operation to be performed on the object. - - All required parameters must be populated in order to send to server. - - :ivar name: The operation to be performed on the object. Required. Known values are: - "SetBlobTier", "SetBlobTags", "SetBlobImmutabilityPolicy", "SetBlobLegalHold", "SetBlobExpiry", - "DeleteBlob", and "UndeleteBlob". - :vartype name: str or ~azure.mgmt.storageactions.models.StorageTaskOperationName - :ivar parameters: Key-value parameters for the operation. - :vartype parameters: dict[str, str] - :ivar on_success: Action to be taken when the operation is successful for a object. Default - value is "continue". - :vartype on_success: str - :ivar on_failure: Action to be taken when the operation fails for a object. Default value is - "break". - :vartype on_failure: str - """ - - _validation = { - "name": {"required": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "parameters": {"key": "parameters", "type": "{str}"}, - "on_success": {"key": "onSuccess", "type": "str"}, - "on_failure": {"key": "onFailure", "type": "str"}, - } - - def __init__( - self, - *, - name: Union[str, "_models.StorageTaskOperationName"], - parameters: Optional[Dict[str, str]] = None, - on_success: Optional[Literal["continue"]] = None, - on_failure: Optional[Literal["break"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: The operation to be performed on the object. Required. Known values are: - "SetBlobTier", "SetBlobTags", "SetBlobImmutabilityPolicy", "SetBlobLegalHold", "SetBlobExpiry", - "DeleteBlob", and "UndeleteBlob". - :paramtype name: str or ~azure.mgmt.storageactions.models.StorageTaskOperationName - :keyword parameters: Key-value parameters for the operation. - :paramtype parameters: dict[str, str] - :keyword on_success: Action to be taken when the operation is successful for a object. Default - value is "continue". - :paramtype on_success: str - :keyword on_failure: Action to be taken when the operation fails for a object. Default value is - "break". - :paramtype on_failure: str - """ - super().__init__(**kwargs) - self.name = name - self.parameters = parameters - self.on_success = on_success - self.on_failure = on_failure - - -class StorageTaskPreviewAction(_serialization.Model): - """Storage Task Preview Action. - - All required parameters must be populated in order to send to server. - - :ivar properties: Properties of the storage task preview. Required. - :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionProperties - """ - - _validation = { - "properties": {"required": True}, - } - - _attribute_map = { - "properties": {"key": "properties", "type": "StorageTaskPreviewActionProperties"}, - } - - def __init__(self, *, properties: "_models.StorageTaskPreviewActionProperties", **kwargs: Any) -> None: - """ - :keyword properties: Properties of the storage task preview. Required. - :paramtype properties: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class StorageTaskPreviewActionCondition(_serialization.Model): - """Represents the storage task conditions to be tested for a match with container and blob - properties. - - All required parameters must be populated in order to send to server. - - :ivar if_property: The condition to be tested for a match with container and blob properties. - Required. - :vartype if_property: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionIfCondition - :ivar else_block_exists: Specify whether the else block is present in the condition. Required. - :vartype else_block_exists: bool - """ - - _validation = { - "if_property": {"required": True}, - "else_block_exists": {"required": True}, - } - - _attribute_map = { - "if_property": {"key": "if", "type": "StorageTaskPreviewActionIfCondition"}, - "else_block_exists": {"key": "elseBlockExists", "type": "bool"}, - } - - def __init__( - self, *, if_property: "_models.StorageTaskPreviewActionIfCondition", else_block_exists: bool, **kwargs: Any - ) -> None: - """ - :keyword if_property: The condition to be tested for a match with container and blob - properties. Required. - :paramtype if_property: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionIfCondition - :keyword else_block_exists: Specify whether the else block is present in the condition. - Required. - :paramtype else_block_exists: bool - """ - super().__init__(**kwargs) - self.if_property = if_property - self.else_block_exists = else_block_exists - - -class StorageTaskPreviewActionIfCondition(_serialization.Model): - """Represents storage task preview action condition. - - :ivar condition: Storage task condition to bes tested for a match. - :vartype condition: str - """ - - _attribute_map = { - "condition": {"key": "condition", "type": "str"}, - } - - def __init__(self, *, condition: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword condition: Storage task condition to bes tested for a match. - :paramtype condition: str - """ - super().__init__(**kwargs) - self.condition = condition - - -class StorageTaskPreviewActionProperties(_serialization.Model): - """Storage task preview action properties. - - All required parameters must be populated in order to send to server. - - :ivar container: Properties of a sample container to test for a match with the preview action. - Required. - :vartype container: ~azure.mgmt.storageactions.models.StorageTaskPreviewContainerProperties - :ivar blobs: Properties of some sample blobs in the container to test for matches with the - preview action. Required. - :vartype blobs: list[~azure.mgmt.storageactions.models.StorageTaskPreviewBlobProperties] - :ivar action: Preview action to test. Required. - :vartype action: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionCondition - """ - - _validation = { - "container": {"required": True}, - "blobs": {"required": True}, - "action": {"required": True}, - } - - _attribute_map = { - "container": {"key": "container", "type": "StorageTaskPreviewContainerProperties"}, - "blobs": {"key": "blobs", "type": "[StorageTaskPreviewBlobProperties]"}, - "action": {"key": "action", "type": "StorageTaskPreviewActionCondition"}, - } - - def __init__( - self, - *, - container: "_models.StorageTaskPreviewContainerProperties", - blobs: List["_models.StorageTaskPreviewBlobProperties"], - action: "_models.StorageTaskPreviewActionCondition", - **kwargs: Any - ) -> None: - """ - :keyword container: Properties of a sample container to test for a match with the preview - action. Required. - :paramtype container: ~azure.mgmt.storageactions.models.StorageTaskPreviewContainerProperties - :keyword blobs: Properties of some sample blobs in the container to test for matches with the - preview action. Required. - :paramtype blobs: list[~azure.mgmt.storageactions.models.StorageTaskPreviewBlobProperties] - :keyword action: Preview action to test. Required. - :paramtype action: ~azure.mgmt.storageactions.models.StorageTaskPreviewActionCondition - """ - super().__init__(**kwargs) - self.container = container - self.blobs = blobs - self.action = action - - -class StorageTaskPreviewBlobProperties(_serialization.Model): - """Storage task preview container properties. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: Name of test blob. - :vartype name: str - :ivar properties: properties key value pairs to be tested for a match against the provided - condition. - :vartype properties: - list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - :ivar metadata: metadata key value pairs to be tested for a match against the provided - condition. - :vartype metadata: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - :ivar tags: tags key value pairs to be tested for a match against the provided condition. - :vartype tags: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - :ivar matched_block: Represents the condition block name that matched blob properties. Known - values are: "If", "Else", "None", "If", and "Else". - :vartype matched_block: str or ~azure.mgmt.storageactions.models.MatchedBlockName - """ - - _validation = { - "matched_block": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "properties": {"key": "properties", "type": "[StorageTaskPreviewKeyValueProperties]"}, - "metadata": {"key": "metadata", "type": "[StorageTaskPreviewKeyValueProperties]"}, - "tags": {"key": "tags", "type": "[StorageTaskPreviewKeyValueProperties]"}, - "matched_block": {"key": "matchedBlock", "type": "str"}, - } - - def __init__( - self, - *, - name: Optional[str] = None, - properties: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, - metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, - tags: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: Name of test blob. - :paramtype name: str - :keyword properties: properties key value pairs to be tested for a match against the provided - condition. - :paramtype properties: - list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - :keyword metadata: metadata key value pairs to be tested for a match against the provided - condition. - :paramtype metadata: - list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - :keyword tags: tags key value pairs to be tested for a match against the provided condition. - :paramtype tags: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - """ - super().__init__(**kwargs) - self.name = name - self.properties = properties - self.metadata = metadata - self.tags = tags - self.matched_block = None - - -class StorageTaskPreviewContainerProperties(_serialization.Model): - """Storage task preview container properties. - - :ivar name: Name of test container. - :vartype name: str - :ivar metadata: metadata key value pairs to be tested for a match against the provided - condition. - :vartype metadata: list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - """ - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "metadata": {"key": "metadata", "type": "[StorageTaskPreviewKeyValueProperties]"}, - } - - def __init__( - self, - *, - name: Optional[str] = None, - metadata: Optional[List["_models.StorageTaskPreviewKeyValueProperties"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword name: Name of test container. - :paramtype name: str - :keyword metadata: metadata key value pairs to be tested for a match against the provided - condition. - :paramtype metadata: - list[~azure.mgmt.storageactions.models.StorageTaskPreviewKeyValueProperties] - """ - super().__init__(**kwargs) - self.name = name - self.metadata = metadata - - -class StorageTaskPreviewKeyValueProperties(_serialization.Model): - """Storage task preview object key value pair properties. - - :ivar key: Represents the key property of the pair. - :vartype key: str - :ivar value: Represents the value property of the pair. - :vartype value: str - """ - - _attribute_map = { - "key": {"key": "key", "type": "str"}, - "value": {"key": "value", "type": "str"}, - } - - def __init__(self, *, key: Optional[str] = None, value: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword key: Represents the key property of the pair. - :paramtype key: str - :keyword value: Represents the value property of the pair. - :paramtype value: str - """ - super().__init__(**kwargs) - self.key = key - self.value = value - - -class StorageTaskProperties(_serialization.Model): - """Properties of the storage task. - - 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 task_version: Storage task version. - :vartype task_version: int - :ivar enabled: Storage Task is enabled when set to true and disabled when set to false. - Required. - :vartype enabled: bool - :ivar description: Text that describes the purpose of the storage task. Required. - :vartype description: str - :ivar action: The storage task action that is executed. Required. - :vartype action: ~azure.mgmt.storageactions.models.StorageTaskAction - :ivar provisioning_state: Represents the provisioning state of the storage task. Known values - are: "ValidateSubscriptionQuotaBegin", "ValidateSubscriptionQuotaEnd", "Accepted", "Creating", - "Succeeded", "Deleting", "Canceled", and "Failed". - :vartype provisioning_state: str or ~azure.mgmt.storageactions.models.ProvisioningState - :ivar creation_time_in_utc: The creation date and time of the storage task in UTC. - :vartype creation_time_in_utc: ~datetime.datetime - """ - - _validation = { - "task_version": {"readonly": True, "maximum": 1, "minimum": 1}, - "enabled": {"required": True}, - "description": {"required": True}, - "action": {"required": True}, - "provisioning_state": {"readonly": True}, - "creation_time_in_utc": {"readonly": True}, - } - - _attribute_map = { - "task_version": {"key": "taskVersion", "type": "int"}, - "enabled": {"key": "enabled", "type": "bool"}, - "description": {"key": "description", "type": "str"}, - "action": {"key": "action", "type": "StorageTaskAction"}, - "provisioning_state": {"key": "provisioningState", "type": "str"}, - "creation_time_in_utc": {"key": "creationTimeInUtc", "type": "iso-8601"}, - } - - def __init__(self, *, enabled: bool, description: str, action: "_models.StorageTaskAction", **kwargs: Any) -> None: - """ - :keyword enabled: Storage Task is enabled when set to true and disabled when set to false. - Required. - :paramtype enabled: bool - :keyword description: Text that describes the purpose of the storage task. Required. - :paramtype description: str - :keyword action: The storage task action that is executed. Required. - :paramtype action: ~azure.mgmt.storageactions.models.StorageTaskAction - """ - super().__init__(**kwargs) - self.task_version = None - self.enabled = enabled - self.description = description - self.action = action - self.provisioning_state = None - self.creation_time_in_utc = None - - -class StorageTaskReportInstance(ProxyResource): - """Storage Tasks run report instance. - - 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}". # pylint: disable=line-too-long - :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.storageactions.models.SystemData - :ivar properties: Storage task execution report for a run instance. - :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskReportProperties - """ - - _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"}, - "properties": {"key": "properties", "type": "StorageTaskReportProperties"}, - } - - def __init__(self, *, properties: Optional["_models.StorageTaskReportProperties"] = None, **kwargs: Any) -> None: - """ - :keyword properties: Storage task execution report for a run instance. - :paramtype properties: ~azure.mgmt.storageactions.models.StorageTaskReportProperties - """ - super().__init__(**kwargs) - self.properties = properties - - -class StorageTaskReportProperties(_serialization.Model): - """Storage task execution report for a run instance. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar task_assignment_id: Resource ID of the Storage Task Assignment associated with this - reported run. - :vartype task_assignment_id: str - :ivar storage_account_id: Resource ID of the Storage Account where this reported run executed. - :vartype storage_account_id: str - :ivar start_time: Start time of the run instance. Filter options such as startTime gt - '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for - DateTime properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype start_time: str - :ivar finish_time: End time of the run instance. Filter options such as startTime gt - '2023-06-26T20:51:24.4494016Z' and other comparison operators can be used as described for - DateTime properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype finish_time: str - :ivar objects_targeted_count: Total number of objects that meet the condition as defined in the - storage task assignment execution context. Filter options such as objectsTargetedCount gt 50 - and other comparison operators can be used as described for Numerical properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype objects_targeted_count: str - :ivar objects_operated_on_count: Total number of objects that meet the storage tasks condition - and were operated upon. Filter options such as objectsOperatedOnCount ge 100 and other - comparison operators can be used as described for Numerical properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype objects_operated_on_count: str - :ivar object_failed_count: Total number of objects where task operation failed when was - attempted. Filter options such as objectFailedCount eq 0 and other comparison operators can be - used as described for Numerical properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype object_failed_count: str - :ivar objects_succeeded_count: Total number of objects where task operation succeeded when was - attempted.Filter options such as objectsSucceededCount gt 150 and other comparison operators - can be used as described for Numerical properties in - https://learn.microsoft.com/en-us/rest/api/storageservices/querying-tables-and-entities#supported-comparison-operators. # pylint: disable=line-too-long - :vartype objects_succeeded_count: str - :ivar run_status_error: Well known Azure Storage error code that represents the error - encountered during execution of the run instance. - :vartype run_status_error: str - :ivar run_status_enum: Represents the status of the execution. Known values are: "InProgress" - and "Finished". - :vartype run_status_enum: str or ~azure.mgmt.storageactions.models.RunStatusEnum - :ivar summary_report_path: Full path to the verbose report stored in the reporting container as - specified in the assignment execution context for the storage account. - :vartype summary_report_path: str - :ivar task_id: Resource ID of the Storage Task applied during this run. - :vartype task_id: str - :ivar task_version: Storage Task Version. - :vartype task_version: str - :ivar run_result: Represents the overall result of the execution for the run instance. Known - values are: "Succeeded" and "Failed". - :vartype run_result: str or ~azure.mgmt.storageactions.models.RunResult - """ - - _validation = { - "task_assignment_id": {"readonly": True}, - "storage_account_id": {"readonly": True}, - "start_time": {"readonly": True}, - "finish_time": {"readonly": True}, - "objects_targeted_count": {"readonly": True}, - "objects_operated_on_count": {"readonly": True}, - "object_failed_count": {"readonly": True}, - "objects_succeeded_count": {"readonly": True}, - "run_status_error": {"readonly": True}, - "run_status_enum": {"readonly": True}, - "summary_report_path": {"readonly": True}, - "task_id": {"readonly": True}, - "task_version": {"readonly": True}, - "run_result": {"readonly": True}, - } - - _attribute_map = { - "task_assignment_id": {"key": "taskAssignmentId", "type": "str"}, - "storage_account_id": {"key": "storageAccountId", "type": "str"}, - "start_time": {"key": "startTime", "type": "str"}, - "finish_time": {"key": "finishTime", "type": "str"}, - "objects_targeted_count": {"key": "objectsTargetedCount", "type": "str"}, - "objects_operated_on_count": {"key": "objectsOperatedOnCount", "type": "str"}, - "object_failed_count": {"key": "objectFailedCount", "type": "str"}, - "objects_succeeded_count": {"key": "objectsSucceededCount", "type": "str"}, - "run_status_error": {"key": "runStatusError", "type": "str"}, - "run_status_enum": {"key": "runStatusEnum", "type": "str"}, - "summary_report_path": {"key": "summaryReportPath", "type": "str"}, - "task_id": {"key": "taskId", "type": "str"}, - "task_version": {"key": "taskVersion", "type": "str"}, - "run_result": {"key": "runResult", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.task_assignment_id = None - self.storage_account_id = None - self.start_time = None - self.finish_time = None - self.objects_targeted_count = None - self.objects_operated_on_count = None - self.object_failed_count = None - self.objects_succeeded_count = None - self.run_status_error = None - self.run_status_enum = None - self.summary_report_path = None - self.task_id = None - self.task_version = None - self.run_result = None - - -class StorageTaskReportSummary(_serialization.Model): - """Fetch Storage Tasks Run Summary. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Gets storage tasks run result summary. - :vartype value: list[~azure.mgmt.storageactions.models.StorageTaskReportInstance] - :ivar next_link: Request URL that can be used to query next page of storage task run results - summary. Returned when the number of run instances and summary reports exceed maximum page - size. - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[StorageTaskReportInstance]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value = None - self.next_link = None - - -class StorageTasksListResult(_serialization.Model): - """The response from the List Storage Task operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Gets the list of storage tasks and their properties. - :vartype value: list[~azure.mgmt.storageactions.models.StorageTask] - :ivar next_link: Request URL that can be used to query next page of storage tasks. Returned - when total number of requested storage tasks exceed maximum page size. - :vartype next_link: str - """ - - _validation = { - "value": {"readonly": True}, - "next_link": {"readonly": True}, - } - - _attribute_map = { - "value": {"key": "value", "type": "[StorageTask]"}, - "next_link": {"key": "nextLink", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.value = None - self.next_link = None - - -class StorageTaskUpdateParameters(_serialization.Model): - """Parameters of the storage task update request. - - :ivar identity: The identity of the resource. - :vartype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity - :ivar tags: Gets or sets a list of key value pairs that describe the resource. These tags can - be used in viewing and grouping this resource (across resource groups). A maximum of 15 tags - can be provided for a resource. Each tag must have a key no greater in length than 128 - characters and a value no greater in length than 256 characters. - :vartype tags: dict[str, str] - :ivar properties: Properties of the storage task. - :vartype properties: ~azure.mgmt.storageactions.models.StorageTaskProperties - """ - - _attribute_map = { - "identity": {"key": "identity", "type": "ManagedServiceIdentity"}, - "tags": {"key": "tags", "type": "{str}"}, - "properties": {"key": "properties", "type": "StorageTaskProperties"}, - } - - def __init__( - self, - *, - identity: Optional["_models.ManagedServiceIdentity"] = None, - tags: Optional[Dict[str, str]] = None, - properties: Optional["_models.StorageTaskProperties"] = None, - **kwargs: Any - ) -> None: - """ - :keyword identity: The identity of the resource. - :paramtype identity: ~azure.mgmt.storageactions.models.ManagedServiceIdentity - :keyword tags: Gets or sets a list of key value pairs that describe the resource. These tags - can be used in viewing and grouping this resource (across resource groups). A maximum of 15 - tags can be provided for a resource. Each tag must have a key no greater in length than 128 - characters and a value no greater in length than 256 characters. - :paramtype tags: dict[str, str] - :keyword properties: Properties of the storage task. - :paramtype properties: ~azure.mgmt.storageactions.models.StorageTaskProperties - """ - super().__init__(**kwargs) - self.identity = identity - self.tags = tags - self.properties = properties - - -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.storageactions.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.storageactions.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.storageactions.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.storageactions.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 UserAssignedIdentity(_serialization.Model): - """User assigned identity properties. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar principal_id: The principal ID of the assigned identity. - :vartype principal_id: str - :ivar client_id: The client ID of the assigned identity. - :vartype client_id: str - """ - - _validation = { - "principal_id": {"readonly": True}, - "client_id": {"readonly": True}, - } - - _attribute_map = { - "principal_id": {"key": "principalId", "type": "str"}, - "client_id": {"key": "clientId", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - self.principal_id = None - self.client_id = None diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_patch.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_patch.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/models/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/__init__.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/__init__.py index 39b5abc1fb68..248b5bd5bf04 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/__init__.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/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 @@ -13,9 +13,8 @@ from ._patch import * # pylint: disable=unused-wildcard-import from ._operations import Operations # type: ignore -from ._storage_tasks_operations import StorageTasksOperations # type: ignore -from ._storage_task_assignment_operations import StorageTaskAssignmentOperations # type: ignore -from ._storage_tasks_report_operations import StorageTasksReportOperations # type: ignore +from ._operations import StorageTasksOperations # type: ignore +from ._operations import StorageTasksOperationGroupOperations # type: ignore from ._patch import __all__ as _patch_all from ._patch import * @@ -24,8 +23,7 @@ __all__ = [ "Operations", "StorageTasksOperations", - "StorageTaskAssignmentOperations", - "StorageTasksReportOperations", + "StorageTasksOperationGroupOperations", ] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_operations.py index c92f26cf52fb..2eed2bdcb54a 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_operations.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_operations.py @@ -1,44 +1,51 @@ +# 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. # -------------------------------------------------------------------------- -import sys -from typing import Any, Callable, Dict, Iterable, Optional, TypeVar +from collections.abc import MutableMapping +from io import IOBase +import json +from typing import Any, Callable, Dict, IO, Iterable, Iterator, List, Optional, TypeVar, Union, cast, overload import urllib.parse +from azure.core import PipelineClient from azure.core.exceptions import ( ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, map_error, ) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse +from azure.core.polling import LROPoller, NoPolling, PollingMethod 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 azure.mgmt.core.polling.arm_polling import ARMPolling from .. import models as _models -from .._serialization import Serializer +from .._configuration import StorageActionsClientConfiguration +from .._utils.model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize +from .._utils.serialization import Deserializer, Serializer -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +JSON = MutableMapping[str, Any] _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 {}) @@ -46,7 +53,7 @@ def build_list_request(**kwargs: Any) -> HttpRequest: accept = _headers.pop("Accept", "application/json") # Construct URL - _url = kwargs.pop("template_url", "/providers/Microsoft.StorageActions/operations") + _url = "/providers/Microsoft.StorageActions/operations" # Construct parameters _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") @@ -57,38 +64,309 @@ def build_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) +def build_storage_tasks_get_request( + resource_group_name: str, storage_task_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-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_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_storage_tasks_create_request( + resource_group_name: str, storage_task_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", "2023-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_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_storage_tasks_update_request( + resource_group_name: str, storage_task_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", "2023-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_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_storage_tasks_delete_request( + resource_group_name: str, storage_task_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-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_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) + + +def build_storage_tasks_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", "2023-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks" + 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_storage_tasks_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", "2023-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.StorageActions/storageTasks" + 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_storage_tasks_list_request( + resource_group_name: str, + storage_task_name: str, + subscription_id: str, + *, + maxpagesize: Optional[int] = None, + filter: 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-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}/reports" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if maxpagesize is not None: + _params["$maxpagesize"] = _SERIALIZER.query("maxpagesize", maxpagesize, "int") + if filter is not None: + _params["$filter"] = _SERIALIZER.query("filter", filter, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_storage_tasks_storage_task_assignment_list_request( # pylint: disable=name-too-long + resource_group_name: str, + storage_task_name: str, + subscription_id: str, + *, + maxpagesize: Optional[int] = 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-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}/storageTaskAssignments" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "storageTaskName": _SERIALIZER.url("storage_task_name", storage_task_name, "str"), + } + + _url: str = _url.format(**path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + if maxpagesize is not None: + _params["$maxpagesize"] = _SERIALIZER.query("maxpagesize", maxpagesize, "int") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_storage_tasks_operation_group_preview_actions_request( # pylint: disable=name-too-long + location: 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", "2023-01-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = "/subscriptions/{subscriptionId}/providers/Microsoft.StorageActions/locations/{location}/previewActions" + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "location": _SERIALIZER.url("location", location, "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="POST", url=_url, params=_params, headers=_headers, **kwargs) + + class Operations: """ .. warning:: **DO NOT** instantiate this class directly. Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.StorageActionsMgmtClient`'s + :class:`~azure.mgmt.storageactions.StorageActionsClient`'s :attr:`operations` attribute. """ - models = _models - - def __init__(self, *args, **kwargs): + def __init__(self, *args, **kwargs) -> None: input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") + self._client: PipelineClient = input_args.pop(0) if input_args else kwargs.pop("client") + self._config: StorageActionsClientConfiguration = 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) -> Iterable["_models.Operation"]: """Lists all of the available Storage Actions Rest API operations. - :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.storageactions.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 +379,17 @@ def list(self, **kwargs: Any) -> Iterable["_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 @@ -121,16 +404,21 @@ def prepare_request(next_link=None): _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - _request.url = self._client.format_url(_request.url) - _request.method = "GET" + path_format_arguments = { + "endpoint": self._serialize.url( + "self._config.base_url", self._config.base_url, "str", skip_quote=True + ), + } + _request.url = self._client.format_url(_request.url, **path_format_arguments) + return _request def extract_data(pipeline_response): - deserialized = self._deserialize("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,9 +431,1231 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response return ItemPaged(get_next, extract_data) + + +class StorageTasksOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.storageactions.StorageActionsClient`'s + :attr:`storage_tasks` 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: StorageActionsClientConfiguration = 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, storage_task_name: str, **kwargs: Any) -> _models.StorageTask: + """Get the storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: StorageTask. The StorageTask is compatible with MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTask + :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.StorageTask] = kwargs.pop("cls", None) + + _request = build_storage_tasks_get_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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 = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.StorageTask, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore + + def _create_initial( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTask, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_storage_tasks_create_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 201, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 201: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: _models.StorageTask, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTask + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. 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: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. 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: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTask, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[None]: + """Asynchronously creates a new storage task resource with the specified parameters. If a storage + task is already created and a subsequent create request is issued with different properties, + the storage task properties will be updated. If a storage task is already created and a + subsequent create or update request is issued with the exact same set of properties, the + request will succeed. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to create a Storage Task. Is one of the following types: + StorageTask, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTask or JSON or IO[bytes] + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._create_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + parameters=parameters, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + def _update_initial( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTaskUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = kwargs.pop("params", {}) or {} + + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _content = None + if isinstance(parameters, (IOBase, bytes)): + _content = parameters + else: + _content = json.dumps(parameters, cls=SdkJSONEncoder, exclude_readonly=True) # type: ignore + + _request = build_storage_tasks_update_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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) + + _stream = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200, 202]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @overload + def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: _models.StorageTaskUpdateParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: An instance of LROPoller that returns StorageTask. The StorageTask is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: JSON, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. 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: An instance of LROPoller that returns StorageTask. The StorageTask is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: IO[bytes], + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. 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: An instance of LROPoller that returns StorageTask. The StorageTask is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + storage_task_name: str, + parameters: Union[_models.StorageTaskUpdateParameters, JSON, IO[bytes]], + **kwargs: Any + ) -> LROPoller[_models.StorageTask]: + """Update storage task properties. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :param parameters: The parameters to provide to update the storage task resource. Is one of the + following types: StorageTaskUpdateParameters, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters or JSON or + IO[bytes] + :return: An instance of LROPoller that returns StorageTask. The StorageTask is compatible with + MutableMapping + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _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.StorageTask] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._update_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + parameters=parameters, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): + response = pipeline_response.http_response + deserialized = _deserialize(_models.StorageTask, response.json()) + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + return deserialized + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[_models.StorageTask].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[_models.StorageTask]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) + + def _delete_initial(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> Iterator[bytes]: + error_map: MutableMapping = { + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, + } + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + + _request = build_storage_tasks_delete_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_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 = True + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [202, 204]: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + response_headers = {} + if response.status_code == 202: + response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) + response_headers["Retry-After"] = self._deserialize("int", response.headers.get("Retry-After")) + + deserialized = response.iter_bytes() + + if cls: + return cls(pipeline_response, deserialized, response_headers) # type: ignore + + return deserialized # type: ignore + + @distributed_trace + def begin_delete(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> LROPoller[None]: + """Delete the storage task resource. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: An instance of LROPoller that returns None + :rtype: ~azure.core.polling.LROPoller[None] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[None] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) + if cont_token is None: + raw_result = self._delete_initial( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, + **kwargs + ) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) + + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements + if cls: + return cls(pipeline_response, None, {}) # type: ignore + + path_format_arguments = { + "endpoint": self._serialize.url("self._config.base_url", self._config.base_url, "str", skip_quote=True), + } + + if polling is True: + polling_method: PollingMethod = cast( + PollingMethod, ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) + ) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling + if cont_token: + return LROPoller[None].from_continuation_token( + polling_method=polling_method, + continuation_token=cont_token, + client=self._client, + deserialization_callback=get_long_running_output, + ) + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore + + @distributed_trace + def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.StorageTask"]: + """Lists all the storage tasks available under the given resource group. + + :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 StorageTask + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.StorageTask]] = 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_storage_tasks_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), 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.StorageTask], 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.ErrorResponse, response.json()) + 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) -> Iterable["_models.StorageTask"]: + """Lists all the storage tasks available under the subscription. + + :return: An iterator like instance of StorageTask + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTask] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + cls: ClsType[List[_models.StorageTask]] = 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_storage_tasks_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), 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.StorageTask], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def list( + self, resource_group_name: str, storage_task_name: str, *, filter: Optional[str] = None, **kwargs: Any + ) -> Iterable["_models.StorageTaskReportInstance"]: + """Fetch the storage tasks run report summary for each assignment. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :keyword filter: Optional. When specified, it can be used to query using reporting properties. + Default value is None. + :paramtype filter: str + :return: An iterator like instance of StorageTaskReportInstance + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTaskReportInstance] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + maxpagesize = kwargs.pop("maxpagesize", None) + cls: ClsType[List[_models.StorageTaskReportInstance]] = 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_storage_tasks_list_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + subscription_id=self._config.subscription_id, + maxpagesize=maxpagesize, + filter=filter, + 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), 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.StorageTaskReportInstance], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + @distributed_trace + def storage_task_assignment_list( + self, resource_group_name: str, storage_task_name: str, **kwargs: Any + ) -> Iterable["_models.StorageTaskAssignment"]: + """Lists Resource IDs of the Storage Task Assignments associated with this Storage Task. + + :param resource_group_name: The name of the resource group. The name is case insensitive. + Required. + :type resource_group_name: str + :param storage_task_name: The name of the storage task within the specified resource group. + Storage task names must be between 3 and 18 characters in length and use numbers and lower-case + letters only. Required. + :type storage_task_name: str + :return: An iterator like instance of StorageTaskAssignment + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTaskAssignment] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = kwargs.pop("headers", {}) or {} + _params = kwargs.pop("params", {}) or {} + + maxpagesize = kwargs.pop("maxpagesize", None) + cls: ClsType[List[_models.StorageTaskAssignment]] = 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_storage_tasks_storage_task_assignment_list_request( + resource_group_name=resource_group_name, + storage_task_name=storage_task_name, + subscription_id=self._config.subscription_id, + maxpagesize=maxpagesize, + 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), 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.StorageTaskAssignment], 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.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + return pipeline_response + + return ItemPaged(get_next, extract_data) + + +class StorageTasksOperationGroupOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.storageactions.StorageActionsClient`'s + :attr:`storage_tasks_operation_group` 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: StorageActionsClientConfiguration = 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") + + @overload + def preview_actions( + self, + location: str, + parameters: _models.StorageTaskPreviewAction, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :return: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def preview_actions( + self, location: str, parameters: JSON, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def preview_actions( + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def preview_actions( + self, location: str, parameters: Union[_models.StorageTaskPreviewAction, JSON, IO[bytes]], **kwargs: Any + ) -> _models.StorageTaskPreviewAction: + """Runs the input conditions against input object metadata properties and designates matched + objects in response. + + :param location: Required. + :type location: str + :param parameters: The parameters to preview action condition. Is one of the following types: + StorageTaskPreviewAction, JSON, IO[bytes] Required. + :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction or JSON or + IO[bytes] + :return: StorageTaskPreviewAction. The StorageTaskPreviewAction is compatible with + MutableMapping + :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction + :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.StorageTaskPreviewAction] = 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_storage_tasks_operation_group_preview_actions_request( + location=location, + 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) + + _stream = kwargs.pop("stream", False) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs + ) + + response = pipeline_response.http_response + + if response.status_code not in [200]: + if _stream: + try: + response.read() # Load the body in memory and close the socket + except (StreamConsumedError, StreamClosedError): + pass + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) + raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) + + if _stream: + deserialized = response.iter_bytes() + else: + deserialized = _deserialize(_models.StorageTaskPreviewAction, response.json()) + + if cls: + return cls(pipeline_response, deserialized, {}) # type: ignore + + return deserialized # type: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_patch.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_patch.py index f7dd32510333..8bcb627aa475 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_patch.py +++ b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_patch.py @@ -1,7 +1,8 @@ -# ------------------------------------ -# Copyright (c) Microsoft Corporation. -# Licensed under the MIT License. -# ------------------------------------ +# 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. +# -------------------------------------------------------------------------- """Customize generated code here. Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_task_assignment_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_task_assignment_operations.py deleted file mode 100644 index 7bb2ba5b96f8..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_task_assignment_operations.py +++ /dev/null @@ -1,191 +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. -# -------------------------------------------------------------------------- -import sys -from typing import Any, Callable, Dict, Iterable, Optional, TypeVar -import urllib.parse - -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 .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -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, - storage_task_name: str, - subscription_id: str, - *, - maxpagesize: Optional[int] = 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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}/storageTaskAssignments", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - if maxpagesize is not None: - _params["$maxpagesize"] = _SERIALIZER.query("maxpagesize", maxpagesize, "int") - _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 StorageTaskAssignmentOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.StorageActionsMgmtClient`'s - :attr:`storage_task_assignment` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, resource_group_name: str, storage_task_name: str, maxpagesize: Optional[int] = None, **kwargs: Any - ) -> Iterable["_models.StorageTaskAssignment"]: - """Lists Resource IDs of the Storage Task Assignments associated with this Storage Task. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param maxpagesize: Optional, specifies the maximum number of Storage Task Assignment Resource - IDs to be included in the list response. Default value is None. - :type maxpagesize: int - :return: An iterator like instance of either StorageTaskAssignment or the result of - cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTaskAssignment] - :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.StorageTaskAssignmentsListResult] = 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, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - maxpagesize=maxpagesize, - 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("StorageTaskAssignmentsListResult", 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_operations.py deleted file mode 100644 index edf963daf1e9..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_operations.py +++ /dev/null @@ -1,1152 +0,0 @@ -# pylint: disable=too-many-lines -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from io import IOBase -import sys -from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload -import urllib.parse - -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) -from azure.core.paging import ItemPaged -from azure.core.pipeline import PipelineResponse -from azure.core.polling import LROPoller, NoPolling, PollingMethod -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 azure.mgmt.core.polling.arm_polling import ARMPolling - -from .. import models as _models -from .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -T = TypeVar("T") -ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - -_SERIALIZER = Serializer() -_SERIALIZER.client_side_validation = False - - -def build_create_request( - resource_group_name: str, storage_task_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-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "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 - 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_delete_request( - resource_group_name: str, storage_task_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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "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="DELETE", url=_url, params=_params, headers=_headers, **kwargs) - - -def build_get_request( - resource_group_name: str, storage_task_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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "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_update_request( - resource_group_name: str, storage_task_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-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "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 - 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_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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.StorageActions/storageTasks") - 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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "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_preview_actions_request(location: 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-01")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/providers/Microsoft.StorageActions/locations/{location}/previewActions", - ) # pylint: disable=line-too-long - path_format_arguments = { - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - "location": _SERIALIZER.url("location", location, "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="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class StorageTasksOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.StorageActionsMgmtClient`'s - :attr:`storage_tasks` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - def _create_initial( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTask, IO[bytes]], - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = 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, "StorageTask") - - _request = build_create_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = 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, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: _models.StorageTask, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTask - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. 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: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_create( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTask, IO[bytes]], - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Asynchronously creates a new storage task resource with the specified parameters. If a storage - task is already created and a subsequent create request is issued with different properties, - the storage task properties will be updated. If a storage task is already created and a - subsequent create or update request is issued with the exact same set of properties, the - request will succeed. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to create a Storage Task. Is either a StorageTask type or a - IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTask or IO[bytes] - :return: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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.StorageTask] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._create_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - parameters=parameters, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.StorageTask].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.StorageTask]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - def _delete_initial(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> Iterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - - _request = build_delete_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [202, 204]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @distributed_trace - def begin_delete(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> LROPoller[None]: - """Delete the storage task resource. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :return: An instance of LROPoller that returns either None or the result of cls(response) - :rtype: ~azure.core.polling.LROPoller[None] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _headers = kwargs.pop("headers", {}) or {} - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - cls: ClsType[None] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._delete_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - api_version=api_version, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements - if cls: - return cls(pipeline_response, None, {}) # type: ignore - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[None].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - @distributed_trace - def get(self, resource_group_name: str, storage_task_name: str, **kwargs: Any) -> _models.StorageTask: - """Get the storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :return: StorageTask or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTask - :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.StorageTask] = kwargs.pop("cls", None) - - _request = build_get_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore - - def _update_initial( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTaskUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> Iterator[bytes]: - error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, - } - error_map.update(kwargs.pop("error_map", {}) or {}) - - _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version)) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = 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, "StorageTaskUpdateParameters") - - _request = build_update_request( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - api_version=api_version, - content_type=content_type, - json=_json, - content=_content, - headers=_headers, - params=_params, - ) - _request.url = self._client.format_url(_request.url) - - _decompress = kwargs.pop("decompress", True) - _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs - ) - - response = pipeline_response.http_response - - if response.status_code not in [200, 202]: - try: - response.read() # Load the body in memory and close the socket - except (StreamConsumedError, StreamClosedError): - pass - map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - response_headers = {} - if response.status_code == 202: - response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) - - deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) - - if cls: - return cls(pipeline_response, deserialized, response_headers) # type: ignore - - return deserialized # type: ignore - - @overload - def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: _models.StorageTaskUpdateParameters, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: IO[bytes], - *, - content_type: str = "application/json", - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. 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: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def begin_update( - self, - resource_group_name: str, - storage_task_name: str, - parameters: Union[_models.StorageTaskUpdateParameters, IO[bytes]], - **kwargs: Any - ) -> LROPoller[_models.StorageTask]: - """Update storage task properties. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param parameters: The parameters to provide to update the storage task resource. Is either a - StorageTaskUpdateParameters type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskUpdateParameters or IO[bytes] - :return: An instance of LROPoller that returns either StorageTask or the result of - cls(response) - :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.storageactions.models.StorageTask] - :raises ~azure.core.exceptions.HttpResponseError: - """ - _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.StorageTask] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) - if cont_token is None: - raw_result = self._update_initial( - resource_group_name=resource_group_name, - storage_task_name=storage_task_name, - parameters=parameters, - api_version=api_version, - content_type=content_type, - cls=lambda x, y, z: x, - headers=_headers, - params=_params, - **kwargs - ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) - - def get_long_running_output(pipeline_response): - deserialized = self._deserialize("StorageTask", pipeline_response.http_response) - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - if polling is True: - polling_method: PollingMethod = cast( - PollingMethod, ARMPolling(lro_delay, lro_options={"final-state-via": "location"}, **kwargs) - ) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling - if cont_token: - return LROPoller[_models.StorageTask].from_continuation_token( - polling_method=polling_method, - continuation_token=cont_token, - client=self._client, - deserialization_callback=get_long_running_output, - ) - return LROPoller[_models.StorageTask]( - self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - @distributed_trace - def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.StorageTask"]: - """Lists all the storage tasks available under the subscription. - - :return: An iterator like instance of either StorageTask or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTask] - :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.StorageTasksListResult] = 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("StorageTasksListResult", 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, 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) -> Iterable["_models.StorageTask"]: - """Lists all the storage tasks available under the given resource group. - - :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 either StorageTask or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTask] - :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.StorageTasksListResult] = 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("StorageTasksListResult", 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) - - @overload - def preview_actions( - self, - location: str, - parameters: _models.StorageTaskPreviewAction, - *, - content_type: str = "application/json", - **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. - Default value is "application/json". - :paramtype content_type: str - :return: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @overload - def preview_actions( - self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. 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: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :raises ~azure.core.exceptions.HttpResponseError: - """ - - @distributed_trace - def preview_actions( - self, location: str, parameters: Union[_models.StorageTaskPreviewAction, IO[bytes]], **kwargs: Any - ) -> _models.StorageTaskPreviewAction: - """Runs the input conditions against input object metadata properties and designates matched - objects in response. - - :param location: The location to perform preview of the actions. Required. - :type location: str - :param parameters: The parameters to preview action condition. Is either a - StorageTaskPreviewAction type or a IO[bytes] type. Required. - :type parameters: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction or IO[bytes] - :return: StorageTaskPreviewAction or the result of cls(response) - :rtype: ~azure.mgmt.storageactions.models.StorageTaskPreviewAction - :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.StorageTaskPreviewAction] = 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, "StorageTaskPreviewAction") - - _request = build_preview_actions_request( - location=location, - 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - deserialized = self._deserialize("StorageTaskPreviewAction", pipeline_response.http_response) - - if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore - - return deserialized # type: ignore diff --git a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_report_operations.py b/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_report_operations.py deleted file mode 100644 index bc98e5df2a1b..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/azure/mgmt/storageactions/operations/_storage_tasks_report_operations.py +++ /dev/null @@ -1,204 +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. -# -------------------------------------------------------------------------- -import sys -from typing import Any, Callable, Dict, Iterable, Optional, TypeVar -import urllib.parse - -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 .._serialization import Serializer - -if sys.version_info >= (3, 9): - from collections.abc import MutableMapping -else: - from typing import MutableMapping # type: ignore -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, - storage_task_name: str, - subscription_id: str, - *, - maxpagesize: Optional[int] = None, - filter: 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-01")) - accept = _headers.pop("Accept", "application/json") - - # Construct URL - _url = kwargs.pop( - "template_url", - "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StorageActions/storageTasks/{storageTaskName}/reports", - ) # pylint: disable=line-too-long - path_format_arguments = { - "resourceGroupName": _SERIALIZER.url( - "resource_group_name", resource_group_name, "str", max_length=90, min_length=1 - ), - "storageTaskName": _SERIALIZER.url( - "storage_task_name", storage_task_name, "str", max_length=18, min_length=3, pattern=r"^[a-z0-9]{3,18}$" - ), - "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), - } - - _url: str = _url.format(**path_format_arguments) # type: ignore - - # Construct parameters - if maxpagesize is not None: - _params["$maxpagesize"] = _SERIALIZER.query("maxpagesize", maxpagesize, "int") - if filter is not None: - _params["$filter"] = _SERIALIZER.query("filter", filter, "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) - - -class StorageTasksReportOperations: - """ - .. warning:: - **DO NOT** instantiate this class directly. - - Instead, you should access the following operations through - :class:`~azure.mgmt.storageactions.StorageActionsMgmtClient`'s - :attr:`storage_tasks_report` attribute. - """ - - models = _models - - def __init__(self, *args, **kwargs): - input_args = list(args) - self._client = input_args.pop(0) if input_args else kwargs.pop("client") - self._config = input_args.pop(0) if input_args else kwargs.pop("config") - self._serialize = input_args.pop(0) if input_args else kwargs.pop("serializer") - self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") - - @distributed_trace - def list( - self, - resource_group_name: str, - storage_task_name: str, - maxpagesize: Optional[int] = None, - filter: Optional[str] = None, - **kwargs: Any - ) -> Iterable["_models.StorageTaskReportInstance"]: - """Fetch the storage tasks run report summary for each assignment. - - :param resource_group_name: The name of the resource group. The name is case insensitive. - Required. - :type resource_group_name: str - :param storage_task_name: The name of the storage task within the specified resource group. - Storage task names must be between 3 and 18 characters in length and use numbers and lower-case - letters only. Required. - :type storage_task_name: str - :param maxpagesize: Optional, specifies the maximum number of Storage Task Assignment Resource - IDs to be included in the list response. Default value is None. - :type maxpagesize: int - :param filter: Optional. When specified, it can be used to query using reporting properties. - Default value is None. - :type filter: str - :return: An iterator like instance of either StorageTaskReportInstance or the result of - cls(response) - :rtype: - ~azure.core.paging.ItemPaged[~azure.mgmt.storageactions.models.StorageTaskReportInstance] - :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.StorageTaskReportSummary] = 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, - storage_task_name=storage_task_name, - subscription_id=self._config.subscription_id, - maxpagesize=maxpagesize, - filter=filter, - 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("StorageTaskReportSummary", 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) - error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response) - raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - - return pipeline_response - - return ItemPaged(get_next, extract_data) diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/operations_list.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/operations_list.py deleted file mode 100644 index 1ef1f23bc1aa..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/operations_list.py +++ /dev/null @@ -1,40 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python operations_list.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="SUBSCRIPTION_ID", - ) - - response = client.operations.list() - for item in response: - print(item) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/misc/OperationsList.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/perform_storage_task_actions_preview.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/perform_storage_task_actions_preview.py deleted file mode 100644 index 4826042fdfe9..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/misc/perform_storage_task_actions_preview.py +++ /dev/null @@ -1,87 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python perform_storage_task_actions_preview.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.preview_actions( - location="eastus", - parameters={ - "properties": { - "action": {"elseBlockExists": True, "if": {"condition": "[[equals(AccessTier, 'Hot')]]"}}, - "blobs": [ - { - "metadata": [{"key": "mKey1", "value": "mValue1"}], - "name": "folder1/file1.txt", - "properties": [ - {"key": "Creation-Time", "value": "Wed, 07 Jun 2023 05:23:29 GMT"}, - {"key": "Last-Modified", "value": "Wed, 07 Jun 2023 05:23:29 GMT"}, - {"key": "Etag", "value": "0x8DB67175454D36D"}, - {"key": "Content-Length", "value": "38619"}, - {"key": "Content-Type", "value": "text/xml"}, - {"key": "Content-Encoding", "value": ""}, - {"key": "Content-Language", "value": ""}, - {"key": "Content-CRC64", "value": ""}, - {"key": "Content-MD5", "value": "njr6iDrmU9+FC89WMK22EA=="}, - {"key": "Cache-Control", "value": ""}, - {"key": "Content-Disposition", "value": ""}, - {"key": "BlobType", "value": "BlockBlob"}, - {"key": "AccessTier", "value": "Hot"}, - {"key": "AccessTierInferred", "value": "true"}, - {"key": "LeaseStatus", "value": "unlocked"}, - {"key": "LeaseState", "value": "available"}, - {"key": "ServerEncrypted", "value": "true"}, - {"key": "TagCount", "value": "1"}, - ], - "tags": [{"key": "tKey1", "value": "tValue1"}], - }, - { - "metadata": [{"key": "mKey2", "value": "mValue2"}], - "name": "folder2/file1.txt", - "properties": [ - {"key": "Creation-Time", "value": "Wed, 06 Jun 2023 05:23:29 GMT"}, - {"key": "Last-Modified", "value": "Wed, 06 Jun 2023 05:23:29 GMT"}, - {"key": "Etag", "value": "0x6FB67175454D36D"}, - ], - "tags": [{"key": "tKey2", "value": "tValue2"}], - }, - ], - "container": { - "metadata": [{"key": "mContainerKey1", "value": "mContainerValue1"}], - "name": "firstContainer", - }, - } - }, - ) - print(response) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/misc/PerformStorageTaskActionsPreview.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/delete_storage_task.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/delete_storage_task.py deleted file mode 100644 index 8a3e06e5fb14..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/delete_storage_task.py +++ /dev/null @@ -1,41 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python delete_storage_task.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - client.storage_tasks.begin_delete( - resource_group_name="res4228", - storage_task_name="mytask1", - ).result() - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksCrud/DeleteStorageTask.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/get_storage_task.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/get_storage_task.py deleted file mode 100644 index 7389d0676c08..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/get_storage_task.py +++ /dev/null @@ -1,42 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python get_storage_task.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.get( - resource_group_name="res4228", - storage_task_name="mytask1", - ) - print(response) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksCrud/GetStorageTask.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/patch_storage_task.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/patch_storage_task.py deleted file mode 100644 index 57bb0965c4d0..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/patch_storage_task.py +++ /dev/null @@ -1,68 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python patch_storage_task.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.begin_update( - resource_group_name="res4228", - storage_task_name="mytask1", - parameters={ - "identity": { - "type": "UserAssigned", - "userAssignedIdentities": { - "/subscriptions/1f31ba14-ce16-4281-b9b4-3e78da6e1616/resourceGroups/res4228/providers/Microsoft.ManagedIdentity/userAssignedIdentities/myUserAssignedIdentity": {} - }, - }, - "properties": { - "action": { - "else": {"operations": [{"name": "DeleteBlob", "onFailure": "break", "onSuccess": "continue"}]}, - "if": { - "condition": "[[equals(AccessTier, 'Cool')]]", - "operations": [ - { - "name": "SetBlobTier", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"tier": "Hot"}, - } - ], - }, - }, - "description": "My Storage task", - "enabled": True, - }, - }, - ).result() - print(response) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksCrud/PatchStorageTask.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/put_storage_task.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/put_storage_task.py deleted file mode 100644 index a538cb2640e7..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_crud/put_storage_task.py +++ /dev/null @@ -1,64 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python put_storage_task.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.begin_create( - resource_group_name="res4228", - storage_task_name="mytask1", - parameters={ - "identity": {"type": "SystemAssigned"}, - "location": "westus", - "properties": { - "action": { - "else": {"operations": [{"name": "DeleteBlob", "onFailure": "break", "onSuccess": "continue"}]}, - "if": { - "condition": "[[equals(AccessTier, 'Cool')]]", - "operations": [ - { - "name": "SetBlobTier", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"tier": "Hot"}, - } - ], - }, - }, - "description": "My Storage task", - "enabled": True, - }, - }, - ).result() - print(response) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksCrud/PutStorageTask.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_task_assignment_ids.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_task_assignment_ids.py deleted file mode 100644 index d35a03e40042..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_task_assignment_ids.py +++ /dev/null @@ -1,43 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python list_storage_task_assignment_ids.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_task_assignment.list( - resource_group_name="rgroup1", - storage_task_name="mytask1", - ) - for item in response: - print(item) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksList/ListStorageTaskAssignmentIds.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_resource_group.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_resource_group.py deleted file mode 100644 index 2c909edd983f..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_resource_group.py +++ /dev/null @@ -1,42 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python list_storage_tasks_by_resource_group.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.list_by_resource_group( - resource_group_name="res6117", - ) - for item in response: - print(item) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksList/ListStorageTasksByResourceGroup.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_subscription.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_subscription.py deleted file mode 100644 index 515e06af5edb..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_by_subscription.py +++ /dev/null @@ -1,40 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python list_storage_tasks_by_subscription.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks.list_by_subscription() - for item in response: - print(item) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksList/ListStorageTasksBySubscription.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_run_report_summary.py b/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_run_report_summary.py deleted file mode 100644 index 16d6999e0028..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_samples/storage_tasks_list/list_storage_tasks_run_report_summary.py +++ /dev/null @@ -1,43 +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 azure.identity import DefaultAzureCredential - -from azure.mgmt.storageactions import StorageActionsMgmtClient - -""" -# PREREQUISITES - pip install azure-identity - pip install azure-mgmt-storageactions -# USAGE - python list_storage_tasks_run_report_summary.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 = StorageActionsMgmtClient( - credential=DefaultAzureCredential(), - subscription_id="1f31ba14-ce16-4281-b9b4-3e78da6e1616", - ) - - response = client.storage_tasks_report.list( - resource_group_name="rgroup1", - storage_task_name="mytask1", - ) - for item in response: - print(item) - - -# x-ms-original-file: specification/storageactions/resource-manager/Microsoft.StorageActions/stable/2023-01-01/examples/storageTasksList/ListStorageTasksRunReportSummary.json -if __name__ == "__main__": - main() diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/conftest.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/conftest.py index 8bb755e68fc6..549e18e8addf 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/conftest.py +++ b/sdk/storageactions/azure-mgmt-storageactions/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,14 +21,14 @@ # For security, please avoid record sensitive identity information in recordings @pytest.fixture(scope="session", autouse=True) def add_sanitizers(test_proxy): - storageactionsmgmt_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - storageactionsmgmt_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - storageactionsmgmt_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - storageactionsmgmt_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=storageactionsmgmt_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=storageactionsmgmt_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=storageactionsmgmt_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=storageactionsmgmt_client_secret, value="00000000-0000-0000-0000-000000000000") + storageactions_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + storageactions_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + storageactions_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + storageactions_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=storageactions_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=storageactions_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=storageactions_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=storageactions_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/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations.py deleted file mode 100644 index d30c60a0dde7..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations.py +++ /dev/null @@ -1,31 +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. -# -------------------------------------------------------------------------- -import pytest -from azure.mgmt.storageactions import StorageActionsMgmtClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTaskAssignmentOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_storage_task_assignment_list(self, resource_group): - response = self.client.storage_task_assignment.list( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations_async.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations_async.py deleted file mode 100644 index a1ace465bc2a..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_task_assignment_operations_async.py +++ /dev/null @@ -1,32 +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. -# -------------------------------------------------------------------------- -import pytest -from azure.mgmt.storageactions.aio import StorageActionsMgmtClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTaskAssignmentOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_storage_task_assignment_list(self, resource_group): - response = self.client.storage_task_assignment.list( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations.py deleted file mode 100644 index b9fe8f0e218d..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations.py +++ /dev/null @@ -1,31 +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. -# -------------------------------------------------------------------------- -import pytest -from azure.mgmt.storageactions import StorageActionsMgmtClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTasksReportOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_storage_tasks_report_list(self, resource_group): - response = self.client.storage_tasks_report.list( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations_async.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations_async.py deleted file mode 100644 index b198da9ddc59..000000000000 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_report_operations_async.py +++ /dev/null @@ -1,32 +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. -# -------------------------------------------------------------------------- -import pytest -from azure.mgmt.storageactions.aio import StorageActionsMgmtClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTasksReportOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_storage_tasks_report_list(self, resource_group): - response = self.client.storage_tasks_report.list( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_operations.py similarity index 72% rename from sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations.py rename to sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_operations.py index 2f8ddd6e400c..ed4f6e2bb97e 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations.py +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_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.storageactions import StorageActionsMgmtClient +from azure.mgmt.storageactions import StorageActionsClient 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 TestStorageActionsMgmtOperations(AzureMgmtRecordedTestCase): +class TestStorageActionsOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient) + self.client = self.create_mgmt_client(StorageActionsClient) @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_operations_list(self, resource_group): - response = self.client.operations.list( - api_version="2023-01-01", - ) + response = self.client.operations.list() result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations_async.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_operations_async.py similarity index 72% rename from sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations_async.py rename to sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_operations_async.py index 585dd579067b..b0a1906d771d 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_operations_async.py +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_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.storageactions.aio import StorageActionsMgmtClient +from azure.mgmt.storageactions.aio import StorageActionsClient 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 TestStorageActionsMgmtOperationsAsync(AzureMgmtRecordedTestCase): +class TestStorageActionsOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient, is_async=True) + self.client = self.create_mgmt_client(StorageActionsClient, 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-01", - ) + response = self.client.operations.list() result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations.py new file mode 100644 index 000000000000..01edb1e67c95 --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations.py @@ -0,0 +1,44 @@ +# 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. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.storageactions import StorageActionsClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestStorageActionsStorageTasksOperationGroupOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(StorageActionsClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_storage_tasks_operation_group_preview_actions(self, resource_group): + response = self.client.storage_tasks_operation_group.preview_actions( + location="str", + parameters={ + "properties": { + "action": {"elseBlockExists": bool, "if": {"condition": "str"}}, + "blobs": [ + { + "matchedBlock": "str", + "metadata": [{"key": "str", "value": "str"}], + "name": "str", + "properties": [{"key": "str", "value": "str"}], + "tags": [{"key": "str", "value": "str"}], + } + ], + "container": {"metadata": [{"key": "str", "value": "str"}], "name": "str"}, + } + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations_async.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations_async.py new file mode 100644 index 000000000000..7b8aa22924f0 --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operation_group_operations_async.py @@ -0,0 +1,45 @@ +# 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. +# -------------------------------------------------------------------------- +import pytest +from azure.mgmt.storageactions.aio import StorageActionsClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestStorageActionsStorageTasksOperationGroupOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(StorageActionsClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_storage_tasks_operation_group_preview_actions(self, resource_group): + response = await self.client.storage_tasks_operation_group.preview_actions( + location="str", + parameters={ + "properties": { + "action": {"elseBlockExists": bool, "if": {"condition": "str"}}, + "blobs": [ + { + "matchedBlock": "str", + "metadata": [{"key": "str", "value": "str"}], + "name": "str", + "properties": [{"key": "str", "value": "str"}], + "tags": [{"key": "str", "value": "str"}], + } + ], + "container": {"metadata": [{"key": "str", "value": "str"}], "name": "str"}, + } + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operations.py similarity index 68% rename from sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations.py rename to sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operations.py index 3fd6617b5a47..bfa1c89998fd 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations.py +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_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.storageactions import StorageActionsMgmtClient +from azure.mgmt.storageactions import StorageActionsClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy @@ -14,9 +14,20 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTasksOperations(AzureMgmtRecordedTestCase): +class TestStorageActionsStorageTasksOperations(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient) + self.client = self.create_mgmt_client(StorageActionsClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_storage_tasks_get(self, resource_group): + response = self.client.storage_tasks.get( + resource_group_name=resource_group.name, + storage_task_name="str", + ) + + # please add some check logic here by yourself + # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy @@ -37,22 +48,12 @@ def test_storage_tasks_begin_create(self, resource_group): "if": { "condition": "str", "operations": [ - { - "name": "str", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"str": "str"}, - } + {"name": "str", "onFailure": "str", "onSuccess": "str", "parameters": {"str": "str"}} ], }, "else": { "operations": [ - { - "name": "str", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"str": "str"}, - } + {"name": "str", "onFailure": "str", "onSuccess": "str", "parameters": {"str": "str"}} ] }, }, @@ -75,36 +76,11 @@ def test_storage_tasks_begin_create(self, resource_group): "tags": {"str": "str"}, "type": "str", }, - api_version="2023-01-01", ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself # ... - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_storage_tasks_begin_delete(self, resource_group): - response = self.client.storage_tasks.begin_delete( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_storage_tasks_get(self, resource_group): - response = self.client.storage_tasks.get( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - - # please add some check logic here by yourself - # ... - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_storage_tasks_begin_update(self, resource_group): @@ -123,34 +99,23 @@ def test_storage_tasks_begin_update(self, resource_group): "if": { "condition": "str", "operations": [ - { - "name": "str", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"str": "str"}, - } + {"name": "str", "onFailure": "str", "onSuccess": "str", "parameters": {"str": "str"}} ], }, "else": { "operations": [ - { - "name": "str", - "onFailure": "break", - "onSuccess": "continue", - "parameters": {"str": "str"}, - } + {"name": "str", "onFailure": "str", "onSuccess": "str", "parameters": {"str": "str"}} ] }, }, + "creationTimeInUtc": "2020-02-20 00:00:00", "description": "str", "enabled": bool, - "creationTimeInUtc": "2020-02-20 00:00:00", "provisioningState": "str", "taskVersion": 0, }, "tags": {"str": "str"}, }, - api_version="2023-01-01", ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself @@ -158,11 +123,12 @@ def test_storage_tasks_begin_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_storage_tasks_list_by_subscription(self, resource_group): - response = self.client.storage_tasks.list_by_subscription( - api_version="2023-01-01", - ) - result = [r for r in response] + def test_storage_tasks_begin_delete(self, resource_group): + response = self.client.storage_tasks.begin_delete( + resource_group_name=resource_group.name, + storage_task_name="str", + ).result() # call '.result()' to poll until service return final result + # please add some check logic here by yourself # ... @@ -171,7 +137,6 @@ def test_storage_tasks_list_by_subscription(self, resource_group): def test_storage_tasks_list_by_resource_group(self, resource_group): response = self.client.storage_tasks.list_by_resource_group( resource_group_name=resource_group.name, - api_version="2023-01-01", ) result = [r for r in response] # please add some check logic here by yourself @@ -179,26 +144,30 @@ def test_storage_tasks_list_by_resource_group(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy - def test_storage_tasks_preview_actions(self, resource_group): - response = self.client.storage_tasks.preview_actions( - location="str", - parameters={ - "properties": { - "action": {"elseBlockExists": bool, "if": {"condition": "str"}}, - "blobs": [ - { - "matchedBlock": "str", - "metadata": [{"key": "str", "value": "str"}], - "name": "str", - "properties": [{"key": "str", "value": "str"}], - "tags": [{"key": "str", "value": "str"}], - } - ], - "container": {"metadata": [{"key": "str", "value": "str"}], "name": "str"}, - } - }, - api_version="2023-01-01", + def test_storage_tasks_list_by_subscription(self, resource_group): + response = self.client.storage_tasks.list_by_subscription() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_storage_tasks_list(self, resource_group): + response = self.client.storage_tasks.list( + resource_group_name=resource_group.name, + storage_task_name="str", ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_storage_tasks_storage_task_assignment_list(self, resource_group): + response = self.client.storage_tasks.storage_task_assignment_list( + resource_group_name=resource_group.name, + storage_task_name="str", + ) + result = [r for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations_async.py b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operations_async.py similarity index 78% rename from sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations_async.py rename to sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_operations_async.py index 6860275744c7..a5e2cc110eba 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_mgmt_storage_tasks_operations_async.py +++ b/sdk/storageactions/azure-mgmt-storageactions/generated_tests/test_storage_actions_storage_tasks_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.storageactions.aio import StorageActionsMgmtClient +from azure.mgmt.storageactions.aio import StorageActionsClient from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer from devtools_testutils.aio import recorded_by_proxy_async @@ -15,9 +15,20 @@ @pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStorageActionsMgmtStorageTasksOperationsAsync(AzureMgmtRecordedTestCase): +class TestStorageActionsStorageTasksOperationsAsync(AzureMgmtRecordedTestCase): def setup_method(self, method): - self.client = self.create_mgmt_client(StorageActionsMgmtClient, is_async=True) + self.client = self.create_mgmt_client(StorageActionsClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_storage_tasks_get(self, resource_group): + response = await self.client.storage_tasks.get( + resource_group_name=resource_group.name, + storage_task_name="str", + ) + + # please add some check logic here by yourself + # ... @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async @@ -41,8 +52,8 @@ async def test_storage_tasks_begin_create(self, resource_group): "operations": [ { "name": "str", - "onFailure": "break", - "onSuccess": "continue", + "onFailure": "str", + "onSuccess": "str", "parameters": {"str": "str"}, } ], @@ -51,8 +62,8 @@ async def test_storage_tasks_begin_create(self, resource_group): "operations": [ { "name": "str", - "onFailure": "break", - "onSuccess": "continue", + "onFailure": "str", + "onSuccess": "str", "parameters": {"str": "str"}, } ] @@ -77,39 +88,12 @@ async def test_storage_tasks_begin_create(self, resource_group): "tags": {"str": "str"}, "type": "str", }, - api_version="2023-01-01", ) ).result() # call '.result()' to poll until service return final result # please add some check logic here by yourself # ... - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_storage_tasks_begin_delete(self, resource_group): - response = await ( - await self.client.storage_tasks.begin_delete( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_storage_tasks_get(self, resource_group): - response = await self.client.storage_tasks.get( - resource_group_name=resource_group.name, - storage_task_name="str", - api_version="2023-01-01", - ) - - # please add some check logic here by yourself - # ... - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async async def test_storage_tasks_begin_update(self, resource_group): @@ -131,8 +115,8 @@ async def test_storage_tasks_begin_update(self, resource_group): "operations": [ { "name": "str", - "onFailure": "break", - "onSuccess": "continue", + "onFailure": "str", + "onSuccess": "str", "parameters": {"str": "str"}, } ], @@ -141,22 +125,21 @@ async def test_storage_tasks_begin_update(self, resource_group): "operations": [ { "name": "str", - "onFailure": "break", - "onSuccess": "continue", + "onFailure": "str", + "onSuccess": "str", "parameters": {"str": "str"}, } ] }, }, + "creationTimeInUtc": "2020-02-20 00:00:00", "description": "str", "enabled": bool, - "creationTimeInUtc": "2020-02-20 00:00:00", "provisioningState": "str", "taskVersion": 0, }, "tags": {"str": "str"}, }, - api_version="2023-01-01", ) ).result() # call '.result()' to poll until service return final result @@ -165,11 +148,14 @@ async def test_storage_tasks_begin_update(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_storage_tasks_list_by_subscription(self, resource_group): - response = self.client.storage_tasks.list_by_subscription( - api_version="2023-01-01", - ) - result = [r async for r in response] + async def test_storage_tasks_begin_delete(self, resource_group): + response = await ( + await self.client.storage_tasks.begin_delete( + resource_group_name=resource_group.name, + storage_task_name="str", + ) + ).result() # call '.result()' to poll until service return final result + # please add some check logic here by yourself # ... @@ -178,7 +164,6 @@ async def test_storage_tasks_list_by_subscription(self, resource_group): async def test_storage_tasks_list_by_resource_group(self, resource_group): response = self.client.storage_tasks.list_by_resource_group( resource_group_name=resource_group.name, - api_version="2023-01-01", ) result = [r async for r in response] # please add some check logic here by yourself @@ -186,26 +171,30 @@ async def test_storage_tasks_list_by_resource_group(self, resource_group): @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy_async - async def test_storage_tasks_preview_actions(self, resource_group): - response = await self.client.storage_tasks.preview_actions( - location="str", - parameters={ - "properties": { - "action": {"elseBlockExists": bool, "if": {"condition": "str"}}, - "blobs": [ - { - "matchedBlock": "str", - "metadata": [{"key": "str", "value": "str"}], - "name": "str", - "properties": [{"key": "str", "value": "str"}], - "tags": [{"key": "str", "value": "str"}], - } - ], - "container": {"metadata": [{"key": "str", "value": "str"}], "name": "str"}, - } - }, - api_version="2023-01-01", + async def test_storage_tasks_list_by_subscription(self, resource_group): + response = self.client.storage_tasks.list_by_subscription() + 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_storage_tasks_list(self, resource_group): + response = self.client.storage_tasks.list( + resource_group_name=resource_group.name, + storage_task_name="str", ) + 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_storage_tasks_storage_task_assignment_list(self, resource_group): + response = self.client.storage_tasks.storage_task_assignment_list( + resource_group_name=resource_group.name, + storage_task_name="str", + ) + result = [r async for r in response] # please add some check logic here by yourself # ... diff --git a/sdk/storageactions/azure-mgmt-storageactions/sdk_packaging.toml b/sdk/storageactions/azure-mgmt-storageactions/sdk_packaging.toml index 20c20ffe0e58..072bf717a36e 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/sdk_packaging.toml +++ b/sdk/storageactions/azure-mgmt-storageactions/sdk_packaging.toml @@ -3,7 +3,7 @@ package_name = "azure-mgmt-storageactions" package_nspkg = "azure-mgmt-nspkg" package_pprint_name = "Storageactions Management" package_doc_id = "" -is_stable = false +is_stable = true is_arm = true need_msrestazure = false need_azuremgmtcore = true diff --git a/sdk/storageactions/azure-mgmt-storageactions/setup.py b/sdk/storageactions/azure-mgmt-storageactions/setup.py index 21abb4c9ff5d..90b3b1590dea 100644 --- a/sdk/storageactions/azure-mgmt-storageactions/setup.py +++ b/sdk/storageactions/azure-mgmt-storageactions/setup.py @@ -49,11 +49,10 @@ 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", + "Development Status :: 5 - Production/Stable", "Programming Language :: Python", "Programming Language :: Python :: 3 :: Only", "Programming Language :: Python :: 3", - "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", @@ -77,7 +76,7 @@ "isodate>=0.6.1", "typing-extensions>=4.6.0", "azure-common>=1.1", - "azure-mgmt-core>=1.3.2", + "azure-mgmt-core>=1.5.0", ], - python_requires=">=3.8", + python_requires=">=3.9", ) diff --git a/sdk/storageactions/azure-mgmt-storageactions/tsp-location.yaml b/sdk/storageactions/azure-mgmt-storageactions/tsp-location.yaml new file mode 100644 index 000000000000..b1c6551ec97e --- /dev/null +++ b/sdk/storageactions/azure-mgmt-storageactions/tsp-location.yaml @@ -0,0 +1,4 @@ +directory: specification/storageactions/StorageAction.Management +commit: b3cda8e13ff3d36e63ea936afa4b6d7fa2cd731f +repo: Azure/azure-rest-api-specs +additionalDirectories: