From a35a0ffdc69418c1cb90e33e2a21ecc1c696f2e1 Mon Sep 17 00:00:00 2001 From: PythonSdkPipelines Date: Thu, 16 Feb 2023 08:45:37 +0000 Subject: [PATCH 1/3] code and test --- .../azure-mgmt-datalake-store/CHANGELOG.md | 6 + .../azure-mgmt-datalake-store/MANIFEST.in | 1 + .../azure-mgmt-datalake-store/README.md | 49 +- .../azure-mgmt-datalake-store/_meta.json | 11 + .../azure/mgmt/datalake/store/__init__.py | 15 +- .../mgmt/datalake/store/_configuration.py | 68 +- ...ta_lake_store_account_management_client.py | 123 +- .../azure/mgmt/datalake/store/_patch.py | 20 + .../mgmt/datalake/store/_serialization.py | 1996 +++++++++++++++++ .../azure/mgmt/datalake/store/_vendor.py | 30 + .../azure/mgmt/datalake/store/_version.py | 2 +- .../azure/mgmt/datalake/store/aio/__init__.py | 15 +- .../mgmt/datalake/store/aio/_configuration.py | 62 +- ...ta_lake_store_account_management_client.py | 103 +- .../azure/mgmt/datalake/store/aio/_patch.py | 20 + .../datalake/store/aio/operations/__init__.py | 18 +- .../aio/operations/_accounts_operations.py | 1222 ++++++---- .../operations/_firewall_rules_operations.py | 645 ++++-- .../aio/operations/_locations_operations.py | 224 +- .../store/aio/operations/_operations.py | 112 +- .../datalake/store/aio/operations/_patch.py | 20 + .../_trusted_id_providers_operations.py | 652 ++++-- .../_virtual_network_rules_operations.py | 650 ++++-- .../mgmt/datalake/store/models/__init__.py | 264 +-- ...e_store_account_management_client_enums.py | 95 +- .../mgmt/datalake/store/models/_models.py | 1695 -------------- .../mgmt/datalake/store/models/_models_py3.py | 1728 +++++++------- .../mgmt/datalake/store/models/_patch.py | 20 + .../datalake/store/operations/__init__.py | 18 +- .../store/operations/_accounts_operations.py | 1543 +++++++++---- .../operations/_firewall_rules_operations.py | 823 +++++-- .../store/operations/_locations_operations.py | 287 ++- .../datalake/store/operations/_operations.py | 139 +- .../mgmt/datalake/store/operations/_patch.py | 20 + .../_trusted_id_providers_operations.py | 829 +++++-- .../_virtual_network_rules_operations.py | 827 +++++-- .../accounts_check_name_availability.py | 41 + .../generated_samples/accounts_create.py | 71 + .../generated_samples/accounts_delete.py | 41 + .../accounts_enable_key_vault.py | 41 + .../generated_samples/accounts_get.py | 41 + .../generated_samples/accounts_list.py | 39 + .../accounts_list_by_resource_group.py | 41 + .../generated_samples/accounts_update.py | 52 + .../firewall_rules_create_or_update.py | 43 + .../firewall_rules_delete.py | 42 + .../generated_samples/firewall_rules_get.py | 42 + .../firewall_rules_list_by_account.py | 42 + .../firewall_rules_update.py | 42 + .../locations_get_capability.py | 40 + .../generated_samples/locations_get_usage.py | 41 + .../generated_samples/operations_list.py | 38 + .../trusted_id_providers_create_or_update.py | 43 + .../trusted_id_providers_delete.py | 42 + .../trusted_id_providers_get.py | 42 + .../trusted_id_providers_list_by_account.py | 42 + .../trusted_id_providers_update.py | 42 + .../virtual_network_rules_create_or_update.py | 43 + .../virtual_network_rules_delete.py | 42 + .../virtual_network_rules_get.py | 42 + .../virtual_network_rules_list_by_account.py | 42 + .../virtual_network_rules_update.py | 42 + .../azure-mgmt-datalake-store/setup.py | 8 +- shared_requirements.txt | 1 + 64 files changed, 10045 insertions(+), 5435 deletions(-) create mode 100644 sdk/datalake/azure-mgmt-datalake-store/_meta.json create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_patch.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_serialization.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_vendor.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_patch.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_patch.py delete mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_patch.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_patch.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_check_name_availability.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_create.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_delete.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_enable_key_vault.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_get.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list_by_resource_group.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_create_or_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_delete.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_get.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_list_by_account.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_capability.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_usage.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/operations_list.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_create_or_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_delete.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_get.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_list_by_account.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_create_or_update.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_delete.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_get.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_list_by_account.py create mode 100644 sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_update.py diff --git a/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md b/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md index 1aecf5116a6f..e965dda625ea 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md +++ b/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md @@ -1,5 +1,11 @@ # Release History +## 1.1.0 (2023-02-16) + +### Features Added + + - Added model CheckNameAvailabilityParametersType + ## 1.0.0 (2020-12-21) **Breaking changes** diff --git a/sdk/datalake/azure-mgmt-datalake-store/MANIFEST.in b/sdk/datalake/azure-mgmt-datalake-store/MANIFEST.in index 619c2afce242..9f4c95472061 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/MANIFEST.in +++ b/sdk/datalake/azure-mgmt-datalake-store/MANIFEST.in @@ -1,5 +1,6 @@ include _meta.json recursive-include tests *.py *.json +recursive-include samples *.py *.md include *.md include azure/__init__.py include azure/mgmt/__init__.py diff --git a/sdk/datalake/azure-mgmt-datalake-store/README.md b/sdk/datalake/azure-mgmt-datalake-store/README.md index 16e3c05eeec3..1b12170fd9db 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/README.md +++ b/sdk/datalake/azure-mgmt-datalake-store/README.md @@ -8,17 +8,52 @@ For a more complete view of Azure libraries, see the [azure sdk python release]( _Azure SDK Python packages support for Python 2.7 has ended 01 January 2022. For more information and questions, please refer to https://github.com/Azure/azure-sdk-for-python/issues/20691_ -# Usage +## Getting started +### Prerequisites -To learn how to use this package, see the [quickstart guide](https://aka.ms/azsdk/python/mgmt) - -For docs and references, see [Python SDK References](https://docs.microsoft.com/python/api/overview/azure/data-lake-store) -Code samples for this package can be found at [Data Lake Store Management](https://docs.microsoft.com/samples/browse/?languages=python&term=Getting%20started%20-%20Managing&terms=Getting%20started%20-%20Managing) on docs.microsoft.com. -Additional code samples for different Azure services are available at [Samples Repo](https://github.com/Azure-Samples/azure-samples-python-management/tree/main/samples/datalakestore) +- Python 3.7+ is required to use this package. +- [Azure subscription](https://azure.microsoft.com/free/) +### Install the package -# Provide Feedback +```bash +pip install azure-mgmt-datalake-store +pip install azure-identity +``` + +### Authentication + +By default, [Azure Active Directory](https://aka.ms/awps/aad) token authentication depends on correct configure of following environment variables. + +- `AZURE_CLIENT_ID` for Azure client ID. +- `AZURE_TENANT_ID` for Azure tenant ID. +- `AZURE_CLIENT_SECRET` for Azure client secret. + +In addition, Azure subscription ID can be configured via environment variable `AZURE_SUBSCRIPTION_ID`. + +With above configuration, client can be authenticated by following code: + +```python +from azure.identity import DefaultAzureCredential +from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient +import os + +sub_id = os.getenv("AZURE_SUBSCRIPTION_ID") +client = DataLakeStoreAccountManagementClient(credential=DefaultAzureCredential(), subscription_id=sub_id) +``` + +## Examples + + +Code samples for this package can be found at [Data Lake Store Management](https://docs.microsoft.com/samples/browse/?languages=python&term=Getting%20started%20-%20Managing&terms=Getting%20started%20-%20Managing) on docs.microsoft.com and [Samples Repo](https://aka.ms/azsdk/python/mgmt/samples) + + +## Troubleshooting + +## Next steps + +## Provide Feedback If you encounter any bugs or have suggestions, please file an issue in the [Issues](https://github.com/Azure/azure-sdk-for-python/issues) diff --git a/sdk/datalake/azure-mgmt-datalake-store/_meta.json b/sdk/datalake/azure-mgmt-datalake-store/_meta.json new file mode 100644 index 000000000000..72cfbe386afb --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/_meta.json @@ -0,0 +1,11 @@ +{ + "commit": "b64c34315bed61363cd4032f1bff1a0b246da490", + "repository_url": "https://github.com/Azure/azure-rest-api-specs", + "autorest": "3.9.2", + "use": [ + "@autorest/python@6.4.0", + "@autorest/modelerfour@4.24.3" + ], + "autorest_command": "autorest specification/datalake-store/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.4.0 --use=@autorest/modelerfour@4.24.3 --version=3.9.2 --version-tolerant=False", + "readme": "specification/datalake-store/resource-manager/readme.md" +} \ No newline at end of file diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/__init__.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/__init__.py index 37a96e4b89ff..64c0cb2ef6e3 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/__init__.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/__init__.py @@ -10,10 +10,17 @@ from ._version import VERSION __version__ = VERSION -__all__ = ['DataLakeStoreAccountManagementClient'] try: - from ._patch import patch_sdk # type: ignore - patch_sdk() + from ._patch import __all__ as _patch_all + from ._patch import * # pylint: disable=unused-wildcard-import except ImportError: - pass + _patch_all = [] +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "DataLakeStoreAccountManagementClient", +] +__all__.extend([p for p in _patch_all if p not in __all__]) + +_patch_sdk() diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_configuration.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_configuration.py index a9e709eb0251..5f2f6a94d8d9 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_configuration.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_configuration.py @@ -6,66 +6,68 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +import sys +from typing import Any, TYPE_CHECKING from azure.core.configuration import Configuration from azure.core.pipeline import policies -from azure.mgmt.core.policies import ARMHttpLoggingPolicy +from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy from ._version import VERSION +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports + if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports - from typing import Any - from azure.core.credentials import TokenCredential -class DataLakeStoreAccountManagementClientConfiguration(Configuration): +class DataLakeStoreAccountManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes """Configuration for DataLakeStoreAccountManagementClient. 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. + :param credential: Credential needed for the client to connect to Azure. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure subscription. The subscription ID forms part of the URI for every service call. + :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure + subscription. The subscription ID forms part of the URI for every service call. Required. :type subscription_id: str + :keyword api_version: Api Version. Default value is "2016-11-01". Note that overriding this + default value may result in unsupported behavior. + :paramtype api_version: str """ - def __init__( - self, - credential, # type: "TokenCredential" - subscription_id, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: + super(DataLakeStoreAccountManagementClientConfiguration, self).__init__(**kwargs) + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", "2016-11-01") + if credential is None: raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") - super(DataLakeStoreAccountManagementClientConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id - self.api_version = "2016-11-01" - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'mgmt-datalake-store/{}'.format(VERSION)) + self.api_version = api_version + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "mgmt-datalake-store/{}".format(VERSION)) self._configure(**kwargs) - def _configure( - self, - **kwargs # type: Any - ): - # type: (...) -> None - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_data_lake_store_account_management_client.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_data_lake_store_account_management_client.py index 934e59bc9590..0e22526ad7eb 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_data_lake_store_account_management_client.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_data_lake_store_account_management_client.py @@ -6,28 +6,30 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING +from copy import deepcopy +from typing import Any, TYPE_CHECKING +from azure.core.rest import HttpRequest, HttpResponse from azure.mgmt.core import ARMPipelineClient -from msrest import Deserializer, Serializer + +from . import models as _models +from ._configuration import DataLakeStoreAccountManagementClientConfiguration +from ._serialization import Deserializer, Serializer +from .operations import ( + AccountsOperations, + FirewallRulesOperations, + LocationsOperations, + Operations, + TrustedIdProvidersOperations, + VirtualNetworkRulesOperations, +) if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Optional - from azure.core.credentials import TokenCredential -from ._configuration import DataLakeStoreAccountManagementClientConfiguration -from .operations import AccountsOperations -from .operations import FirewallRulesOperations -from .operations import VirtualNetworkRulesOperations -from .operations import TrustedIdProvidersOperations -from .operations import Operations -from .operations import LocationsOperations -from . import models - -class DataLakeStoreAccountManagementClient(object): +class DataLakeStoreAccountManagementClient: # pylint: disable=client-accepts-api-version-keyword """Creates an Azure Data Lake Store account management client. :ivar accounts: AccountsOperations operations @@ -35,61 +37,84 @@ class DataLakeStoreAccountManagementClient(object): :ivar firewall_rules: FirewallRulesOperations operations :vartype firewall_rules: azure.mgmt.datalake.store.operations.FirewallRulesOperations :ivar virtual_network_rules: VirtualNetworkRulesOperations operations - :vartype virtual_network_rules: azure.mgmt.datalake.store.operations.VirtualNetworkRulesOperations + :vartype virtual_network_rules: + azure.mgmt.datalake.store.operations.VirtualNetworkRulesOperations :ivar trusted_id_providers: TrustedIdProvidersOperations operations - :vartype trusted_id_providers: azure.mgmt.datalake.store.operations.TrustedIdProvidersOperations + :vartype trusted_id_providers: + azure.mgmt.datalake.store.operations.TrustedIdProvidersOperations :ivar operations: Operations operations :vartype operations: azure.mgmt.datalake.store.operations.Operations :ivar locations: LocationsOperations operations :vartype locations: azure.mgmt.datalake.store.operations.LocationsOperations - :param credential: Credential needed for the client to connect to Azure. + :param credential: Credential needed for the client to connect to Azure. Required. :type credential: ~azure.core.credentials.TokenCredential - :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure subscription. The subscription ID forms part of the URI for every service call. + :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure + subscription. The subscription ID forms part of the URI for every service call. Required. :type subscription_id: str - :param str base_url: Service URL - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :param base_url: Service URL. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: Api Version. Default value is "2016-11-01". Note that overriding this + default value may result in unsupported behavior. + :paramtype api_version: str + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. """ def __init__( self, - credential, # type: "TokenCredential" - subscription_id, # type: str - base_url=None, # type: Optional[str] - **kwargs # type: Any - ): - # type: (...) -> None - if not base_url: - base_url = 'https://management.azure.com' - self._config = DataLakeStoreAccountManagementClientConfiguration(credential, subscription_id, **kwargs) + credential: "TokenCredential", + subscription_id: str, + base_url: str = "https://management.azure.com", + **kwargs: Any + ) -> None: + self._config = DataLakeStoreAccountManagementClientConfiguration( + credential=credential, subscription_id=subscription_id, **kwargs + ) self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs) - client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) - self._serialize.client_side_validation = False self._deserialize = Deserializer(client_models) - - self.accounts = AccountsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.firewall_rules = FirewallRulesOperations( - self._client, self._config, self._serialize, self._deserialize) + self._serialize.client_side_validation = False + self.accounts = AccountsOperations(self._client, self._config, self._serialize, self._deserialize) + self.firewall_rules = FirewallRulesOperations(self._client, self._config, self._serialize, self._deserialize) self.virtual_network_rules = VirtualNetworkRulesOperations( - self._client, self._config, self._serialize, self._deserialize) + self._client, self._config, self._serialize, self._deserialize + ) self.trusted_id_providers = TrustedIdProvidersOperations( - self._client, self._config, self._serialize, self._deserialize) - self.operations = Operations( - self._client, self._config, self._serialize, self._deserialize) - self.locations = LocationsOperations( - self._client, self._config, self._serialize, self._deserialize) - - def close(self): - # type: () -> None + self._client, self._config, self._serialize, self._deserialize + ) + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize) + + def _send_request(self, request: HttpRequest, **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) + + + For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. + :return: The response of your network call. Does not do error handling on your response. + :rtype: ~azure.core.rest.HttpResponse + """ + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, **kwargs) + + def close(self) -> None: self._client.close() - def __enter__(self): - # type: () -> DataLakeStoreAccountManagementClient + def __enter__(self) -> "DataLakeStoreAccountManagementClient": self._client.__enter__() return self - def __exit__(self, *exc_details): - # type: (Any) -> None + def __exit__(self, *exc_details: Any) -> None: self._client.__exit__(*exc_details) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_patch.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_serialization.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_serialization.py new file mode 100644 index 000000000000..f17c068e833e --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_serialization.py @@ -0,0 +1,1996 @@ +# -------------------------------------------------------------------------- +# +# 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. +# +# -------------------------------------------------------------------------- + +# pylint: skip-file +# pyright: reportUnnecessaryTypeIgnoreComment=false + +from base64 import b64decode, b64encode +import calendar +import datetime +import decimal +import email +from enum import Enum +import json +import logging +import re +import sys +import codecs +from typing import ( + Dict, + Any, + cast, + Optional, + Union, + AnyStr, + IO, + Mapping, + Callable, + TypeVar, + MutableMapping, + Type, + List, + Mapping, +) + +try: + from urllib import quote # type: ignore +except ImportError: + from urllib.parse import quote +import xml.etree.ElementTree as ET + +import isodate # type: ignore + +from azure.core.exceptions import DeserializationError, SerializationError, raise_with_traceback +from azure.core.serialization import NULL as AzureCoreNull + +_BOM = codecs.BOM_UTF8.decode(encoding="utf-8") + +ModelType = TypeVar("ModelType", bound="Model") +JSON = MutableMapping[str, Any] + + +class RawDeserializer: + + # Accept "text" because we're open minded people... + JSON_REGEXP = re.compile(r"^(application|text)/([a-z+.]+\+)?json$") + + # Name used in context + CONTEXT_NAME = "deserialized_data" + + @classmethod + def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: Optional[str] = None) -> Any: + """Decode data according to content-type. + + Accept a stream of data as well, but will be load at once in memory for now. + + If no content-type, will return the string version (not bytes, not stream) + + :param data: Input, could be bytes or stream (will be decoded with UTF8) or text + :type data: str or bytes or IO + :param str content_type: The content type. + """ + if hasattr(data, "read"): + # Assume a stream + data = cast(IO, data).read() + + if isinstance(data, bytes): + data_as_str = data.decode(encoding="utf-8-sig") + else: + # Explain to mypy the correct type. + data_as_str = cast(str, data) + + # Remove Byte Order Mark if present in string + data_as_str = data_as_str.lstrip(_BOM) + + if content_type is None: + return data + + if cls.JSON_REGEXP.match(content_type): + try: + return json.loads(data_as_str) + except ValueError as err: + raise DeserializationError("JSON is invalid: {}".format(err), err) + elif "xml" in (content_type or []): + try: + + try: + if isinstance(data, unicode): # type: ignore + # If I'm Python 2.7 and unicode XML will scream if I try a "fromstring" on unicode string + data_as_str = data_as_str.encode(encoding="utf-8") # type: ignore + except NameError: + pass + + return ET.fromstring(data_as_str) # nosec + except ET.ParseError: + # It might be because the server has an issue, and returned JSON with + # content-type XML.... + # So let's try a JSON load, and if it's still broken + # let's flow the initial exception + def _json_attemp(data): + try: + return True, json.loads(data) + except ValueError: + return False, None # Don't care about this one + + success, json_result = _json_attemp(data) + if success: + return json_result + # If i'm here, it's not JSON, it's not XML, let's scream + # and raise the last context in this block (the XML exception) + # The function hack is because Py2.7 messes up with exception + # context otherwise. + _LOGGER.critical("Wasn't XML not JSON, failing") + raise_with_traceback(DeserializationError, "XML is invalid") + raise DeserializationError("Cannot deserialize content-type: {}".format(content_type)) + + @classmethod + def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], headers: Mapping) -> Any: + """Deserialize from HTTP response. + + Use bytes and headers to NOT use any requests/aiohttp or whatever + specific implementation. + Headers will tested for "content-type" + """ + # Try to use content-type from headers if available + content_type = None + if "content-type" in headers: + content_type = headers["content-type"].split(";")[0].strip().lower() + # Ouch, this server did not declare what it sent... + # Let's guess it's JSON... + # Also, since Autorest was considering that an empty body was a valid JSON, + # need that test as well.... + else: + content_type = "application/json" + + if body_bytes: + return cls.deserialize_from_text(body_bytes, content_type) + return None + + +try: + basestring # type: ignore + unicode_str = unicode # type: ignore +except NameError: + basestring = str + unicode_str = str + +_LOGGER = logging.getLogger(__name__) + +try: + _long_type = long # type: ignore +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.""" + return datetime.timedelta(0) + + def tzname(self, dt): + """Timestamp representation.""" + return "Z" + + def dst(self, dt): + """No daylight saving for UTC.""" + 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): + 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 + +_FLATTEN = re.compile(r"(? None: + self.additional_properties: Dict[str, Any] = {} + for k in kwargs: + if k not in self._attribute_map: + _LOGGER.warning("%s is not a known attribute of class %s and will be ignored", k, self.__class__) + elif k in self._validation and self._validation[k].get("readonly", False): + _LOGGER.warning("Readonly attribute %s will be ignored in class %s", k, self.__class__) + else: + setattr(self, k, kwargs[k]) + + def __eq__(self, other: Any) -> bool: + """Compare objects by comparing all attributes.""" + if isinstance(other, self.__class__): + return self.__dict__ == other.__dict__ + return False + + def __ne__(self, other: Any) -> bool: + """Compare objects by comparing all attributes.""" + return not self.__eq__(other) + + def __str__(self) -> str: + return str(self.__dict__) + + @classmethod + def enable_additional_properties_sending(cls) -> None: + cls._attribute_map["additional_properties"] = {"key": "", "type": "{object}"} + + @classmethod + def is_xml_model(cls) -> bool: + try: + cls._xml_map # type: ignore + except AttributeError: + return False + return True + + @classmethod + def _create_xml_node(cls): + """Create XML node.""" + try: + xml_map = cls._xml_map # type: ignore + except AttributeError: + xml_map = {} + + return _create_xml_node(xml_map.get("name", cls.__name__), xml_map.get("prefix", None), xml_map.get("ns", None)) + + def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: + """Return the JSON that would be sent to azure from this model. + + This is an alias to `as_dict(full_restapi_key_transformer, keep_readonly=False)`. + + If you want XML serialization, you can pass the kwargs is_xml=True. + + :param bool keep_readonly: If you want to serialize the readonly attributes + :returns: A dict JSON compatible object + :rtype: dict + """ + serializer = Serializer(self._infer_class_models()) + return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) + + def as_dict( + self, + keep_readonly: bool = True, + key_transformer: Callable[[str, Dict[str, Any], Any], Any] = attribute_transformer, + **kwargs: Any + ) -> JSON: + """Return a dict that can be serialized using json.dump. + + Advanced usage might optionally use a callback as parameter: + + .. code::python + + def my_key_transformer(key, attr_desc, value): + return key + + Key is the attribute name used in Python. Attr_desc + is a dict of metadata. Currently contains 'type' with the + msrest type and 'key' with the RestAPI encoded key. + Value is the current value in this object. + + The string returned will be used to serialize the key. + If the return type is a list, this is considered hierarchical + result dict. + + See the three examples in this file: + + - attribute_transformer + - full_restapi_key_transformer + - last_restapi_key_transformer + + If you want XML serialization, you can pass the kwargs is_xml=True. + + :param function key_transformer: A key transformer function. + :returns: A dict JSON compatible object + :rtype: dict + """ + serializer = Serializer(self._infer_class_models()) + return serializer._serialize(self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs) + + @classmethod + def _infer_class_models(cls): + try: + str_models = cls.__module__.rsplit(".", 1)[0] + models = sys.modules[str_models] + client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} + if cls.__name__ not in client_models: + raise ValueError("Not Autorest generated code") + except Exception: + # Assume it's not Autorest generated (tests?). Add ourselves as dependencies. + client_models = {cls.__name__: cls} + return client_models + + @classmethod + def deserialize(cls: Type[ModelType], data: Any, content_type: Optional[str] = None) -> ModelType: + """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 + """ + deserializer = Deserializer(cls._infer_class_models()) + return deserializer(cls.__name__, data, content_type=content_type) + + @classmethod + def from_dict( + cls: Type[ModelType], + data: Any, + key_extractors: Optional[Callable[[str, Dict[str, Any], Any], Any]] = None, + content_type: Optional[str] = None, + ) -> ModelType: + """Parse a dict using given key extractor return a model. + + By default consider key + extractors (rest_key_case_insensitive_extractor, attribute_key_case_insensitive_extractor + and last_rest_key_case_insensitive_extractor) + + :param dict data: A dict using RestAPI structure + :param str content_type: JSON by default, set application/xml if XML. + :returns: An instance of this model + :raises: DeserializationError if something went wrong + """ + deserializer = Deserializer(cls._infer_class_models()) + deserializer.key_extractors = ( # type: ignore + [ # type: ignore + attribute_key_case_insensitive_extractor, + rest_key_case_insensitive_extractor, + last_rest_key_case_insensitive_extractor, + ] + if key_extractors is None + else key_extractors + ) + return deserializer(cls.__name__, data, content_type=content_type) + + @classmethod + def _flatten_subtype(cls, key, objects): + if "_subtype_map" not in cls.__dict__: + return {} + result = dict(cls._subtype_map[key]) + for valuetype in cls._subtype_map[key].values(): + result.update(objects[valuetype]._flatten_subtype(key, objects)) + return result + + @classmethod + def _classify(cls, response, objects): + """Check the class _subtype_map for any child classes. + We want to ignore any inherited _subtype_maps. + Remove the polymorphic key from the initial data. + """ + for subtype_key in cls.__dict__.get("_subtype_map", {}).keys(): + subtype_value = None + + if not isinstance(response, ET.Element): + rest_api_response_key = cls._get_rest_key_parts(subtype_key)[-1] + subtype_value = response.pop(rest_api_response_key, None) or response.pop(subtype_key, None) + else: + subtype_value = xml_key_extractor(subtype_key, cls._attribute_map[subtype_key], response) + if subtype_value: + # Try to match base class. Can be class name only + # (bug to fix in Autorest to support x-ms-discriminator-name) + if cls.__name__ == subtype_value: + return cls + flatten_mapping_type = cls._flatten_subtype(subtype_key, objects) + try: + return objects[flatten_mapping_type[subtype_value]] # type: ignore + except KeyError: + _LOGGER.warning( + "Subtype value %s has no mapping, use base class %s.", + subtype_value, + cls.__name__, + ) + break + else: + _LOGGER.warning("Discriminator %s is absent or null, use base class %s.", subtype_key, cls.__name__) + break + return cls + + @classmethod + def _get_rest_key_parts(cls, attr_key): + """Get the RestAPI key of this attr, split it and decode part + :param str attr_key: Attribute key must be in attribute_map. + :returns: A list of RestAPI part + :rtype: list + """ + rest_split_key = _FLATTEN.split(cls._attribute_map[attr_key]["key"]) + return [_decode_attribute_map_key(key_part) for key_part in rest_split_key] + + +def _decode_attribute_map_key(key): + """This decode a key in an _attribute_map to the actual key we want to look at + inside the received data. + + :param str key: A key string from the generated code + """ + return key.replace("\\.", ".") + + +class Serializer(object): + """Request object model serializer.""" + + basic_types = {str: "str", int: "int", bool: "bool", float: "float"} + + _xml_basic_types_serializers = {"bool": lambda x: str(x).lower()} + days = {0: "Mon", 1: "Tue", 2: "Wed", 3: "Thu", 4: "Fri", 5: "Sat", 6: "Sun"} + months = { + 1: "Jan", + 2: "Feb", + 3: "Mar", + 4: "Apr", + 5: "May", + 6: "Jun", + 7: "Jul", + 8: "Aug", + 9: "Sep", + 10: "Oct", + 11: "Nov", + 12: "Dec", + } + validation = { + "min_length": lambda x, y: len(x) < y, + "max_length": lambda x, y: len(x) > y, + "minimum": lambda x, y: x < y, + "maximum": lambda x, y: x > y, + "minimum_ex": lambda x, y: x <= y, + "maximum_ex": lambda x, y: x >= y, + "min_items": lambda x, y: len(x) < y, + "max_items": lambda x, y: len(x) > y, + "pattern": lambda x, y: not re.match(y, x, re.UNICODE), + "unique": lambda x, y: len(x) != len(set(x)), + "multiple": lambda x, y: x % y != 0, + } + + def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]] = None): + self.serialize_type = { + "iso-8601": Serializer.serialize_iso, + "rfc-1123": Serializer.serialize_rfc, + "unix-time": Serializer.serialize_unix, + "duration": Serializer.serialize_duration, + "date": Serializer.serialize_date, + "time": Serializer.serialize_time, + "decimal": Serializer.serialize_decimal, + "long": Serializer.serialize_long, + "bytearray": Serializer.serialize_bytearray, + "base64": Serializer.serialize_base64, + "object": self.serialize_object, + "[]": self.serialize_iter, + "{}": self.serialize_dict, + } + self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} + self.key_transformer = full_restapi_key_transformer + self.client_side_validation = True + + def _serialize(self, target_obj, data_type=None, **kwargs): + """Serialize data into a string according to type. + + :param 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. + """ + key_transformer = kwargs.get("key_transformer", self.key_transformer) + keep_readonly = kwargs.get("keep_readonly", False) + if target_obj is None: + return None + + attr_name = None + class_name = target_obj.__class__.__name__ + + if data_type: + return self.serialize_data(target_obj, data_type, **kwargs) + + if not hasattr(target_obj, "_attribute_map"): + data_type = type(target_obj).__name__ + if data_type in self.basic_types.values(): + return self.serialize_data(target_obj, data_type, **kwargs) + + # Force "is_xml" kwargs if we detect a XML model + try: + is_xml_model_serialization = kwargs["is_xml"] + except KeyError: + is_xml_model_serialization = kwargs.setdefault("is_xml", target_obj.is_xml_model()) + + serialized = {} + if is_xml_model_serialization: + serialized = target_obj._create_xml_node() + try: + attributes = target_obj._attribute_map + for attr, attr_desc in attributes.items(): + attr_name = attr + if not keep_readonly and target_obj._validation.get(attr_name, {}).get("readonly", False): + continue + + if attr_name == "additional_properties" and attr_desc["key"] == "": + if target_obj.additional_properties is not None: + serialized.update(target_obj.additional_properties) + continue + try: + + orig_attr = getattr(target_obj, attr) + if is_xml_model_serialization: + pass # Don't provide "transformer" for XML for now. Keep "orig_attr" + else: # JSON + keys, orig_attr = key_transformer(attr, attr_desc.copy(), orig_attr) + keys = keys if isinstance(keys, list) else [keys] + + kwargs["serialization_ctxt"] = attr_desc + new_attr = self.serialize_data(orig_attr, attr_desc["type"], **kwargs) + + if is_xml_model_serialization: + xml_desc = attr_desc.get("xml", {}) + xml_name = xml_desc.get("name", attr_desc["key"]) + xml_prefix = xml_desc.get("prefix", None) + xml_ns = xml_desc.get("ns", None) + if xml_desc.get("attr", False): + if xml_ns: + ET.register_namespace(xml_prefix, xml_ns) + xml_name = "{}{}".format(xml_ns, xml_name) + serialized.set(xml_name, new_attr) # type: ignore + continue + if xml_desc.get("text", False): + serialized.text = new_attr # type: ignore + continue + if isinstance(new_attr, list): + serialized.extend(new_attr) # type: ignore + elif isinstance(new_attr, ET.Element): + # If the down XML has no XML/Name, we MUST replace the tag with the local tag. But keeping the namespaces. + if "name" not in getattr(orig_attr, "_xml_map", {}): + splitted_tag = new_attr.tag.split("}") + if len(splitted_tag) == 2: # Namespace + new_attr.tag = "}".join([splitted_tag[0], xml_name]) + else: + new_attr.tag = xml_name + serialized.append(new_attr) # type: ignore + else: # That's a basic type + # Integrate namespace if necessary + local_node = _create_xml_node(xml_name, xml_prefix, xml_ns) + local_node.text = unicode_str(new_attr) + serialized.append(local_node) # type: ignore + else: # JSON + for k in reversed(keys): # type: ignore + new_attr = {k: new_attr} + + _new_attr = new_attr + _serialized = serialized + for k in keys: # type: ignore + if k not in _serialized: + _serialized.update(_new_attr) # type: ignore + _new_attr = _new_attr[k] # type: ignore + _serialized = _serialized[k] + except ValueError: + continue + + except (AttributeError, KeyError, TypeError) as err: + msg = "Attribute {} in object {} cannot be serialized.\n{}".format(attr_name, class_name, str(target_obj)) + raise_with_traceback(SerializationError, msg, err) + else: + return serialized + + def body(self, data, data_type, **kwargs): + """Serialize data intended for a request body. + + :param 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 + """ + + # Just in case this is a dict + internal_data_type_str = data_type.strip("[]{}") + internal_data_type = self.dependencies.get(internal_data_type_str, None) + try: + is_xml_model_serialization = kwargs["is_xml"] + except KeyError: + if internal_data_type and issubclass(internal_data_type, Model): + is_xml_model_serialization = kwargs.setdefault("is_xml", internal_data_type.is_xml_model()) + else: + is_xml_model_serialization = False + if internal_data_type and not isinstance(internal_data_type, Enum): + try: + deserializer = Deserializer(self.dependencies) + # Since it's on serialization, it's almost sure that format is not JSON REST + # We're not able to deal with additional properties for now. + deserializer.additional_properties_detection = False + if is_xml_model_serialization: + deserializer.key_extractors = [ # type: ignore + attribute_key_case_insensitive_extractor, + ] + else: + deserializer.key_extractors = [ + rest_key_case_insensitive_extractor, + attribute_key_case_insensitive_extractor, + last_rest_key_case_insensitive_extractor, + ] + data = deserializer._deserialize(data_type, data) + except DeserializationError as err: + raise_with_traceback(SerializationError, "Unable to build a model: " + str(err), err) + + return self._serialize(data, data_type, **kwargs) + + def url(self, name, data, data_type, **kwargs): + """Serialize data intended for a URL path. + + :param 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 + """ + try: + output = self.serialize_data(data, data_type, **kwargs) + if data_type == "bool": + output = json.dumps(output) + + if kwargs.get("skip_quote") is True: + output = str(output) + else: + output = quote(str(output), safe="") + except SerializationError: + raise TypeError("{} must be type {}.".format(name, data_type)) + else: + return output + + def query(self, name, data, data_type, **kwargs): + """Serialize data intended for a URL query. + + :param 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 + """ + try: + # Treat the list aside, since we don't want to encode the div separator + if data_type.startswith("["): + internal_data_type = data_type[1:-1] + data = [self.serialize_data(d, internal_data_type, **kwargs) if d is not None else "" for d in data] + if not kwargs.get("skip_quote", False): + data = [quote(str(d), safe="") for d in data] + return str(self.serialize_iter(data, internal_data_type, **kwargs)) + + # Not a list, regular serialization + output = self.serialize_data(data, data_type, **kwargs) + if data_type == "bool": + output = json.dumps(output) + if kwargs.get("skip_quote") is True: + output = str(output) + else: + output = quote(str(output), safe="") + except SerializationError: + raise TypeError("{} must be type {}.".format(name, data_type)) + else: + return str(output) + + def header(self, name, data, data_type, **kwargs): + """Serialize data intended for a request header. + + :param 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 + """ + try: + if data_type in ["[str]"]: + data = ["" if d is None else d for d in data] + + output = self.serialize_data(data, data_type, **kwargs) + if data_type == "bool": + output = json.dumps(output) + except SerializationError: + raise TypeError("{} must be type {}.".format(name, data_type)) + else: + return str(output) + + def serialize_data(self, data, data_type, **kwargs): + """Serialize generic data according to supplied data type. + + :param data: The data to be serialized. + :param str data_type: The type to be serialized from. + :param bool required: Whether it's essential that the data not be + empty or None + :raises: AttributeError if required data is None. + :raises: ValueError if data is None + :raises: SerializationError if serialization fails. + """ + if data is None: + raise ValueError("No value for given attribute") + + try: + if data is AzureCoreNull: + return None + if data_type in self.basic_types.values(): + return self.serialize_basic(data, data_type, **kwargs) + + elif data_type in self.serialize_type: + return self.serialize_type[data_type](data, **kwargs) + + # If dependencies is empty, try with current data class + # It has to be a subclass of Enum anyway + enum_type = self.dependencies.get(data_type, data.__class__) + if issubclass(enum_type, Enum): + return Serializer.serialize_enum(data, enum_obj=enum_type) + + iter_type = data_type[0] + data_type[-1] + if iter_type in self.serialize_type: + return self.serialize_type[iter_type](data, data_type[1:-1], **kwargs) + + except (ValueError, TypeError) as err: + msg = "Unable to serialize value: {!r} as type: {!r}." + raise_with_traceback(SerializationError, msg.format(data, data_type), err) + else: + return self._serialize(data, **kwargs) + + @classmethod + def _get_custom_serializers(cls, data_type, **kwargs): + custom_serializer = kwargs.get("basic_types_serializers", {}).get(data_type) + if custom_serializer: + return custom_serializer + if kwargs.get("is_xml", False): + return cls._xml_basic_types_serializers.get(data_type) + + @classmethod + def serialize_basic(cls, data, data_type, **kwargs): + """Serialize basic builting data type. + Serializes objects to str, int, float or bool. + + Possible kwargs: + - basic_types_serializers dict[str, callable] : If set, use the callable as serializer + - is_xml bool : If set, use xml_basic_types_serializers + + :param data: Object to be serialized. + :param str data_type: Type of object in the iterable. + """ + custom_serializer = cls._get_custom_serializers(data_type, **kwargs) + if custom_serializer: + return custom_serializer(data) + if data_type == "str": + return cls.serialize_unicode(data) + return eval(data_type)(data) # nosec + + @classmethod + def serialize_unicode(cls, data): + """Special handling for serializing unicode strings in Py2. + Encode to UTF-8 if unicode, otherwise handle as a str. + + :param data: Object to be serialized. + :rtype: str + """ + try: # If I received an enum, return its value + return data.value + except AttributeError: + pass + + try: + if isinstance(data, unicode): # type: ignore + # Don't change it, JSON and XML ElementTree are totally able + # to serialize correctly u'' strings + return data + except NameError: + return str(data) + else: + return str(data) + + def serialize_iter(self, data, iter_type, div=None, **kwargs): + """Serialize iterable. + + Supported kwargs: + - serialization_ctxt dict : The current entry of _attribute_map, or same format. + serialization_ctxt['type'] should be same as data_type. + - is_xml bool : If set, serialize as XML + + :param list attr: Object to be serialized. + :param str iter_type: Type of object in the iterable. + :param bool required: Whether the objects in the iterable must + not be None or empty. + :param str div: If set, this str will be used to combine the elements + in the iterable into a combined string. Default is 'None'. + :rtype: list, str + """ + if isinstance(data, str): + raise SerializationError("Refuse str type as a valid iter type.") + + serialization_ctxt = kwargs.get("serialization_ctxt", {}) + is_xml = kwargs.get("is_xml", False) + + serialized = [] + for d in data: + try: + serialized.append(self.serialize_data(d, iter_type, **kwargs)) + except ValueError: + serialized.append(None) + + if div: + serialized = ["" if s is None else str(s) for s in serialized] + serialized = div.join(serialized) + + if "xml" in serialization_ctxt or is_xml: + # XML serialization is more complicated + xml_desc = serialization_ctxt.get("xml", {}) + xml_name = xml_desc.get("name") + if not xml_name: + xml_name = serialization_ctxt["key"] + + # Create a wrap node if necessary (use the fact that Element and list have "append") + is_wrapped = xml_desc.get("wrapped", False) + node_name = xml_desc.get("itemsName", xml_name) + if is_wrapped: + final_result = _create_xml_node(xml_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + else: + final_result = [] + # All list elements to "local_node" + for el in serialized: + if isinstance(el, ET.Element): + el_node = el + else: + el_node = _create_xml_node(node_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + if el is not None: # Otherwise it writes "None" :-p + el_node.text = str(el) + final_result.append(el_node) + return final_result + return serialized + + def serialize_dict(self, attr, dict_type, **kwargs): + """Serialize a dictionary of objects. + + :param dict attr: Object to be serialized. + :param str dict_type: Type of object in the dictionary. + :param bool required: Whether the objects in the dictionary must + not be None or empty. + :rtype: dict + """ + serialization_ctxt = kwargs.get("serialization_ctxt", {}) + serialized = {} + for key, value in attr.items(): + try: + serialized[self.serialize_unicode(key)] = self.serialize_data(value, dict_type, **kwargs) + except ValueError: + serialized[self.serialize_unicode(key)] = None + + if "xml" in serialization_ctxt: + # XML serialization is more complicated + xml_desc = serialization_ctxt["xml"] + xml_name = xml_desc["name"] + + final_result = _create_xml_node(xml_name, xml_desc.get("prefix", None), xml_desc.get("ns", None)) + for key, value in serialized.items(): + ET.SubElement(final_result, key).text = value + return final_result + + return serialized + + def serialize_object(self, attr, **kwargs): + """Serialize a generic object. + This will be handled as a dictionary. If object passed in is not + a basic type (str, int, float, dict, list) it will simply be + cast to str. + + :param dict attr: Object to be serialized. + :rtype: dict or str + """ + if attr is None: + return None + if isinstance(attr, ET.Element): + return attr + obj_type = type(attr) + if obj_type in self.basic_types: + return self.serialize_basic(attr, self.basic_types[obj_type], **kwargs) + if obj_type is _long_type: + return self.serialize_long(attr) + if obj_type is unicode_str: + return self.serialize_unicode(attr) + if obj_type is datetime.datetime: + return self.serialize_iso(attr) + if obj_type is datetime.date: + return self.serialize_date(attr) + if obj_type is datetime.time: + return self.serialize_time(attr) + if obj_type is datetime.timedelta: + return self.serialize_duration(attr) + if obj_type is decimal.Decimal: + return self.serialize_decimal(attr) + + # If it's a model or I know this dependency, serialize as a Model + elif obj_type in self.dependencies.values() or isinstance(attr, Model): + return self._serialize(attr) + + if obj_type == dict: + serialized = {} + for key, value in attr.items(): + try: + serialized[self.serialize_unicode(key)] = self.serialize_object(value, **kwargs) + except ValueError: + serialized[self.serialize_unicode(key)] = None + return serialized + + if obj_type == list: + serialized = [] + for obj in attr: + try: + serialized.append(self.serialize_object(obj, **kwargs)) + except ValueError: + pass + return serialized + return str(attr) + + @staticmethod + def serialize_enum(attr, enum_obj=None): + try: + result = attr.value + except AttributeError: + result = attr + try: + enum_obj(result) # type: ignore + return result + except ValueError: + for enum_value in enum_obj: # type: ignore + if enum_value.value.lower() == str(attr).lower(): + return enum_value.value + error = "{!r} is not valid value for enum {!r}" + raise SerializationError(error.format(attr, enum_obj)) + + @staticmethod + def serialize_bytearray(attr, **kwargs): + """Serialize bytearray into base-64 string. + + :param attr: Object to be serialized. + :rtype: str + """ + return b64encode(attr).decode() + + @staticmethod + def serialize_base64(attr, **kwargs): + """Serialize str into base-64 string. + + :param attr: Object to be serialized. + :rtype: str + """ + encoded = b64encode(attr).decode("ascii") + return encoded.strip("=").replace("+", "-").replace("/", "_") + + @staticmethod + def serialize_decimal(attr, **kwargs): + """Serialize Decimal object to float. + + :param attr: Object to be serialized. + :rtype: float + """ + return float(attr) + + @staticmethod + def serialize_long(attr, **kwargs): + """Serialize long (Py2) or int (Py3). + + :param attr: Object to be serialized. + :rtype: int/long + """ + return _long_type(attr) + + @staticmethod + def serialize_date(attr, **kwargs): + """Serialize Date object into ISO-8601 formatted string. + + :param Date attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_date(attr) + t = "{:04}-{:02}-{:02}".format(attr.year, attr.month, attr.day) + return t + + @staticmethod + def serialize_time(attr, **kwargs): + """Serialize Time object into ISO-8601 formatted string. + + :param datetime.time attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_time(attr) + t = "{:02}:{:02}:{:02}".format(attr.hour, attr.minute, attr.second) + if attr.microsecond: + t += ".{:02}".format(attr.microsecond) + return t + + @staticmethod + def serialize_duration(attr, **kwargs): + """Serialize TimeDelta object into ISO-8601 formatted string. + + :param TimeDelta attr: Object to be serialized. + :rtype: str + """ + if isinstance(attr, str): + attr = isodate.parse_duration(attr) + return isodate.duration_isoformat(attr) + + @staticmethod + def serialize_rfc(attr, **kwargs): + """Serialize Datetime object into RFC-1123 formatted string. + + :param Datetime attr: Object to be serialized. + :rtype: str + :raises: TypeError if format invalid. + """ + try: + if not attr.tzinfo: + _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") + utc = attr.utctimetuple() + except AttributeError: + raise TypeError("RFC1123 object must be valid Datetime object.") + + return "{}, {:02} {} {:04} {:02}:{:02}:{:02} GMT".format( + Serializer.days[utc.tm_wday], + utc.tm_mday, + Serializer.months[utc.tm_mon], + utc.tm_year, + utc.tm_hour, + utc.tm_min, + utc.tm_sec, + ) + + @staticmethod + def serialize_iso(attr, **kwargs): + """Serialize Datetime object into ISO-8601 formatted string. + + :param Datetime attr: Object to be serialized. + :rtype: str + :raises: SerializationError if format invalid. + """ + if isinstance(attr, str): + attr = isodate.parse_datetime(attr) + try: + if not attr.tzinfo: + _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") + utc = attr.utctimetuple() + if utc.tm_year > 9999 or utc.tm_year < 1: + raise OverflowError("Hit max or min date") + + microseconds = str(attr.microsecond).rjust(6, "0").rstrip("0").ljust(3, "0") + if microseconds: + microseconds = "." + microseconds + date = "{:04}-{:02}-{:02}T{:02}:{:02}:{:02}".format( + utc.tm_year, utc.tm_mon, utc.tm_mday, utc.tm_hour, utc.tm_min, utc.tm_sec + ) + return date + microseconds + "Z" + except (ValueError, OverflowError) as err: + msg = "Unable to serialize datetime object." + raise_with_traceback(SerializationError, msg, err) + except AttributeError as err: + msg = "ISO-8601 object must be valid Datetime object." + raise_with_traceback(TypeError, msg, err) + + @staticmethod + def serialize_unix(attr, **kwargs): + """Serialize Datetime object into IntTime format. + This is represented as seconds. + + :param Datetime attr: Object to be serialized. + :rtype: int + :raises: SerializationError if format invalid + """ + if isinstance(attr, int): + return attr + try: + if not attr.tzinfo: + _LOGGER.warning("Datetime with no tzinfo will be considered UTC.") + return int(calendar.timegm(attr.utctimetuple())) + except AttributeError: + raise TypeError("Unix time object must be valid Datetime object.") + + +def rest_key_extractor(attr, attr_desc, data): + key = attr_desc["key"] + working_data = data + + while "." in key: + # Need the cast, as for some reasons "split" is typed as list[str | Any] + dict_keys = cast(List[str], _FLATTEN.split(key)) + if len(dict_keys) == 1: + key = _decode_attribute_map_key(dict_keys[0]) + break + working_key = _decode_attribute_map_key(dict_keys[0]) + working_data = working_data.get(working_key, data) + if working_data is None: + # If at any point while following flatten JSON path see None, it means + # that all properties under are None as well + # https://github.com/Azure/msrest-for-python/issues/197 + return None + key = ".".join(dict_keys[1:]) + + return working_data.get(key) + + +def rest_key_case_insensitive_extractor(attr, attr_desc, data): + key = attr_desc["key"] + working_data = data + + while "." in key: + dict_keys = _FLATTEN.split(key) + if len(dict_keys) == 1: + key = _decode_attribute_map_key(dict_keys[0]) + break + working_key = _decode_attribute_map_key(dict_keys[0]) + working_data = attribute_key_case_insensitive_extractor(working_key, None, working_data) + if working_data is None: + # If at any point while following flatten JSON path see None, it means + # that all properties under are None as well + # https://github.com/Azure/msrest-for-python/issues/197 + return None + key = ".".join(dict_keys[1:]) + + if working_data: + return attribute_key_case_insensitive_extractor(key, None, working_data) + + +def last_rest_key_extractor(attr, attr_desc, data): + """Extract the attribute in "data" based on the last part of the JSON path key.""" + key = attr_desc["key"] + dict_keys = _FLATTEN.split(key) + return attribute_key_extractor(dict_keys[-1], None, data) + + +def last_rest_key_case_insensitive_extractor(attr, attr_desc, data): + """Extract the attribute in "data" based on the last part of the JSON path key. + + This is the case insensitive version of "last_rest_key_extractor" + """ + key = attr_desc["key"] + dict_keys = _FLATTEN.split(key) + return attribute_key_case_insensitive_extractor(dict_keys[-1], None, data) + + +def attribute_key_extractor(attr, _, data): + return data.get(attr) + + +def attribute_key_case_insensitive_extractor(attr, _, data): + found_key = None + lower_attr = attr.lower() + for key in data: + if lower_attr == key.lower(): + found_key = key + break + + return data.get(found_key) + + +def _extract_name_from_internal_type(internal_type): + """Given an internal type XML description, extract correct XML name with namespace. + + :param dict internal_type: An model type + :rtype: tuple + :returns: A tuple XML name + namespace dict + """ + internal_type_xml_map = getattr(internal_type, "_xml_map", {}) + xml_name = internal_type_xml_map.get("name", internal_type.__name__) + xml_ns = internal_type_xml_map.get("ns", None) + if xml_ns: + xml_name = "{}{}".format(xml_ns, xml_name) + return xml_name + + +def xml_key_extractor(attr, attr_desc, data): + if isinstance(data, dict): + return None + + # Test if this model is XML ready first + if not isinstance(data, ET.Element): + return None + + xml_desc = attr_desc.get("xml", {}) + xml_name = xml_desc.get("name", attr_desc["key"]) + + # Look for a children + is_iter_type = attr_desc["type"].startswith("[") + is_wrapped = xml_desc.get("wrapped", False) + internal_type = attr_desc.get("internalType", None) + internal_type_xml_map = getattr(internal_type, "_xml_map", {}) + + # Integrate namespace if necessary + xml_ns = xml_desc.get("ns", internal_type_xml_map.get("ns", None)) + if xml_ns: + xml_name = "{}{}".format(xml_ns, xml_name) + + # If it's an attribute, that's simple + if xml_desc.get("attr", False): + return data.get(xml_name) + + # If it's x-ms-text, that's simple too + if xml_desc.get("text", False): + return data.text + + # Scenario where I take the local name: + # - Wrapped node + # - Internal type is an enum (considered basic types) + # - Internal type has no XML/Name node + if is_wrapped or (internal_type and (issubclass(internal_type, Enum) or "name" not in internal_type_xml_map)): + children = data.findall(xml_name) + # If internal type has a local name and it's not a list, I use that name + elif not is_iter_type and internal_type and "name" in internal_type_xml_map: + xml_name = _extract_name_from_internal_type(internal_type) + children = data.findall(xml_name) + # That's an array + else: + if internal_type: # Complex type, ignore itemsName and use the complex type name + items_name = _extract_name_from_internal_type(internal_type) + else: + items_name = xml_desc.get("itemsName", xml_name) + children = data.findall(items_name) + + if len(children) == 0: + if is_iter_type: + if is_wrapped: + return None # is_wrapped no node, we want None + else: + return [] # not wrapped, assume empty list + return None # Assume it's not there, maybe an optional node. + + # If is_iter_type and not wrapped, return all found children + if is_iter_type: + if not is_wrapped: + return children + else: # 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( + xml_name + ) + ) + return list(children[0]) # Might be empty list and that's ok. + + # Here it's not a itertype, we should have found one element only or empty + if len(children) > 1: + raise DeserializationError("Find several XML '{}' where it was not expected".format(xml_name)) + return children[0] + + +class Deserializer(object): + """Response object model deserializer. + + :param dict classes: Class type dictionary for deserializing complex types. + :ivar list key_extractors: Ordered list of extractors to be used by this deserializer. + """ + + basic_types = {str: "str", int: "int", bool: "bool", float: "float"} + + 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}]?") + + def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]] = None): + self.deserialize_type = { + "iso-8601": Deserializer.deserialize_iso, + "rfc-1123": Deserializer.deserialize_rfc, + "unix-time": Deserializer.deserialize_unix, + "duration": Deserializer.deserialize_duration, + "date": Deserializer.deserialize_date, + "time": Deserializer.deserialize_time, + "decimal": Deserializer.deserialize_decimal, + "long": Deserializer.deserialize_long, + "bytearray": Deserializer.deserialize_bytearray, + "base64": Deserializer.deserialize_base64, + "object": self.deserialize_object, + "[]": self.deserialize_iter, + "{}": self.deserialize_dict, + } + self.deserialize_expected_types = { + "duration": (isodate.Duration, datetime.timedelta), + "iso-8601": (datetime.datetime), + } + self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} + self.key_extractors = [rest_key_extractor, xml_key_extractor] + # Additional properties only works if the "rest_key_extractor" is used to + # extract the keys. Making it to work whatever the key extractor is too much + # complicated, with no real scenario for now. + # So adding a flag to disable additional properties detection. This flag should be + # used if your expect the deserialization to NOT come from a JSON REST syntax. + # Otherwise, result are unexpected + self.additional_properties_detection = True + + def __call__(self, target_obj, response_data, content_type=None): + """Call the deserializer to process a REST response. + + :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. + :return: Deserialized object. + """ + data = self._unpack_content(response_data, content_type) + return self._deserialize(target_obj, data) + + def _deserialize(self, target_obj, data): + """Call the deserializer on a model. + + Data needs to be already deserialized as JSON or XML ElementTree + + :param str target_obj: Target data type to deserialize to. + :param object data: Object to deserialize. + :raises: DeserializationError if deserialization fails. + :return: Deserialized object. + """ + # This is already a model, go recursive just in case + if hasattr(data, "_attribute_map"): + constants = [name for name, config in getattr(data, "_validation", {}).items() if config.get("constant")] + try: + for attr, mapconfig in data._attribute_map.items(): + if attr in constants: + continue + value = getattr(data, attr) + if value is None: + continue + local_type = mapconfig["type"] + internal_data_type = local_type.strip("[]{}") + if internal_data_type not in self.dependencies or isinstance(internal_data_type, Enum): + continue + setattr(data, attr, self._deserialize(local_type, value)) + return data + except AttributeError: + return + + response, class_name = self._classify_target(target_obj, data) + + if isinstance(response, basestring): + return self.deserialize_data(data, response) + elif isinstance(response, type) and issubclass(response, Enum): + return self.deserialize_enum(data, response) + + if data is None: + return data + try: + attributes = response._attribute_map # type: ignore + d_attrs = {} + for attr, attr_desc in attributes.items(): + # Check empty string. If it's not empty, someone has a real "additionalProperties"... + if attr == "additional_properties" and attr_desc["key"] == "": + continue + raw_value = None + # Enhance attr_desc with some dynamic data + attr_desc = attr_desc.copy() # Do a copy, do not change the real one + internal_data_type = attr_desc["type"].strip("[]{}") + if internal_data_type in self.dependencies: + attr_desc["internalType"] = self.dependencies[internal_data_type] + + for key_extractor in self.key_extractors: + found_value = key_extractor(attr, attr_desc, data) + if found_value is not None: + if raw_value is not None and raw_value != found_value: + msg = ( + "Ignoring extracted value '%s' from %s for key '%s'" + " (duplicate extraction, follow extractors order)" + ) + _LOGGER.warning(msg, found_value, key_extractor, attr) + continue + raw_value = found_value + + value = self.deserialize_data(raw_value, attr_desc["type"]) + d_attrs[attr] = value + except (AttributeError, TypeError, KeyError) as err: + msg = "Unable to deserialize to object: " + class_name # type: ignore + raise_with_traceback(DeserializationError, msg, err) + else: + additional_properties = self._build_additional_properties(attributes, data) + return self._instantiate_model(response, d_attrs, additional_properties) + + def _build_additional_properties(self, attribute_map, data): + if not self.additional_properties_detection: + return None + if "additional_properties" in attribute_map and attribute_map.get("additional_properties", {}).get("key") != "": + # Check empty string. If it's not empty, someone has a real "additionalProperties" + return None + if isinstance(data, ET.Element): + data = {el.tag: el.text for el in data} + + known_keys = { + _decode_attribute_map_key(_FLATTEN.split(desc["key"])[0]) + for desc in attribute_map.values() + if desc["key"] != "" + } + present_keys = set(data.keys()) + missing_keys = present_keys - known_keys + return {key: data[key] for key in missing_keys} + + def _classify_target(self, target, data): + """Check to see whether the deserialization target object can + be classified into a subclass. + Once classification has been determined, initialize object. + + :param str target: The target object type to deserialize to. + :param str/dict data: The response data to deserialize. + """ + if target is None: + return None, None + + if isinstance(target, basestring): + try: + target = self.dependencies[target] + except KeyError: + return target, target + + try: + target = target._classify(data, self.dependencies) + except AttributeError: + pass # Target is not a Model, no classify + return target, target.__class__.__name__ # type: ignore + + def failsafe_deserialize(self, target_obj, data, content_type=None): + """Ignores any errors encountered in deserialization, + and falls back to not deserializing the object. Recommended + for use in error deserialization, as we want to return the + HttpResponseError to users, and not have them deal with + a deserialization error. + + :param str target_obj: The target object type to deserialize to. + :param str/dict data: The response data to deserialize. + :param str content_type: Swagger "produces" if available. + """ + try: + return self(target_obj, data, content_type=content_type) + except: + _LOGGER.debug( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + @staticmethod + def _unpack_content(raw_data, content_type=None): + """Extract the correct structure for deserialization. + + If raw_data is a PipelineResponse, try to extract the result of RawDeserializer. + if we can't, raise. Your Pipeline should have a RawDeserializer. + + If not a pipeline response and raw_data is bytes or string, use content-type + to decode it. If no content-type, try JSON. + + If raw_data is something else, bypass all logic and return it directly. + + :param raw_data: Data to be processed. + :param content_type: How to parse if raw_data is a string/bytes. + :raises JSONDecodeError: If JSON is requested and parsing is impossible. + :raises UnicodeDecodeError: If bytes is not UTF8 + """ + # Assume this is enough to detect a Pipeline Response without importing it + context = getattr(raw_data, "context", {}) + if context: + if RawDeserializer.CONTEXT_NAME in context: + return context[RawDeserializer.CONTEXT_NAME] + raise ValueError("This pipeline didn't have the RawDeserializer policy; can't deserialize") + + # Assume this is enough to recognize universal_http.ClientResponse without importing it + if hasattr(raw_data, "body"): + return RawDeserializer.deserialize_from_http_generics(raw_data.text(), raw_data.headers) + + # Assume this enough to recognize requests.Response without importing it. + if hasattr(raw_data, "_content_consumed"): + return RawDeserializer.deserialize_from_http_generics(raw_data.text, raw_data.headers) + + if isinstance(raw_data, (basestring, bytes)) or hasattr(raw_data, "read"): + return RawDeserializer.deserialize_from_text(raw_data, content_type) # type: ignore + return raw_data + + def _instantiate_model(self, response, attrs, additional_properties=None): + """Instantiate a response model passing in deserialized args. + + :param response: The response model class. + :param d_attrs: The deserialized response attributes. + """ + if callable(response): + subtype = getattr(response, "_subtype_map", {}) + try: + readonly = [k for k, v in response._validation.items() if v.get("readonly")] + const = [k for k, v in response._validation.items() if v.get("constant")] + kwargs = {k: v for k, v in attrs.items() if k not in subtype and k not in readonly + const} + response_obj = response(**kwargs) + for attr in readonly: + setattr(response_obj, attr, attrs.get(attr)) + if additional_properties: + response_obj.additional_properties = additional_properties + return response_obj + except TypeError as err: + msg = "Unable to deserialize {} into model {}. ".format(kwargs, response) # type: ignore + raise DeserializationError(msg + str(err)) + else: + try: + for attr, value in attrs.items(): + setattr(response, attr, value) + return response + except Exception as exp: + msg = "Unable to populate response model. " + msg += "Type: {}, Error: {}".format(type(response), exp) + raise DeserializationError(msg) + + def deserialize_data(self, data, data_type): + """Process data for deserialization according to data type. + + :param str data: The response string to be deserialized. + :param str data_type: The type to deserialize to. + :raises: DeserializationError if deserialization fails. + :return: Deserialized object. + """ + if data is None: + return data + + try: + if not data_type: + return data + if data_type in self.basic_types.values(): + return self.deserialize_basic(data, data_type) + if data_type in self.deserialize_type: + if isinstance(data, self.deserialize_expected_types.get(data_type, tuple())): + return data + + is_a_text_parsing_type = lambda x: x not in ["object", "[]", r"{}"] + if isinstance(data, ET.Element) and is_a_text_parsing_type(data_type) and not data.text: + return None + data_val = self.deserialize_type[data_type](data) + return data_val + + iter_type = data_type[0] + data_type[-1] + if iter_type in self.deserialize_type: + return self.deserialize_type[iter_type](data, data_type[1:-1]) + + obj_type = self.dependencies[data_type] + if issubclass(obj_type, Enum): + if isinstance(data, ET.Element): + data = data.text + return self.deserialize_enum(data, obj_type) + + except (ValueError, TypeError, AttributeError) as err: + msg = "Unable to deserialize response data." + msg += " Data: {}, {}".format(data, data_type) + raise_with_traceback(DeserializationError, msg, err) + else: + return self._deserialize(obj_type, data) + + def deserialize_iter(self, attr, iter_type): + """Deserialize an iterable. + + :param list attr: Iterable to be deserialized. + :param str iter_type: The type of object in the iterable. + :rtype: list + """ + if attr is None: + return None + if isinstance(attr, ET.Element): # If I receive an element here, get the children + attr = list(attr) + if not isinstance(attr, (list, set)): + raise DeserializationError("Cannot deserialize as [{}] an object of type {}".format(iter_type, type(attr))) + return [self.deserialize_data(a, iter_type) for a in attr] + + def deserialize_dict(self, attr, dict_type): + """Deserialize a dictionary. + + :param dict/list attr: Dictionary to be deserialized. Also accepts + a list of key, value pairs. + :param str dict_type: The object type of the items in the dictionary. + :rtype: dict + """ + if isinstance(attr, list): + return {x["key"]: self.deserialize_data(x["value"], dict_type) for x in attr} + + if isinstance(attr, ET.Element): + # Transform value into {"Key": "value"} + attr = {el.tag: el.text for el in attr} + return {k: self.deserialize_data(v, dict_type) for k, v in attr.items()} + + def deserialize_object(self, attr, **kwargs): + """Deserialize a generic object. + This will be handled as a dictionary. + + :param dict attr: Dictionary to be deserialized. + :rtype: dict + :raises: TypeError if non-builtin datatype encountered. + """ + if attr is None: + return None + if isinstance(attr, ET.Element): + # Do no recurse on XML, just return the tree as-is + return attr + if isinstance(attr, basestring): + return self.deserialize_basic(attr, "str") + obj_type = type(attr) + if obj_type in self.basic_types: + return self.deserialize_basic(attr, self.basic_types[obj_type]) + if obj_type is _long_type: + return self.deserialize_long(attr) + + if obj_type == dict: + deserialized = {} + for key, value in attr.items(): + try: + deserialized[key] = self.deserialize_object(value, **kwargs) + except ValueError: + deserialized[key] = None + return deserialized + + if obj_type == list: + deserialized = [] + for obj in attr: + try: + deserialized.append(self.deserialize_object(obj, **kwargs)) + except ValueError: + pass + return deserialized + + else: + error = "Cannot deserialize generic object with type: " + raise TypeError(error + str(obj_type)) + + def deserialize_basic(self, attr, data_type): + """Deserialize basic builtin data type from string. + Will attempt to convert to str, int, float and bool. + This function will also accept '1', '0', 'true' and 'false' as + valid bool values. + + :param str attr: response string to be deserialized. + :param str data_type: deserialization data type. + :rtype: str, int, float or bool + :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 + if isinstance(attr, ET.Element): + attr = attr.text + if not attr: + if data_type == "str": + # None or '', node is empty string. + return "" + else: + # None or '', node with a strong type is None. + # Don't try to model "empty bool" or "empty int" + return None + + if data_type == "bool": + if attr in [True, False, 1, 0]: + return bool(attr) + elif isinstance(attr, basestring): + if attr.lower() in ["true", "1"]: + return True + elif attr.lower() in ["false", "0"]: + return False + raise TypeError("Invalid boolean value: {}".format(attr)) + + if data_type == "str": + return self.deserialize_unicode(attr) + return eval(data_type)(attr) # nosec + + @staticmethod + def deserialize_unicode(data): + """Preserve unicode objects in Python 2, otherwise return data + as a string. + + :param str data: response string to be deserialized. + :rtype: str or unicode + """ + # We might be here because we have an enum modeled as string, + # and we try to deserialize a partial dict with enum inside + if isinstance(data, Enum): + return data + + # Consider this is real string + try: + if isinstance(data, unicode): # type: ignore + return data + except NameError: + return str(data) + else: + return str(data) + + @staticmethod + def deserialize_enum(data, enum_obj): + """Deserialize string into enum object. + + If the string is not a valid enum value it will be returned as-is + and a warning will be logged. + + :param str data: Response string to be deserialized. If this value is + None or invalid it will be returned as-is. + :param Enum enum_obj: Enum object to deserialize to. + :rtype: Enum + """ + if isinstance(data, enum_obj) or data is None: + return data + if isinstance(data, Enum): + data = data.value + if isinstance(data, int): + # Workaround. We might consider remove it in the future. + # https://github.com/Azure/azure-rest-api-specs/issues/141 + try: + return list(enum_obj.__members__.values())[data] + except IndexError: + error = "{!r} is not a valid index for enum {!r}" + raise DeserializationError(error.format(data, enum_obj)) + try: + return enum_obj(str(data)) + except ValueError: + for enum_value in enum_obj: + if enum_value.value.lower() == str(data).lower(): + return enum_value + # We don't fail anymore for unknown value, we deserialize as a string + _LOGGER.warning("Deserializer is not able to find %s as valid enum in %s", data, enum_obj) + return Deserializer.deserialize_unicode(data) + + @staticmethod + def deserialize_bytearray(attr): + """Deserialize string into bytearray. + + :param str attr: response string to be deserialized. + :rtype: bytearray + :raises: TypeError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + return bytearray(b64decode(attr)) # type: ignore + + @staticmethod + def deserialize_base64(attr): + """Deserialize base64 encoded string into string. + + :param str attr: response string to be deserialized. + :rtype: bytearray + :raises: TypeError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + padding = "=" * (3 - (len(attr) + 3) % 4) # type: ignore + attr = attr + padding # type: ignore + encoded = attr.replace("-", "+").replace("_", "/") + return b64decode(encoded) + + @staticmethod + def deserialize_decimal(attr): + """Deserialize string into Decimal object. + + :param str attr: response string to be deserialized. + :rtype: Decimal + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + return decimal.Decimal(attr) # type: ignore + except decimal.DecimalException as err: + msg = "Invalid decimal {}".format(attr) + raise_with_traceback(DeserializationError, msg, err) + + @staticmethod + def deserialize_long(attr): + """Deserialize string into long (Py2) or int (Py3). + + :param str attr: response string to be deserialized. + :rtype: long or int + :raises: ValueError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + return _long_type(attr) # type: ignore + + @staticmethod + def deserialize_duration(attr): + """Deserialize ISO-8601 formatted string into TimeDelta object. + + :param str attr: response string to be deserialized. + :rtype: TimeDelta + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + duration = isodate.parse_duration(attr) + except (ValueError, OverflowError, AttributeError) as err: + msg = "Cannot deserialize duration object." + raise_with_traceback(DeserializationError, msg, err) + else: + return duration + + @staticmethod + def deserialize_date(attr): + """Deserialize ISO-8601 formatted string into Date object. + + :param str attr: response string to be deserialized. + :rtype: Date + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + if re.search(r"[^\W\d_]", attr, re.I + re.U): # type: ignore + raise DeserializationError("Date must have only digits and -. Received: %s" % attr) + # This must NOT use defaultmonth/defaultday. Using None ensure this raises an exception. + return isodate.parse_date(attr, defaultmonth=None, defaultday=None) + + @staticmethod + def deserialize_time(attr): + """Deserialize ISO-8601 formatted string into time object. + + :param str attr: response string to be deserialized. + :rtype: datetime.time + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + if re.search(r"[^\W\d_]", attr, re.I + re.U): # type: ignore + raise DeserializationError("Date must have only digits and -. Received: %s" % attr) + return isodate.parse_time(attr) + + @staticmethod + def deserialize_rfc(attr): + """Deserialize RFC-1123 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: Datetime + :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)) + ) + if not date_obj.tzinfo: + date_obj = date_obj.astimezone(tz=TZ_UTC) + except ValueError as err: + msg = "Cannot deserialize to rfc datetime object." + raise_with_traceback(DeserializationError, msg, err) + else: + return date_obj + + @staticmethod + def deserialize_iso(attr): + """Deserialize ISO-8601 formatted string into Datetime object. + + :param str attr: response string to be deserialized. + :rtype: Datetime + :raises: DeserializationError if string format invalid. + """ + if isinstance(attr, ET.Element): + attr = attr.text + try: + attr = attr.upper() # type: ignore + match = Deserializer.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") + except (ValueError, OverflowError, AttributeError) as err: + msg = "Cannot deserialize datetime object." + raise_with_traceback(DeserializationError, msg, err) + else: + return date_obj + + @staticmethod + def deserialize_unix(attr): + """Serialize Datetime object into IntTime format. + This is represented as seconds. + + :param int attr: Object to be serialized. + :rtype: Datetime + :raises: DeserializationError if format invalid + """ + if isinstance(attr, ET.Element): + attr = int(attr.text) # type: ignore + try: + date_obj = datetime.datetime.fromtimestamp(attr, TZ_UTC) + except ValueError as err: + msg = "Cannot deserialize to unix datetime object." + raise_with_traceback(DeserializationError, msg, err) + else: + return date_obj diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_vendor.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_vendor.py new file mode 100644 index 000000000000..bd0df84f5319 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_vendor.py @@ -0,0 +1,30 @@ +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from typing import List, cast + +from azure.core.pipeline.transport import HttpRequest + + +def _convert_request(request, files=None): + data = request.content if not files else None + request = HttpRequest(method=request.method, url=request.url, headers=request.headers, data=data) + if files: + request.set_formdata_body(files) + return request + + +def _format_url_section(template, **kwargs): + components = template.split("/") + while components: + try: + return template.format(**kwargs) + except KeyError as key: + # Need the cast, as for some reasons "split" is typed as list[str | Any] + formatted_components = cast(List[str], template.split("/")) + components = [c for c in formatted_components if "{}".format(key.args[0]) not in c] + template = "/".join(components) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py index c47f66669f1b..59deb8c7263b 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py @@ -6,4 +6,4 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.0.0" +VERSION = "1.1.0" diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/__init__.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/__init__.py index bcc73f33ddbb..4e17276d41d4 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/__init__.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/__init__.py @@ -7,4 +7,17 @@ # -------------------------------------------------------------------------- from ._data_lake_store_account_management_client import DataLakeStoreAccountManagementClient -__all__ = ['DataLakeStoreAccountManagementClient'] + +try: + from ._patch import __all__ as _patch_all + from ._patch import * # pylint: disable=unused-wildcard-import +except ImportError: + _patch_all = [] +from ._patch import patch_sdk as _patch_sdk + +__all__ = [ + "DataLakeStoreAccountManagementClient", +] +__all__.extend([p for p in _patch_all if p not in __all__]) + +_patch_sdk() diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_configuration.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_configuration.py index 881f1fcfe1f2..c07cdff76a19 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_configuration.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_configuration.py @@ -6,62 +6,68 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +import sys from typing import Any, TYPE_CHECKING from azure.core.configuration import Configuration from azure.core.pipeline import policies -from azure.mgmt.core.policies import ARMHttpLoggingPolicy +from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy from .._version import VERSION +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports + if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials_async import AsyncTokenCredential -class DataLakeStoreAccountManagementClientConfiguration(Configuration): +class DataLakeStoreAccountManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes """Configuration for DataLakeStoreAccountManagementClient. 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. + :param credential: Credential needed for the client to connect to Azure. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure subscription. The subscription ID forms part of the URI for every service call. + :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure + subscription. The subscription ID forms part of the URI for every service call. Required. :type subscription_id: str + :keyword api_version: Api Version. Default value is "2016-11-01". Note that overriding this + default value may result in unsupported behavior. + :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - subscription_id: str, - **kwargs: Any - ) -> None: + def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: + super(DataLakeStoreAccountManagementClientConfiguration, self).__init__(**kwargs) + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", "2016-11-01") + if credential is None: raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") - super(DataLakeStoreAccountManagementClientConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id - self.api_version = "2016-11-01" - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'mgmt-datalake-store/{}'.format(VERSION)) + self.api_version = api_version + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "mgmt-datalake-store/{}".format(VERSION)) self._configure(**kwargs) - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_data_lake_store_account_management_client.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_data_lake_store_account_management_client.py index 36e16b7000ac..dd6dd8467e7e 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_data_lake_store_account_management_client.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_data_lake_store_account_management_client.py @@ -6,26 +6,30 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, TYPE_CHECKING +from copy import deepcopy +from typing import Any, Awaitable, TYPE_CHECKING +from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.mgmt.core import AsyncARMPipelineClient -from msrest import Deserializer, Serializer + +from .. import models as _models +from .._serialization import Deserializer, Serializer +from ._configuration import DataLakeStoreAccountManagementClientConfiguration +from .operations import ( + AccountsOperations, + FirewallRulesOperations, + LocationsOperations, + Operations, + TrustedIdProvidersOperations, + VirtualNetworkRulesOperations, +) if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials_async import AsyncTokenCredential -from ._configuration import DataLakeStoreAccountManagementClientConfiguration -from .operations import AccountsOperations -from .operations import FirewallRulesOperations -from .operations import VirtualNetworkRulesOperations -from .operations import TrustedIdProvidersOperations -from .operations import Operations -from .operations import LocationsOperations -from .. import models - -class DataLakeStoreAccountManagementClient(object): +class DataLakeStoreAccountManagementClient: # pylint: disable=client-accepts-api-version-keyword """Creates an Azure Data Lake Store account management client. :ivar accounts: AccountsOperations operations @@ -33,50 +37,77 @@ class DataLakeStoreAccountManagementClient(object): :ivar firewall_rules: FirewallRulesOperations operations :vartype firewall_rules: azure.mgmt.datalake.store.aio.operations.FirewallRulesOperations :ivar virtual_network_rules: VirtualNetworkRulesOperations operations - :vartype virtual_network_rules: azure.mgmt.datalake.store.aio.operations.VirtualNetworkRulesOperations + :vartype virtual_network_rules: + azure.mgmt.datalake.store.aio.operations.VirtualNetworkRulesOperations :ivar trusted_id_providers: TrustedIdProvidersOperations operations - :vartype trusted_id_providers: azure.mgmt.datalake.store.aio.operations.TrustedIdProvidersOperations + :vartype trusted_id_providers: + azure.mgmt.datalake.store.aio.operations.TrustedIdProvidersOperations :ivar operations: Operations operations :vartype operations: azure.mgmt.datalake.store.aio.operations.Operations :ivar locations: LocationsOperations operations :vartype locations: azure.mgmt.datalake.store.aio.operations.LocationsOperations - :param credential: Credential needed for the client to connect to Azure. + :param credential: Credential needed for the client to connect to Azure. Required. :type credential: ~azure.core.credentials_async.AsyncTokenCredential - :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure subscription. The subscription ID forms part of the URI for every service call. + :param subscription_id: Gets subscription credentials which uniquely identify Microsoft Azure + subscription. The subscription ID forms part of the URI for every service call. Required. :type subscription_id: str - :param str base_url: Service URL - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :param base_url: Service URL. Default value is "https://management.azure.com". + :type base_url: str + :keyword api_version: Api Version. Default value is "2016-11-01". Note that overriding this + default value may result in unsupported behavior. + :paramtype api_version: str + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. """ def __init__( self, credential: "AsyncTokenCredential", subscription_id: str, - base_url: Optional[str] = None, + base_url: str = "https://management.azure.com", **kwargs: Any ) -> None: - if not base_url: - base_url = 'https://management.azure.com' - self._config = DataLakeStoreAccountManagementClientConfiguration(credential, subscription_id, **kwargs) + self._config = DataLakeStoreAccountManagementClientConfiguration( + credential=credential, subscription_id=subscription_id, **kwargs + ) self._client = AsyncARMPipelineClient(base_url=base_url, config=self._config, **kwargs) - client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)} + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) - self._serialize.client_side_validation = False self._deserialize = Deserializer(client_models) - - self.accounts = AccountsOperations( - self._client, self._config, self._serialize, self._deserialize) - self.firewall_rules = FirewallRulesOperations( - self._client, self._config, self._serialize, self._deserialize) + self._serialize.client_side_validation = False + self.accounts = AccountsOperations(self._client, self._config, self._serialize, self._deserialize) + self.firewall_rules = FirewallRulesOperations(self._client, self._config, self._serialize, self._deserialize) self.virtual_network_rules = VirtualNetworkRulesOperations( - self._client, self._config, self._serialize, self._deserialize) + self._client, self._config, self._serialize, self._deserialize + ) self.trusted_id_providers = TrustedIdProvidersOperations( - self._client, self._config, self._serialize, self._deserialize) - self.operations = Operations( - self._client, self._config, self._serialize, self._deserialize) - self.locations = LocationsOperations( - self._client, self._config, self._serialize, self._deserialize) + self._client, self._config, self._serialize, self._deserialize + ) + self.operations = Operations(self._client, self._config, self._serialize, self._deserialize) + self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize) + + def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + """Runs the network request through the client's chained policies. + + >>> from azure.core.rest import HttpRequest + >>> request = HttpRequest("GET", "https://www.example.org/") + + >>> response = await client._send_request(request) + + + For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. + :return: The response of your network call. Does not do error handling on your response. + :rtype: ~azure.core.rest.AsyncHttpResponse + """ + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, **kwargs) async def close(self) -> None: await self._client.close() @@ -85,5 +116,5 @@ async def __aenter__(self) -> "DataLakeStoreAccountManagementClient": await self._client.__aenter__() return self - async def __aexit__(self, *exc_details) -> None: + async def __aexit__(self, *exc_details: Any) -> None: await self._client.__aexit__(*exc_details) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_patch.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/__init__.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/__init__.py index ad9f979648cf..1189bb013e90 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/__init__.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/__init__.py @@ -13,11 +13,17 @@ from ._operations import Operations from ._locations_operations import LocationsOperations +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'AccountsOperations', - 'FirewallRulesOperations', - 'VirtualNetworkRulesOperations', - 'TrustedIdProvidersOperations', - 'Operations', - 'LocationsOperations', + "AccountsOperations", + "FirewallRulesOperations", + "VirtualNetworkRulesOperations", + "TrustedIdProvidersOperations", + "Operations", + "LocationsOperations", ] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_accounts_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_accounts_operations.py index 9ad51dca6a24..6b2060f5806e 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_accounts_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_accounts_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,44 +6,70 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar, Union -import warnings +import sys +from typing import Any, AsyncIterable, 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, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod +from azure.core.rest import 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 - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._accounts_operations import ( + build_check_name_availability_request, + build_create_request, + build_delete_request, + build_enable_key_vault_request, + build_get_request, + build_list_by_resource_group_request, + build_list_request, + build_update_request, +) + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class AccountsOperations: - """AccountsOperations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class AccountsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`accounts` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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, filter: Optional[str] = None, @@ -51,86 +78,102 @@ def list( select: Optional[str] = None, orderby: Optional[str] = None, count: Optional[bool] = None, - **kwargs - ) -> AsyncIterable["_models.DataLakeStoreAccountListResult"]: + **kwargs: Any + ) -> AsyncIterable["_models.DataLakeStoreAccountBasic"]: """Lists the Data Lake Store accounts within the subscription. The response includes a link to the next page of results, if any. - :param filter: OData filter. Optional. + :param filter: OData filter. Optional. Default value is None. :type filter: str - :param top: The number of items to return. Optional. + :param top: The number of items to return. Optional. Default value is None. :type top: int - :param skip: The number of items to skip over before returning elements. Optional. + :param skip: The number of items to skip over before returning elements. Optional. Default + value is None. :type skip: int :param select: OData Select statement. Limits the properties on each entry to just those - requested, e.g. Categories?$select=CategoryName,Description. Optional. + requested, e.g. Categories?$select=CategoryName,Description. Optional. Default value is None. :type select: str :param orderby: OrderBy clause. One or more comma-separated expressions with an optional "asc" (the default) or "desc" depending on the order you'd like the values sorted, e.g. - Categories?$orderby=CategoryName desc. Optional. + Categories?$orderby=CategoryName desc. Optional. Default value is None. :type orderby: str :param count: The Boolean value of true or false to request a count of the matching resources - included with the resources in the response, e.g. Categories?$count=true. Optional. + included with the resources in the response, e.g. Categories?$count=true. Optional. Default + value is None. :type count: bool :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DataLakeStoreAccountListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either DataLakeStoreAccountBasic or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountBasic] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccountListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccountListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int', minimum=1) - if skip is not None: - query_parameters['$skip'] = self._serialize.query("skip", skip, 'int', minimum=1) - if select is not None: - query_parameters['$select'] = self._serialize.query("select", select, 'str') - if orderby is not None: - query_parameters['$orderby'] = self._serialize.query("orderby", orderby, 'str') - if count is not None: - query_parameters['$count'] = self._serialize.query("count", count, 'bool') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + skip=skip, + select=select, + orderby=orderby, + count=count, + api_version=api_version, + template_url=self.list.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccountListResult', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccountListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -139,11 +182,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/accounts'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + + list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/accounts"} + @distributed_trace def list_by_resource_group( self, resource_group_name: str, @@ -153,89 +196,105 @@ def list_by_resource_group( select: Optional[str] = None, orderby: Optional[str] = None, count: Optional[bool] = None, - **kwargs - ) -> AsyncIterable["_models.DataLakeStoreAccountListResult"]: + **kwargs: Any + ) -> AsyncIterable["_models.DataLakeStoreAccountBasic"]: """Lists the Data Lake Store accounts within a specific resource group. The response includes a link to the next page of results, if any. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param filter: OData filter. Optional. + :param filter: OData filter. Optional. Default value is None. :type filter: str - :param top: The number of items to return. Optional. + :param top: The number of items to return. Optional. Default value is None. :type top: int - :param skip: The number of items to skip over before returning elements. Optional. + :param skip: The number of items to skip over before returning elements. Optional. Default + value is None. :type skip: int :param select: OData Select statement. Limits the properties on each entry to just those - requested, e.g. Categories?$select=CategoryName,Description. Optional. + requested, e.g. Categories?$select=CategoryName,Description. Optional. Default value is None. :type select: str :param orderby: OrderBy clause. One or more comma-separated expressions with an optional "asc" (the default) or "desc" depending on the order you'd like the values sorted, e.g. - Categories?$orderby=CategoryName desc. Optional. + Categories?$orderby=CategoryName desc. Optional. Default value is None. :type orderby: str :param count: A Boolean value of true or false to request a count of the matching resources - included with the resources in the response, e.g. Categories?$count=true. Optional. + included with the resources in the response, e.g. Categories?$count=true. Optional. Default + value is None. :type count: bool :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DataLakeStoreAccountListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either DataLakeStoreAccountBasic or the result of + cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountBasic] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccountListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccountListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int', minimum=1) - if skip is not None: - query_parameters['$skip'] = self._serialize.query("skip", skip, 'int', minimum=1) - if select is not None: - query_parameters['$select'] = self._serialize.query("select", select, 'str') - if orderby is not None: - query_parameters['$orderby'] = self._serialize.query("orderby", orderby, 'str') - if count is not None: - query_parameters['$count'] = self._serialize.query("count", count, 'bool') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + resource_group_name=resource_group_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + skip=skip, + select=select, + orderby=orderby, + count=count, + api_version=api_version, + template_url=self.list_by_resource_group.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccountListResult', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccountListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -244,50 +303,63 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + + list_by_resource_group.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts" + } async def _create_initial( self, resource_group_name: str, account_name: str, - parameters: "_models.CreateDataLakeStoreAccountParameters", - **kwargs - ) -> "_models.DataLakeStoreAccount": - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] + parameters: Union[_models.CreateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> _models.DataLakeStoreAccount: error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._create_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateDataLakeStoreAccountParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.DataLakeStoreAccount] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateDataLakeStoreAccountParameters") + + request = build_create_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self._create_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200, 201]: @@ -295,185 +367,289 @@ async def _create_initial( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - _create_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return deserialized # type: ignore + + _create_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + @overload async def begin_create( self, resource_group_name: str, account_name: str, - parameters: "_models.CreateDataLakeStoreAccountParameters", - **kwargs - ) -> AsyncLROPoller["_models.DataLakeStoreAccount"]: + parameters: _models.CreateDataLakeStoreAccountParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: """Creates the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param parameters: Parameters supplied to create the Data Lake Store account. + :param parameters: Parameters supplied to create the Data Lake Store account. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateDataLakeStoreAccountParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. - :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] :raises ~azure.core.exceptions.HttpResponseError: """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + + @overload + async def begin_create( + self, + resource_group_name: str, + account_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: + """Creates the specified Data Lake Store account. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to create the Data Lake Store account. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. + :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_create( + self, + resource_group_name: str, + account_name: str, + parameters: Union[_models.CreateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: + """Creates the specified Data Lake Store account. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to create the Data Lake Store account. Is either a + CreateDataLakeStoreAccountParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateDataLakeStoreAccountParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. + :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.DataLakeStoreAccount] = 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, account_name=account_name, parameters=parameters, - cls=lambda x,y,z: x, + api_version=api_version, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) - - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) - + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = AsyncNoPolling() - else: polling_method = polling + if polling is True: + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_create.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - async def get( - self, - resource_group_name: str, - account_name: str, - **kwargs - ) -> "_models.DataLakeStoreAccount": + begin_create.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @distributed_trace_async + async def get(self, resource_group_name: str, account_name: str, **kwargs: Any) -> _models.DataLakeStoreAccount: """Gets the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: DataLakeStoreAccount, or the result of cls(response) + :return: DataLakeStoreAccount or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.DataLakeStoreAccount - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccount] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } async def _update_initial( self, resource_group_name: str, account_name: str, - parameters: "_models.UpdateDataLakeStoreAccountParameters", - **kwargs - ) -> "_models.DataLakeStoreAccount": - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] + parameters: Union[_models.UpdateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> _models.DataLakeStoreAccount: error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._update_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'UpdateDataLakeStoreAccountParameters') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.DataLakeStoreAccount] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "UpdateDataLakeStoreAccountParameters") + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self._update_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: @@ -481,122 +657,214 @@ async def _update_initial( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 202: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - _update_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return deserialized # type: ignore + _update_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @overload async def begin_update( self, resource_group_name: str, account_name: str, - parameters: "_models.UpdateDataLakeStoreAccountParameters", - **kwargs - ) -> AsyncLROPoller["_models.DataLakeStoreAccount"]: + parameters: _models.UpdateDataLakeStoreAccountParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: """Updates the specified Data Lake Store account information. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param parameters: Parameters supplied to update the Data Lake Store account. + :param parameters: Parameters supplied to update the Data Lake Store account. Required. :type parameters: ~azure.mgmt.datalake.store.models.UpdateDataLakeStoreAccountParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. - :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result of cls(response) - :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] :raises ~azure.core.exceptions.HttpResponseError: """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + + @overload + async def begin_update( + self, + resource_group_name: str, + account_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: + """Updates the specified Data Lake Store account information. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to update the Data Lake Store account. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. + :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def begin_update( + self, + resource_group_name: str, + account_name: str, + parameters: Union[_models.UpdateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> AsyncLROPoller[_models.DataLakeStoreAccount]: + """Updates the specified Data Lake Store account information. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to update the Data Lake Store account. Is either a + UpdateDataLakeStoreAccountParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateDataLakeStoreAccountParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. + :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of AsyncLROPoller that returns either DataLakeStoreAccount or the result + of cls(response) + :rtype: + ~azure.core.polling.AsyncLROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.DataLakeStoreAccount] = 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, account_name=account_name, parameters=parameters, - cls=lambda x,y,z: x, + api_version=api_version, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) - - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) - + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = AsyncNoPolling() - else: polling_method = polling + if polling is True: + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - async def _delete_initial( - self, - resource_group_name: str, - account_name: str, - **kwargs + begin_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + async def _delete_initial( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, **kwargs: Any ) -> None: - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self._delete_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self._delete_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 202, 204]: @@ -606,115 +874,123 @@ async def _delete_initial( if cls: return cls(pipeline_response, None, {}) - _delete_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + _delete_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } - async def begin_delete( - self, - resource_group_name: str, - account_name: str, - **kwargs - ) -> AsyncLROPoller[None]: + @distributed_trace_async + async def begin_delete(self, resource_group_name: str, account_name: str, **kwargs: Any) -> AsyncLROPoller[None]: """Deletes the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for + this operation to not poll, or pass in your own initialized polling object for a personal + polling strategy. :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :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: """ - polling = kwargs.pop('polling', True) # type: Union[bool, AsyncPollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType[None] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + 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( + raw_result = await self._delete_initial( # type: ignore resource_group_name=resource_group_name, account_name=account_name, - cls=lambda x,y,z: x, + api_version=api_version, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) + kwargs.pop("error_map", None) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) - - def get_long_running_output(pipeline_response): + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: return cls(pipeline_response, None, {}) - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = AsyncARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = AsyncNoPolling() - else: polling_method = polling + if polling is True: + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - async def enable_key_vault( - self, - resource_group_name: str, - account_name: str, - **kwargs + begin_delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @distributed_trace_async + async def enable_key_vault( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, **kwargs: Any ) -> None: """Attempts to enable a user managed Key Vault for encryption of the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.enable_key_vault.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_enable_key_vault_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.enable_key_vault.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.post(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: @@ -724,66 +1000,130 @@ async def enable_key_vault( if cls: return cls(pipeline_response, None, {}) - enable_key_vault.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/enableKeyVault'} # type: ignore + enable_key_vault.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/enableKeyVault" + } + @overload async def check_name_availability( self, location: str, - parameters: "_models.CheckNameAvailabilityParameters", - **kwargs - ) -> "_models.NameAvailabilityInformation": + parameters: _models.CheckNameAvailabilityParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NameAvailabilityInformation: """Checks whether the specified account name is available or taken. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Required. :type parameters: ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: NameAvailabilityInformation, or the result of cls(response) + :return: NameAvailabilityInformation or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def check_name_availability( + self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.NameAvailabilityInformation: + """Checks whether the specified account name is available or taken. + + :param location: The resource location without whitespace. Required. + :type location: str + :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: NameAvailabilityInformation or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def check_name_availability( + self, location: str, parameters: Union[_models.CheckNameAvailabilityParameters, IO], **kwargs: Any + ) -> _models.NameAvailabilityInformation: + """Checks whether the specified account name is available or taken. + + :param location: The resource location without whitespace. Required. + :type location: str + :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Is either a CheckNameAvailabilityParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: NameAvailabilityInformation or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.NameAvailabilityInformation"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.check_name_availability.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CheckNameAvailabilityParameters') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.NameAvailabilityInformation] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CheckNameAvailabilityParameters") + + request = build_check_name_availability_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.check_name_availability.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('NameAvailabilityInformation', pipeline_response) + deserialized = self._deserialize("NameAvailabilityInformation", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - check_name_availability.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/checkNameAvailability'} # type: ignore + + check_name_availability.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/checkNameAvailability" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_firewall_rules_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_firewall_rules_operations.py index 4ef2a8c20692..c495eb470c68 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_firewall_rules_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_firewall_rules_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,103 +6,141 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar -import warnings +import sys +from typing import Any, AsyncIterable, Callable, Dict, IO, Optional, TypeVar, Union, overload +import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._firewall_rules_operations import ( + build_create_or_update_request, + build_delete_request, + build_get_request, + build_list_by_account_request, + build_update_request, +) + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class FirewallRulesOperations: - """FirewallRulesOperations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class FirewallRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`firewall_rules` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name: str, - account_name: str, - **kwargs - ) -> AsyncIterable["_models.FirewallRuleListResult"]: + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> AsyncIterable["_models.FirewallRule"]: """Lists the Data Lake Store firewall rules within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either FirewallRuleListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.FirewallRuleListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either FirewallRule or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.FirewallRule] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRuleListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.FirewallRuleListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('FirewallRuleListResult', pipeline_response) + deserialized = self._deserialize("FirewallRuleListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -110,263 +149,435 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules" + } + + @overload async def create_or_update( self, resource_group_name: str, account_name: str, firewall_rule_name: str, - parameters: "_models.CreateOrUpdateFirewallRuleParameters", - **kwargs - ) -> "_models.FirewallRule": + parameters: _models.CreateOrUpdateFirewallRuleParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: """Creates or updates the specified firewall rule. During update, the firewall rule with the specified name will be replaced with this new firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to create or update. + :param firewall_rule_name: The name of the firewall rule to create or update. Required. :type firewall_rule_name: str - :param parameters: Parameters supplied to create or update the firewall rule. + :param parameters: Parameters supplied to create or update the firewall rule. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateFirewallRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: + """Creates or updates the specified firewall rule. During update, the firewall rule with the + specified name will be replaced with this new firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to create or update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to create or update the firewall rule. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Union[_models.CreateOrUpdateFirewallRuleParameters, IO], + **kwargs: Any + ) -> _models.FirewallRule: + """Creates or updates the specified firewall rule. During update, the firewall rule with the + specified name will be replaced with this new firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to create or update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to create or update the firewall rule. Is either a + CreateOrUpdateFirewallRuleParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateFirewallRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateFirewallRuleParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.FirewallRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateFirewallRuleParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @distributed_trace_async async def get( - self, - resource_group_name: str, - account_name: str, - firewall_rule_name: str, - **kwargs - ) -> "_models.FirewallRule": + self, resource_group_name: str, account_name: str, firewall_rule_name: str, **kwargs: Any + ) -> _models.FirewallRule: """Gets the specified Data Lake Store firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to retrieve. + :param firewall_rule_name: The name of the firewall rule to retrieve. Required. :type firewall_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.FirewallRule] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @overload async def update( self, resource_group_name: str, account_name: str, firewall_rule_name: str, - parameters: Optional["_models.UpdateFirewallRuleParameters"] = None, - **kwargs - ) -> "_models.FirewallRule": + parameters: Optional[_models.UpdateFirewallRuleParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: """Updates the specified firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to update. + :param firewall_rule_name: The name of the firewall rule to update. Required. :type firewall_rule_name: str - :param parameters: Parameters supplied to update the firewall rule. + :param parameters: Parameters supplied to update the firewall rule. Default value is None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateFirewallRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: + """Updates the specified firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to update the firewall rule. Default value is None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Optional[Union[_models.UpdateFirewallRuleParameters, IO]] = None, + **kwargs: Any + ) -> _models.FirewallRule: + """Updates the specified firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to update the firewall rule. Is either a + UpdateFirewallRuleParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateFirewallRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateFirewallRuleParameters') + api_version: Literal["2016-11-01"] = 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.FirewallRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateFirewallRuleParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore - async def delete( - self, - resource_group_name: str, - account_name: str, - firewall_rule_name: str, - **kwargs + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, firewall_rule_name: str, **kwargs: Any ) -> None: """Deletes the specified firewall rule from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to delete. + :param firewall_rule_name: The name of the firewall rule to delete. Required. :type firewall_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -376,4 +587,6 @@ async def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_locations_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_locations_operations.py index 6344204350ce..7af28a572a71 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_locations_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_locations_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,83 +6,101 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar -import warnings +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, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._locations_operations import build_get_capability_request, build_get_usage_request + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class LocationsOperations: - """LocationsOperations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class LocationsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`locations` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config - - async def get_capability( - self, - location: str, - **kwargs - ) -> Optional["_models.CapabilityInformation"]: + 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_async + async def get_capability(self, location: str, **kwargs: Any) -> Optional[_models.CapabilityInformation]: """Gets subscription-level properties and limits for Data Lake Store specified by resource location. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: CapabilityInformation, or the result of cls(response) + :return: CapabilityInformation or None or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.CapabilityInformation or None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[Optional["_models.CapabilityInformation"]] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get_capability.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[Optional[_models.CapabilityInformation]] = kwargs.pop("cls", None) + + request = build_get_capability_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get_capability.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 404]: @@ -90,72 +109,90 @@ async def get_capability( deserialized = None if response.status_code == 200: - deserialized = self._deserialize('CapabilityInformation', pipeline_response) + deserialized = self._deserialize("CapabilityInformation", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get_capability.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/capability'} # type: ignore - def get_usage( - self, - location: str, - **kwargs - ) -> AsyncIterable["_models.UsageListResult"]: + get_capability.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/capability" + } + + @distributed_trace + def get_usage(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: """Gets the current usage count and the limit for the resources of the location under the subscription. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either UsageListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.UsageListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either Usage or the result of cls(response) + :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.Usage] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.UsageListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.UsageListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.get_usage.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_get_usage_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get_usage.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('UsageListResult', pipeline_response) + deserialized = self._deserialize("UsageListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + list_of_elem = cls(list_of_elem) # type: ignore return None, AsyncList(list_of_elem) async def get_next(next_link=None): request = prepare_request(next_link) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -164,7 +201,8 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - get_usage.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/usages'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + + get_usage.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/usages" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_operations.py index 0252f15cbccf..efd91a316f1b 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,83 +6,104 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Callable, Dict, Generic, Optional, TypeVar -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, Optional, TypeVar + +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._operations import build_list_request + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class Operations: - """Operations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class Operations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`operations` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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 list( - self, - **kwargs - ) -> "_models.OperationListResult": + @distributed_trace_async + async def list(self, **kwargs: Any) -> _models.OperationListResult: """Lists all of the available Data Lake Store REST API operations. :keyword callable cls: A custom type or function that will be passed the direct response - :return: OperationListResult, or the result of cls(response) + :return: OperationListResult or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.OperationListResult - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.OperationListResult"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct URL - url = self.list.metadata['url'] # type: ignore + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + request = build_list_request( + api_version=api_version, + template_url=self.list.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('OperationListResult', pipeline_response) + deserialized = self._deserialize("OperationListResult", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - list.metadata = {'url': '/providers/Microsoft.DataLakeStore/operations'} # type: ignore + + list.metadata = {"url": "/providers/Microsoft.DataLakeStore/operations"} diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_patch.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_trusted_id_providers_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_trusted_id_providers_operations.py index 2579c72572c1..2d644c8c7a9b 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_trusted_id_providers_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_trusted_id_providers_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,104 +6,143 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar -import warnings +import sys +from typing import Any, AsyncIterable, Callable, Dict, IO, Optional, TypeVar, Union, overload +import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._trusted_id_providers_operations import ( + build_create_or_update_request, + build_delete_request, + build_get_request, + build_list_by_account_request, + build_update_request, +) + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class TrustedIdProvidersOperations: - """TrustedIdProvidersOperations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class TrustedIdProvidersOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`trusted_id_providers` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name: str, - account_name: str, - **kwargs - ) -> AsyncIterable["_models.TrustedIdProviderListResult"]: + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> AsyncIterable["_models.TrustedIdProvider"]: """Lists the Data Lake Store trusted identity providers within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either TrustedIdProviderListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.TrustedIdProviderListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either TrustedIdProvider or the result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.TrustedIdProvider] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProviderListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.TrustedIdProviderListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('TrustedIdProviderListResult', pipeline_response) + deserialized = self._deserialize("TrustedIdProviderListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -111,265 +151,447 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders" + } + @overload async def create_or_update( self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, - parameters: "_models.CreateOrUpdateTrustedIdProviderParameters", - **kwargs - ) -> "_models.TrustedIdProvider": + parameters: _models.CreateOrUpdateTrustedIdProviderParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: """Creates or updates the specified trusted identity provider. During update, the trusted identity provider with the specified name will be replaced with this new provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider. This is used for - differentiation of providers in the account. + differentiation of providers in the account. Required. :type trusted_id_provider_name: str :param parameters: Parameters supplied to create or replace the trusted identity provider. + Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateTrustedIdProviderParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Creates or updates the specified trusted identity provider. During update, the trusted identity + provider with the specified name will be replaced with this new provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to create or replace the trusted identity provider. + Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Union[_models.CreateOrUpdateTrustedIdProviderParameters, IO], + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Creates or updates the specified trusted identity provider. During update, the trusted identity + provider with the specified name will be replaced with this new provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to create or replace the trusted identity provider. Is + either a CreateOrUpdateTrustedIdProviderParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateTrustedIdProviderParameters + or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateTrustedIdProviderParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.TrustedIdProvider] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateTrustedIdProviderParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @distributed_trace_async async def get( - self, - resource_group_name: str, - account_name: str, - trusted_id_provider_name: str, - **kwargs - ) -> "_models.TrustedIdProvider": + self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, **kwargs: Any + ) -> _models.TrustedIdProvider: """Gets the specified Data Lake Store trusted identity provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider to retrieve. + Required. :type trusted_id_provider_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.TrustedIdProvider] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @overload async def update( self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, - parameters: Optional["_models.UpdateTrustedIdProviderParameters"] = None, - **kwargs - ) -> "_models.TrustedIdProvider": + parameters: Optional[_models.UpdateTrustedIdProviderParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: """Updates the specified trusted identity provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider. This is used for - differentiation of providers in the account. + differentiation of providers in the account. Required. :type trusted_id_provider_name: str - :param parameters: Parameters supplied to update the trusted identity provider. + :param parameters: Parameters supplied to update the trusted identity provider. Default value + is None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Updates the specified trusted identity provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to update the trusted identity provider. Default value + is None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Optional[Union[_models.UpdateTrustedIdProviderParameters, IO]] = None, + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Updates the specified trusted identity provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to update the trusted identity provider. Is either a + UpdateTrustedIdProviderParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateTrustedIdProviderParameters') + api_version: Literal["2016-11-01"] = 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.TrustedIdProvider] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateTrustedIdProviderParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore - async def delete( - self, - resource_group_name: str, - account_name: str, - trusted_id_provider_name: str, - **kwargs + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, **kwargs: Any ) -> None: """Deletes the specified trusted identity provider from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param trusted_id_provider_name: The name of the trusted identity provider to delete. + :param trusted_id_provider_name: The name of the trusted identity provider to delete. Required. :type trusted_id_provider_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -379,4 +601,6 @@ async def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_virtual_network_rules_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_virtual_network_rules_operations.py index fa1adf3d443e..caf690972899 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_virtual_network_rules_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/aio/operations/_virtual_network_rules_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,103 +6,142 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, AsyncIterable, Callable, Dict, Generic, Optional, TypeVar -import warnings +import sys +from typing import Any, AsyncIterable, Callable, Dict, IO, Optional, TypeVar, Union, overload +import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import AsyncHttpResponse, HttpRequest +from azure.core.pipeline.transport import AsyncHttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.tracing.decorator_async import distributed_trace_async +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models - -T = TypeVar('T') +from ..._vendor import _convert_request +from ...operations._virtual_network_rules_operations import ( + build_create_or_update_request, + build_delete_request, + build_get_request, + build_list_by_account_request, + build_update_request, +) + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class VirtualNetworkRulesOperations: - """VirtualNetworkRulesOperations async operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +class VirtualNetworkRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.aio.DataLakeStoreAccountManagementClient`'s + :attr:`virtual_network_rules` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer) -> None: - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name: str, - account_name: str, - **kwargs - ) -> AsyncIterable["_models.VirtualNetworkRuleListResult"]: + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> AsyncIterable["_models.VirtualNetworkRule"]: """Lists the Data Lake Store virtual network rules within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either VirtualNetworkRuleListResult or the result of cls(response) - :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.VirtualNetworkRuleListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either VirtualNetworkRule or the result of cls(response) + :rtype: + ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.datalake.store.models.VirtualNetworkRule] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRuleListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.VirtualNetworkRuleListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request async def extract_data(pipeline_response): - deserialized = self._deserialize('VirtualNetworkRuleListResult', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRuleListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -110,263 +150,441 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules'} # type: ignore + return AsyncItemPaged(get_next, extract_data) + + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules" + } + @overload async def create_or_update( self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, - parameters: "_models.CreateOrUpdateVirtualNetworkRuleParameters", - **kwargs - ) -> "_models.VirtualNetworkRule": + parameters: _models.CreateOrUpdateVirtualNetworkRuleParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: """Creates or updates the specified virtual network rule. During update, the virtual network rule with the specified name will be replaced with this new virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. :type virtual_network_rule_name: str - :param parameters: Parameters supplied to create or update the virtual network rule. + :param parameters: Parameters supplied to create or update the virtual network rule. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateVirtualNetworkRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Creates or updates the specified virtual network rule. During update, the virtual network rule + with the specified name will be replaced with this new virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to create or update the virtual network rule. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def create_or_update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Union[_models.CreateOrUpdateVirtualNetworkRuleParameters, IO], + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Creates or updates the specified virtual network rule. During update, the virtual network rule + with the specified name will be replaced with this new virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to create or update the virtual network rule. Is either + a CreateOrUpdateVirtualNetworkRuleParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateVirtualNetworkRuleParameters + or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateVirtualNetworkRuleParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.VirtualNetworkRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateVirtualNetworkRuleParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @distributed_trace_async async def get( - self, - resource_group_name: str, - account_name: str, - virtual_network_rule_name: str, - **kwargs - ) -> "_models.VirtualNetworkRule": + self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, **kwargs: Any + ) -> _models.VirtualNetworkRule: """Gets the specified Data Lake Store virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to retrieve. + :param virtual_network_rule_name: The name of the virtual network rule to retrieve. Required. :type virtual_network_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.VirtualNetworkRule] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @overload async def update( self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, - parameters: Optional["_models.UpdateVirtualNetworkRuleParameters"] = None, - **kwargs - ) -> "_models.VirtualNetworkRule": + parameters: Optional[_models.UpdateVirtualNetworkRuleParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: """Updates the specified virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to update. + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. :type virtual_network_rule_name: str - :param parameters: Parameters supplied to update the virtual network rule. + :param parameters: Parameters supplied to update the virtual network rule. Default value is + None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + async def update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Updates the specified virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to update the virtual network rule. Default value is + None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace_async + async def update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Optional[Union[_models.UpdateVirtualNetworkRuleParameters, IO]] = None, + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Updates the specified virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to update the virtual network rule. Is either a + UpdateVirtualNetworkRuleParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateVirtualNetworkRuleParameters') + api_version: Literal["2016-11-01"] = 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.VirtualNetworkRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateVirtualNetworkRuleParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore - async def delete( - self, - resource_group_name: str, - account_name: str, - virtual_network_rule_name: str, - **kwargs + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @distributed_trace_async + async def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, **kwargs: Any ) -> None: """Deletes the specified virtual network rule from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to delete. + :param virtual_network_rule_name: The name of the virtual network rule to delete. Required. :type virtual_network_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = await self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -376,4 +594,6 @@ async def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/__init__.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/__init__.py index 91f2b0a2e465..a3c3a3770ed9 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/__init__.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/__init__.py @@ -6,158 +6,120 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -try: - from ._models_py3 import CapabilityInformation - from ._models_py3 import CheckNameAvailabilityParameters - from ._models_py3 import CreateDataLakeStoreAccountParameters - from ._models_py3 import CreateFirewallRuleWithAccountParameters - from ._models_py3 import CreateOrUpdateFirewallRuleParameters - from ._models_py3 import CreateOrUpdateTrustedIdProviderParameters - from ._models_py3 import CreateOrUpdateVirtualNetworkRuleParameters - from ._models_py3 import CreateTrustedIdProviderWithAccountParameters - from ._models_py3 import CreateVirtualNetworkRuleWithAccountParameters - from ._models_py3 import DataLakeStoreAccount - from ._models_py3 import DataLakeStoreAccountBasic - from ._models_py3 import DataLakeStoreAccountListResult - from ._models_py3 import DataLakeStoreAccountProperties - from ._models_py3 import DataLakeStoreAccountPropertiesBasic - from ._models_py3 import EncryptionConfig - from ._models_py3 import EncryptionIdentity - from ._models_py3 import FirewallRule - from ._models_py3 import FirewallRuleListResult - from ._models_py3 import KeyVaultMetaInfo - from ._models_py3 import NameAvailabilityInformation - from ._models_py3 import Operation - from ._models_py3 import OperationDisplay - from ._models_py3 import OperationListResult - from ._models_py3 import Resource - from ._models_py3 import SubResource - from ._models_py3 import TrustedIdProvider - from ._models_py3 import TrustedIdProviderListResult - from ._models_py3 import UpdateDataLakeStoreAccountParameters - from ._models_py3 import UpdateEncryptionConfig - from ._models_py3 import UpdateFirewallRuleParameters - from ._models_py3 import UpdateFirewallRuleWithAccountParameters - from ._models_py3 import UpdateKeyVaultMetaInfo - from ._models_py3 import UpdateTrustedIdProviderParameters - from ._models_py3 import UpdateTrustedIdProviderWithAccountParameters - from ._models_py3 import UpdateVirtualNetworkRuleParameters - from ._models_py3 import UpdateVirtualNetworkRuleWithAccountParameters - from ._models_py3 import Usage - from ._models_py3 import UsageListResult - from ._models_py3 import UsageName - from ._models_py3 import VirtualNetworkRule - from ._models_py3 import VirtualNetworkRuleListResult -except (SyntaxError, ImportError): - from ._models import CapabilityInformation # type: ignore - from ._models import CheckNameAvailabilityParameters # type: ignore - from ._models import CreateDataLakeStoreAccountParameters # type: ignore - from ._models import CreateFirewallRuleWithAccountParameters # type: ignore - from ._models import CreateOrUpdateFirewallRuleParameters # type: ignore - from ._models import CreateOrUpdateTrustedIdProviderParameters # type: ignore - from ._models import CreateOrUpdateVirtualNetworkRuleParameters # type: ignore - from ._models import CreateTrustedIdProviderWithAccountParameters # type: ignore - from ._models import CreateVirtualNetworkRuleWithAccountParameters # type: ignore - from ._models import DataLakeStoreAccount # type: ignore - from ._models import DataLakeStoreAccountBasic # type: ignore - from ._models import DataLakeStoreAccountListResult # type: ignore - from ._models import DataLakeStoreAccountProperties # type: ignore - from ._models import DataLakeStoreAccountPropertiesBasic # type: ignore - from ._models import EncryptionConfig # type: ignore - from ._models import EncryptionIdentity # type: ignore - from ._models import FirewallRule # type: ignore - from ._models import FirewallRuleListResult # type: ignore - from ._models import KeyVaultMetaInfo # type: ignore - from ._models import NameAvailabilityInformation # type: ignore - from ._models import Operation # type: ignore - from ._models import OperationDisplay # type: ignore - from ._models import OperationListResult # type: ignore - from ._models import Resource # type: ignore - from ._models import SubResource # type: ignore - from ._models import TrustedIdProvider # type: ignore - from ._models import TrustedIdProviderListResult # type: ignore - from ._models import UpdateDataLakeStoreAccountParameters # type: ignore - from ._models import UpdateEncryptionConfig # type: ignore - from ._models import UpdateFirewallRuleParameters # type: ignore - from ._models import UpdateFirewallRuleWithAccountParameters # type: ignore - from ._models import UpdateKeyVaultMetaInfo # type: ignore - from ._models import UpdateTrustedIdProviderParameters # type: ignore - from ._models import UpdateTrustedIdProviderWithAccountParameters # type: ignore - from ._models import UpdateVirtualNetworkRuleParameters # type: ignore - from ._models import UpdateVirtualNetworkRuleWithAccountParameters # type: ignore - from ._models import Usage # type: ignore - from ._models import UsageListResult # type: ignore - from ._models import UsageName # type: ignore - from ._models import VirtualNetworkRule # type: ignore - from ._models import VirtualNetworkRuleListResult # type: ignore +from ._models_py3 import CapabilityInformation +from ._models_py3 import CheckNameAvailabilityParameters +from ._models_py3 import CreateDataLakeStoreAccountParameters +from ._models_py3 import CreateFirewallRuleWithAccountParameters +from ._models_py3 import CreateOrUpdateFirewallRuleParameters +from ._models_py3 import CreateOrUpdateTrustedIdProviderParameters +from ._models_py3 import CreateOrUpdateVirtualNetworkRuleParameters +from ._models_py3 import CreateTrustedIdProviderWithAccountParameters +from ._models_py3 import CreateVirtualNetworkRuleWithAccountParameters +from ._models_py3 import DataLakeStoreAccount +from ._models_py3 import DataLakeStoreAccountBasic +from ._models_py3 import DataLakeStoreAccountListResult +from ._models_py3 import DataLakeStoreAccountProperties +from ._models_py3 import DataLakeStoreAccountPropertiesBasic +from ._models_py3 import EncryptionConfig +from ._models_py3 import EncryptionIdentity +from ._models_py3 import FirewallRule +from ._models_py3 import FirewallRuleListResult +from ._models_py3 import KeyVaultMetaInfo +from ._models_py3 import NameAvailabilityInformation +from ._models_py3 import Operation +from ._models_py3 import OperationDisplay +from ._models_py3 import OperationListResult +from ._models_py3 import Resource +from ._models_py3 import SubResource +from ._models_py3 import TrustedIdProvider +from ._models_py3 import TrustedIdProviderListResult +from ._models_py3 import UpdateDataLakeStoreAccountParameters +from ._models_py3 import UpdateEncryptionConfig +from ._models_py3 import UpdateFirewallRuleParameters +from ._models_py3 import UpdateFirewallRuleWithAccountParameters +from ._models_py3 import UpdateKeyVaultMetaInfo +from ._models_py3 import UpdateTrustedIdProviderParameters +from ._models_py3 import UpdateTrustedIdProviderWithAccountParameters +from ._models_py3 import UpdateVirtualNetworkRuleParameters +from ._models_py3 import UpdateVirtualNetworkRuleWithAccountParameters +from ._models_py3 import Usage +from ._models_py3 import UsageListResult +from ._models_py3 import UsageName +from ._models_py3 import VirtualNetworkRule +from ._models_py3 import VirtualNetworkRuleListResult -from ._data_lake_store_account_management_client_enums import ( - DataLakeStoreAccountState, - DataLakeStoreAccountStatus, - EncryptionConfigType, - EncryptionProvisioningState, - EncryptionState, - FirewallAllowAzureIpsState, - FirewallState, - OperationOrigin, - SubscriptionState, - TierType, - TrustedIdProviderState, - UsageUnit, -) +from ._data_lake_store_account_management_client_enums import CheckNameAvailabilityParametersType +from ._data_lake_store_account_management_client_enums import DataLakeStoreAccountState +from ._data_lake_store_account_management_client_enums import DataLakeStoreAccountStatus +from ._data_lake_store_account_management_client_enums import EncryptionConfigType +from ._data_lake_store_account_management_client_enums import EncryptionProvisioningState +from ._data_lake_store_account_management_client_enums import EncryptionState +from ._data_lake_store_account_management_client_enums import FirewallAllowAzureIpsState +from ._data_lake_store_account_management_client_enums import FirewallState +from ._data_lake_store_account_management_client_enums import OperationOrigin +from ._data_lake_store_account_management_client_enums import SubscriptionState +from ._data_lake_store_account_management_client_enums import TierType +from ._data_lake_store_account_management_client_enums import TrustedIdProviderState +from ._data_lake_store_account_management_client_enums import UsageUnit +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk __all__ = [ - 'CapabilityInformation', - 'CheckNameAvailabilityParameters', - 'CreateDataLakeStoreAccountParameters', - 'CreateFirewallRuleWithAccountParameters', - 'CreateOrUpdateFirewallRuleParameters', - 'CreateOrUpdateTrustedIdProviderParameters', - 'CreateOrUpdateVirtualNetworkRuleParameters', - 'CreateTrustedIdProviderWithAccountParameters', - 'CreateVirtualNetworkRuleWithAccountParameters', - 'DataLakeStoreAccount', - 'DataLakeStoreAccountBasic', - 'DataLakeStoreAccountListResult', - 'DataLakeStoreAccountProperties', - 'DataLakeStoreAccountPropertiesBasic', - 'EncryptionConfig', - 'EncryptionIdentity', - 'FirewallRule', - 'FirewallRuleListResult', - 'KeyVaultMetaInfo', - 'NameAvailabilityInformation', - 'Operation', - 'OperationDisplay', - 'OperationListResult', - 'Resource', - 'SubResource', - 'TrustedIdProvider', - 'TrustedIdProviderListResult', - 'UpdateDataLakeStoreAccountParameters', - 'UpdateEncryptionConfig', - 'UpdateFirewallRuleParameters', - 'UpdateFirewallRuleWithAccountParameters', - 'UpdateKeyVaultMetaInfo', - 'UpdateTrustedIdProviderParameters', - 'UpdateTrustedIdProviderWithAccountParameters', - 'UpdateVirtualNetworkRuleParameters', - 'UpdateVirtualNetworkRuleWithAccountParameters', - 'Usage', - 'UsageListResult', - 'UsageName', - 'VirtualNetworkRule', - 'VirtualNetworkRuleListResult', - 'DataLakeStoreAccountState', - 'DataLakeStoreAccountStatus', - 'EncryptionConfigType', - 'EncryptionProvisioningState', - 'EncryptionState', - 'FirewallAllowAzureIpsState', - 'FirewallState', - 'OperationOrigin', - 'SubscriptionState', - 'TierType', - 'TrustedIdProviderState', - 'UsageUnit', + "CapabilityInformation", + "CheckNameAvailabilityParameters", + "CreateDataLakeStoreAccountParameters", + "CreateFirewallRuleWithAccountParameters", + "CreateOrUpdateFirewallRuleParameters", + "CreateOrUpdateTrustedIdProviderParameters", + "CreateOrUpdateVirtualNetworkRuleParameters", + "CreateTrustedIdProviderWithAccountParameters", + "CreateVirtualNetworkRuleWithAccountParameters", + "DataLakeStoreAccount", + "DataLakeStoreAccountBasic", + "DataLakeStoreAccountListResult", + "DataLakeStoreAccountProperties", + "DataLakeStoreAccountPropertiesBasic", + "EncryptionConfig", + "EncryptionIdentity", + "FirewallRule", + "FirewallRuleListResult", + "KeyVaultMetaInfo", + "NameAvailabilityInformation", + "Operation", + "OperationDisplay", + "OperationListResult", + "Resource", + "SubResource", + "TrustedIdProvider", + "TrustedIdProviderListResult", + "UpdateDataLakeStoreAccountParameters", + "UpdateEncryptionConfig", + "UpdateFirewallRuleParameters", + "UpdateFirewallRuleWithAccountParameters", + "UpdateKeyVaultMetaInfo", + "UpdateTrustedIdProviderParameters", + "UpdateTrustedIdProviderWithAccountParameters", + "UpdateVirtualNetworkRuleParameters", + "UpdateVirtualNetworkRuleWithAccountParameters", + "Usage", + "UsageListResult", + "UsageName", + "VirtualNetworkRule", + "VirtualNetworkRuleListResult", + "CheckNameAvailabilityParametersType", + "DataLakeStoreAccountState", + "DataLakeStoreAccountStatus", + "EncryptionConfigType", + "EncryptionProvisioningState", + "EncryptionState", + "FirewallAllowAzureIpsState", + "FirewallState", + "OperationOrigin", + "SubscriptionState", + "TierType", + "TrustedIdProviderState", + "UsageUnit", ] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_data_lake_store_account_management_client_enums.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_data_lake_store_account_management_client_enums.py index cf54bcdedb73..eea78254f8a3 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_data_lake_store_account_management_client_enums.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_data_lake_store_account_management_client_enums.py @@ -6,36 +6,27 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from enum import Enum, EnumMeta -from six import with_metaclass - -class _CaseInsensitiveEnumMeta(EnumMeta): - def __getitem__(self, name): - return super().__getitem__(name.upper()) - - def __getattr__(cls, name): - """Return the enum member matching `name` - We use __getattr__ instead of descriptors or inserting into the enum - class' __dict__ in order to support `name` and `value` being both - properties for enum members (which live in the class' __dict__) and - enum members themselves. - """ - try: - return cls._member_map_[name.upper()] - except KeyError: - raise AttributeError(name) - - -class DataLakeStoreAccountState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The state of the Data Lake Store account. +from enum import Enum +from azure.core import CaseInsensitiveEnumMeta + + +class CheckNameAvailabilityParametersType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The resource type. Note: This should not be set by the user, as the constant value is + Microsoft.DataLakeStore/accounts. """ + MICROSOFT_DATA_LAKE_STORE_ACCOUNTS = "Microsoft.DataLakeStore/accounts" + + +class DataLakeStoreAccountState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The state of the Data Lake Store account.""" + ACTIVE = "Active" SUSPENDED = "Suspended" -class DataLakeStoreAccountStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The provisioning status of the Data Lake Store account. - """ + +class DataLakeStoreAccountStatus(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The provisioning status of the Data Lake Store account.""" FAILED = "Failed" CREATING = "Creating" @@ -49,7 +40,8 @@ class DataLakeStoreAccountStatus(with_metaclass(_CaseInsensitiveEnumMeta, str, E UNDELETING = "Undeleting" CANCELED = "Canceled" -class EncryptionConfigType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): + +class EncryptionConfigType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The type of encryption configuration being used. Currently the only supported types are 'UserManaged' and 'ServiceManaged'. """ @@ -57,21 +49,22 @@ class EncryptionConfigType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): USER_MANAGED = "UserManaged" SERVICE_MANAGED = "ServiceManaged" -class EncryptionProvisioningState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The current state of encryption provisioning for this Data Lake Store account. - """ + +class EncryptionProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The current state of encryption provisioning for this Data Lake Store account.""" CREATING = "Creating" SUCCEEDED = "Succeeded" -class EncryptionState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The current state of encryption for this Data Lake Store account. - """ + +class EncryptionState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The current state of encryption for this Data Lake Store account.""" ENABLED = "Enabled" DISABLED = "Disabled" -class FirewallAllowAzureIpsState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): + +class FirewallAllowAzureIpsState(str, Enum, metaclass=CaseInsensitiveEnumMeta): """The current state of allowing or disallowing IPs originating within Azure through the firewall. If the firewall is disabled, this is not enforced. """ @@ -79,24 +72,24 @@ class FirewallAllowAzureIpsState(with_metaclass(_CaseInsensitiveEnumMeta, str, E ENABLED = "Enabled" DISABLED = "Disabled" -class FirewallState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The current state of the IP address firewall for this Data Lake Store account. - """ + +class FirewallState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The current state of the IP address firewall for this Data Lake Store account.""" ENABLED = "Enabled" DISABLED = "Disabled" -class OperationOrigin(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The intended executor of the operation. - """ + +class OperationOrigin(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The intended executor of the operation.""" USER = "user" SYSTEM = "system" USER_SYSTEM = "user,system" -class SubscriptionState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The subscription state. - """ + +class SubscriptionState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The subscription state.""" REGISTERED = "Registered" SUSPENDED = "Suspended" @@ -104,9 +97,9 @@ class SubscriptionState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): UNREGISTERED = "Unregistered" WARNED = "Warned" -class TierType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The commitment tier to use for next month. - """ + +class TierType(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The commitment tier to use for next month.""" CONSUMPTION = "Consumption" COMMITMENT1_TB = "Commitment_1TB" @@ -116,16 +109,16 @@ class TierType(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): COMMITMENT1_PB = "Commitment_1PB" COMMITMENT5_PB = "Commitment_5PB" -class TrustedIdProviderState(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """The current state of the trusted identity provider feature for this Data Lake Store account. - """ + +class TrustedIdProviderState(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """The current state of the trusted identity provider feature for this Data Lake Store account.""" ENABLED = "Enabled" DISABLED = "Disabled" -class UsageUnit(with_metaclass(_CaseInsensitiveEnumMeta, str, Enum)): - """Gets the unit of measurement. - """ + +class UsageUnit(str, Enum, metaclass=CaseInsensitiveEnumMeta): + """Gets the unit of measurement.""" COUNT = "Count" BYTES = "Bytes" diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models.py deleted file mode 100644 index d27e5d49e658..000000000000 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models.py +++ /dev/null @@ -1,1695 +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 msrest.serialization - - -class CapabilityInformation(msrest.serialization.Model): - """Subscription-level properties and limits for Data Lake Store. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar subscription_id: The subscription credentials that uniquely identifies the subscription. - :vartype subscription_id: str - :ivar state: The subscription state. Possible values include: "Registered", "Suspended", - "Deleted", "Unregistered", "Warned". - :vartype state: str or ~azure.mgmt.datalake.store.models.SubscriptionState - :ivar max_account_count: The maximum supported number of accounts under this subscription. - :vartype max_account_count: int - :ivar account_count: The current number of accounts under this subscription. - :vartype account_count: int - :ivar migration_state: The Boolean value of true or false to indicate the maintenance state. - :vartype migration_state: bool - """ - - _validation = { - 'subscription_id': {'readonly': True}, - 'state': {'readonly': True}, - 'max_account_count': {'readonly': True}, - 'account_count': {'readonly': True}, - 'migration_state': {'readonly': True}, - } - - _attribute_map = { - 'subscription_id': {'key': 'subscriptionId', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'max_account_count': {'key': 'maxAccountCount', 'type': 'int'}, - 'account_count': {'key': 'accountCount', 'type': 'int'}, - 'migration_state': {'key': 'migrationState', 'type': 'bool'}, - } - - def __init__( - self, - **kwargs - ): - super(CapabilityInformation, self).__init__(**kwargs) - self.subscription_id = None - self.state = None - self.max_account_count = None - self.account_count = None - self.migration_state = None - - -class CheckNameAvailabilityParameters(msrest.serialization.Model): - """Data Lake Store account name availability check parameters. - - 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 Azure. - - :param name: Required. The Data Lake Store name to check availability for. - :type name: str - :ivar type: Required. The resource type. Note: This should not be set by the user, as the - constant value is Microsoft.DataLakeStore/accounts. Default value: - "Microsoft.DataLakeStore/accounts". - :vartype type: str - """ - - _validation = { - 'name': {'required': True}, - 'type': {'required': True, 'constant': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - type = "Microsoft.DataLakeStore/accounts" - - def __init__( - self, - **kwargs - ): - super(CheckNameAvailabilityParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - - -class CreateDataLakeStoreAccountParameters(msrest.serialization.Model): - """CreateDataLakeStoreAccountParameters. - - All required parameters must be populated in order to send to Azure. - - :param location: Required. The resource location. - :type location: str - :param tags: A set of tags. The resource tags. - :type tags: dict[str, str] - :param identity: The Key Vault encryption identity, if any. - :type identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity - :param default_group: The default owner group for all new folders and files created in the Data - Lake Store account. - :type default_group: str - :param encryption_config: The Key Vault encryption configuration. - :type encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :param encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". - :type encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState - :param firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :type firewall_rules: - list[~azure.mgmt.datalake.store.models.CreateFirewallRuleWithAccountParameters] - :param virtual_network_rules: The list of virtual network rules associated with this Data Lake - Store account. - :type virtual_network_rules: - list[~azure.mgmt.datalake.store.models.CreateVirtualNetworkRuleWithAccountParameters] - :param firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". - :type firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :param firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :type firewall_allow_azure_ips: str or - ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :param trusted_id_providers: The list of trusted identity providers associated with this Data - Lake Store account. - :type trusted_id_providers: - list[~azure.mgmt.datalake.store.models.CreateTrustedIdProviderWithAccountParameters] - :param trusted_id_provider_state: The current state of the trusted identity provider feature - for this Data Lake Store account. Possible values include: "Enabled", "Disabled". - :type trusted_id_provider_state: str or - ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :param new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :type new_tier: str or ~azure.mgmt.datalake.store.models.TierType - """ - - _validation = { - 'location': {'required': True}, - } - - _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'identity': {'key': 'identity', 'type': 'EncryptionIdentity'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'properties.encryptionState', 'type': 'str'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[CreateFirewallRuleWithAccountParameters]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[CreateVirtualNetworkRuleWithAccountParameters]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[CreateTrustedIdProviderWithAccountParameters]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateDataLakeStoreAccountParameters, self).__init__(**kwargs) - self.location = kwargs['location'] - self.tags = kwargs.get('tags', None) - self.identity = kwargs.get('identity', None) - self.default_group = kwargs.get('default_group', None) - self.encryption_config = kwargs.get('encryption_config', None) - self.encryption_state = kwargs.get('encryption_state', None) - self.firewall_rules = kwargs.get('firewall_rules', None) - self.virtual_network_rules = kwargs.get('virtual_network_rules', None) - self.firewall_state = kwargs.get('firewall_state', None) - self.firewall_allow_azure_ips = kwargs.get('firewall_allow_azure_ips', None) - self.trusted_id_providers = kwargs.get('trusted_id_providers', None) - self.trusted_id_provider_state = kwargs.get('trusted_id_provider_state', None) - self.new_tier = kwargs.get('new_tier', None) - - -class CreateFirewallRuleWithAccountParameters(msrest.serialization.Model): - """The parameters used to create a new firewall rule while creating a new Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the firewall rule to create. - :type name: str - :param start_ip_address: Required. The start IP address for the firewall rule. This can be - either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: Required. The end IP address for the firewall rule. This can be either - ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str - """ - - _validation = { - 'name': {'required': True}, - 'start_ip_address': {'required': True}, - 'end_ip_address': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateFirewallRuleWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.start_ip_address = kwargs['start_ip_address'] - self.end_ip_address = kwargs['end_ip_address'] - - -class CreateOrUpdateFirewallRuleParameters(msrest.serialization.Model): - """The parameters used to create a new firewall rule. - - All required parameters must be populated in order to send to Azure. - - :param start_ip_address: Required. The start IP address for the firewall rule. This can be - either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: Required. The end IP address for the firewall rule. This can be either - ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str - """ - - _validation = { - 'start_ip_address': {'required': True}, - 'end_ip_address': {'required': True}, - } - - _attribute_map = { - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateOrUpdateFirewallRuleParameters, self).__init__(**kwargs) - self.start_ip_address = kwargs['start_ip_address'] - self.end_ip_address = kwargs['end_ip_address'] - - -class CreateOrUpdateTrustedIdProviderParameters(msrest.serialization.Model): - """The parameters used to create a new trusted identity provider. - - All required parameters must be populated in order to send to Azure. - - :param id_provider: Required. The URL of this trusted identity provider. - :type id_provider: str - """ - - _validation = { - 'id_provider': {'required': True}, - } - - _attribute_map = { - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateOrUpdateTrustedIdProviderParameters, self).__init__(**kwargs) - self.id_provider = kwargs['id_provider'] - - -class CreateOrUpdateVirtualNetworkRuleParameters(msrest.serialization.Model): - """The parameters used to create a new virtual network rule. - - All required parameters must be populated in order to send to Azure. - - :param subnet_id: Required. The resource identifier for the subnet. - :type subnet_id: str - """ - - _validation = { - 'subnet_id': {'required': True}, - } - - _attribute_map = { - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateOrUpdateVirtualNetworkRuleParameters, self).__init__(**kwargs) - self.subnet_id = kwargs['subnet_id'] - - -class CreateTrustedIdProviderWithAccountParameters(msrest.serialization.Model): - """The parameters used to create a new trusted identity provider while creating a new Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the trusted identity provider to create. - :type name: str - :param id_provider: Required. The URL of this trusted identity provider. - :type id_provider: str - """ - - _validation = { - 'name': {'required': True}, - 'id_provider': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateTrustedIdProviderWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.id_provider = kwargs['id_provider'] - - -class CreateVirtualNetworkRuleWithAccountParameters(msrest.serialization.Model): - """The parameters used to create a new virtual network rule while creating a new Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the virtual network rule to create. - :type name: str - :param subnet_id: Required. The resource identifier for the subnet. - :type subnet_id: str - """ - - _validation = { - 'name': {'required': True}, - 'subnet_id': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(CreateVirtualNetworkRuleWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.subnet_id = kwargs['subnet_id'] - - -class Resource(msrest.serialization.Model): - """The resource model definition. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar location: The resource location. - :vartype location: str - :ivar tags: A set of tags. The resource tags. - :vartype tags: dict[str, str] - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - } - - def __init__( - self, - **kwargs - ): - super(Resource, self).__init__(**kwargs) - self.id = None - self.name = None - self.type = None - self.location = None - self.tags = None - - -class DataLakeStoreAccount(Resource): - """Data Lake Store account information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar location: The resource location. - :vartype location: str - :ivar tags: A set of tags. The resource tags. - :vartype tags: dict[str, str] - :ivar identity: The Key Vault encryption identity, if any. - :vartype identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity - :ivar account_id: The unique identifier associated with this Data Lake Store account. - :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". - :vartype provisioning_state: str or - ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", - "Suspended". - :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState - :ivar creation_time: The account creation time. - :vartype creation_time: ~datetime.datetime - :ivar last_modified_time: The account last modified time. - :vartype last_modified_time: ~datetime.datetime - :ivar endpoint: The full CName endpoint for this account. - :vartype endpoint: str - :ivar default_group: The default owner group for all new folders and files created in the Data - Lake Store account. - :vartype default_group: str - :ivar encryption_config: The Key Vault encryption configuration. - :vartype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :ivar encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". - :vartype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState - :ivar encryption_provisioning_state: The current state of encryption provisioning for this Data - Lake Store account. Possible values include: "Creating", "Succeeded". - :vartype encryption_provisioning_state: str or - ~azure.mgmt.datalake.store.models.EncryptionProvisioningState - :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :vartype firewall_rules: list[~azure.mgmt.datalake.store.models.FirewallRule] - :ivar virtual_network_rules: The list of virtual network rules associated with this Data Lake - Store account. - :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.VirtualNetworkRule] - :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". - :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :vartype firewall_allow_azure_ips: str or - ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :ivar trusted_id_providers: The list of trusted identity providers associated with this Data - Lake Store account. - :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.TrustedIdProvider] - :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for - this Data Lake Store account. Possible values include: "Enabled", "Disabled". - :vartype trusted_id_provider_state: str or - ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :ivar new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType - :ivar current_tier: The commitment tier in use for the current month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :vartype current_tier: str or ~azure.mgmt.datalake.store.models.TierType - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, - 'identity': {'readonly': True}, - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - 'default_group': {'readonly': True}, - 'encryption_config': {'readonly': True}, - 'encryption_state': {'readonly': True}, - 'encryption_provisioning_state': {'readonly': True}, - 'firewall_rules': {'readonly': True}, - 'virtual_network_rules': {'readonly': True}, - 'firewall_state': {'readonly': True}, - 'firewall_allow_azure_ips': {'readonly': True}, - 'trusted_id_providers': {'readonly': True}, - 'trusted_id_provider_state': {'readonly': True}, - 'new_tier': {'readonly': True}, - 'current_tier': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'identity': {'key': 'identity', 'type': 'EncryptionIdentity'}, - 'account_id': {'key': 'properties.accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'creation_time': {'key': 'properties.creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'properties.lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'properties.endpoint', 'type': 'str'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'properties.encryptionState', 'type': 'str'}, - 'encryption_provisioning_state': {'key': 'properties.encryptionProvisioningState', 'type': 'str'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[FirewallRule]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[VirtualNetworkRule]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[TrustedIdProvider]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, - 'current_tier': {'key': 'properties.currentTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccount, self).__init__(**kwargs) - self.identity = None - self.account_id = None - self.provisioning_state = None - self.state = None - self.creation_time = None - self.last_modified_time = None - self.endpoint = None - self.default_group = None - self.encryption_config = None - self.encryption_state = None - self.encryption_provisioning_state = None - self.firewall_rules = None - self.virtual_network_rules = None - self.firewall_state = None - self.firewall_allow_azure_ips = None - self.trusted_id_providers = None - self.trusted_id_provider_state = None - self.new_tier = None - self.current_tier = None - - -class DataLakeStoreAccountBasic(Resource): - """Basic Data Lake Store account information, returned on list calls. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar location: The resource location. - :vartype location: str - :ivar tags: A set of tags. The resource tags. - :vartype tags: dict[str, str] - :ivar account_id: The unique identifier associated with this Data Lake Store account. - :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". - :vartype provisioning_state: str or - ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", - "Suspended". - :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState - :ivar creation_time: The account creation time. - :vartype creation_time: ~datetime.datetime - :ivar last_modified_time: The account last modified time. - :vartype last_modified_time: ~datetime.datetime - :ivar endpoint: The full CName endpoint for this account. - :vartype endpoint: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'account_id': {'key': 'properties.accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'creation_time': {'key': 'properties.creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'properties.lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'properties.endpoint', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountBasic, self).__init__(**kwargs) - self.account_id = None - self.provisioning_state = None - self.state = None - self.creation_time = None - self.last_modified_time = None - self.endpoint = None - - -class DataLakeStoreAccountListResult(msrest.serialization.Model): - """Data Lake Store account list information response. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: The results of the list operation. - :vartype value: list[~azure.mgmt.datalake.store.models.DataLakeStoreAccountBasic] - :ivar next_link: The link (url) to the next page of results. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[DataLakeStoreAccountBasic]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountListResult, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class DataLakeStoreAccountPropertiesBasic(msrest.serialization.Model): - """The basic account specific properties that are associated with an underlying Data Lake Store account. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar account_id: The unique identifier associated with this Data Lake Store account. - :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". - :vartype provisioning_state: str or - ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", - "Suspended". - :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState - :ivar creation_time: The account creation time. - :vartype creation_time: ~datetime.datetime - :ivar last_modified_time: The account last modified time. - :vartype last_modified_time: ~datetime.datetime - :ivar endpoint: The full CName endpoint for this account. - :vartype endpoint: str - """ - - _validation = { - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - } - - _attribute_map = { - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'provisioningState', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'creation_time': {'key': 'creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'endpoint', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountPropertiesBasic, self).__init__(**kwargs) - self.account_id = None - self.provisioning_state = None - self.state = None - self.creation_time = None - self.last_modified_time = None - self.endpoint = None - - -class DataLakeStoreAccountProperties(DataLakeStoreAccountPropertiesBasic): - """Data Lake Store account properties information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar account_id: The unique identifier associated with this Data Lake Store account. - :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". - :vartype provisioning_state: str or - ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", - "Suspended". - :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState - :ivar creation_time: The account creation time. - :vartype creation_time: ~datetime.datetime - :ivar last_modified_time: The account last modified time. - :vartype last_modified_time: ~datetime.datetime - :ivar endpoint: The full CName endpoint for this account. - :vartype endpoint: str - :ivar default_group: The default owner group for all new folders and files created in the Data - Lake Store account. - :vartype default_group: str - :ivar encryption_config: The Key Vault encryption configuration. - :vartype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :ivar encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". - :vartype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState - :ivar encryption_provisioning_state: The current state of encryption provisioning for this Data - Lake Store account. Possible values include: "Creating", "Succeeded". - :vartype encryption_provisioning_state: str or - ~azure.mgmt.datalake.store.models.EncryptionProvisioningState - :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :vartype firewall_rules: list[~azure.mgmt.datalake.store.models.FirewallRule] - :ivar virtual_network_rules: The list of virtual network rules associated with this Data Lake - Store account. - :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.VirtualNetworkRule] - :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". - :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :vartype firewall_allow_azure_ips: str or - ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :ivar trusted_id_providers: The list of trusted identity providers associated with this Data - Lake Store account. - :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.TrustedIdProvider] - :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for - this Data Lake Store account. Possible values include: "Enabled", "Disabled". - :vartype trusted_id_provider_state: str or - ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :ivar new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType - :ivar current_tier: The commitment tier in use for the current month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :vartype current_tier: str or ~azure.mgmt.datalake.store.models.TierType - """ - - _validation = { - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - 'default_group': {'readonly': True}, - 'encryption_config': {'readonly': True}, - 'encryption_state': {'readonly': True}, - 'encryption_provisioning_state': {'readonly': True}, - 'firewall_rules': {'readonly': True}, - 'virtual_network_rules': {'readonly': True}, - 'firewall_state': {'readonly': True}, - 'firewall_allow_azure_ips': {'readonly': True}, - 'trusted_id_providers': {'readonly': True}, - 'trusted_id_provider_state': {'readonly': True}, - 'new_tier': {'readonly': True}, - 'current_tier': {'readonly': True}, - } - - _attribute_map = { - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'provisioningState', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'creation_time': {'key': 'creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'endpoint', 'type': 'str'}, - 'default_group': {'key': 'defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'encryptionState', 'type': 'str'}, - 'encryption_provisioning_state': {'key': 'encryptionProvisioningState', 'type': 'str'}, - 'firewall_rules': {'key': 'firewallRules', 'type': '[FirewallRule]'}, - 'virtual_network_rules': {'key': 'virtualNetworkRules', 'type': '[VirtualNetworkRule]'}, - 'firewall_state': {'key': 'firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'trustedIdProviders', 'type': '[TrustedIdProvider]'}, - 'trusted_id_provider_state': {'key': 'trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'newTier', 'type': 'str'}, - 'current_tier': {'key': 'currentTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountProperties, self).__init__(**kwargs) - self.default_group = None - self.encryption_config = None - self.encryption_state = None - self.encryption_provisioning_state = None - self.firewall_rules = None - self.virtual_network_rules = None - self.firewall_state = None - self.firewall_allow_azure_ips = None - self.trusted_id_providers = None - self.trusted_id_provider_state = None - self.new_tier = None - self.current_tier = None - - -class EncryptionConfig(msrest.serialization.Model): - """The encryption configuration for the account. - - All required parameters must be populated in order to send to Azure. - - :param type: Required. The type of encryption configuration being used. Currently the only - supported types are 'UserManaged' and 'ServiceManaged'. Possible values include: "UserManaged", - "ServiceManaged". - :type type: str or ~azure.mgmt.datalake.store.models.EncryptionConfigType - :param key_vault_meta_info: The Key Vault information for connecting to user managed encryption - keys. - :type key_vault_meta_info: ~azure.mgmt.datalake.store.models.KeyVaultMetaInfo - """ - - _validation = { - 'type': {'required': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'key_vault_meta_info': {'key': 'keyVaultMetaInfo', 'type': 'KeyVaultMetaInfo'}, - } - - def __init__( - self, - **kwargs - ): - super(EncryptionConfig, self).__init__(**kwargs) - self.type = kwargs['type'] - self.key_vault_meta_info = kwargs.get('key_vault_meta_info', None) - - -class EncryptionIdentity(msrest.serialization.Model): - """The encryption identity properties. - - 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 Azure. - - :ivar type: Required. The type of encryption being used. Currently the only supported type is - 'SystemAssigned'. Default value: "SystemAssigned". - :vartype type: str - :ivar principal_id: The principal identifier associated with the encryption. - :vartype principal_id: str - :ivar tenant_id: The tenant identifier associated with the encryption. - :vartype tenant_id: str - """ - - _validation = { - 'type': {'required': True, 'constant': True}, - 'principal_id': {'readonly': True}, - 'tenant_id': {'readonly': True}, - } - - _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'principal_id': {'key': 'principalId', 'type': 'str'}, - 'tenant_id': {'key': 'tenantId', 'type': 'str'}, - } - - type = "SystemAssigned" - - def __init__( - self, - **kwargs - ): - super(EncryptionIdentity, self).__init__(**kwargs) - self.principal_id = None - self.tenant_id = None - - -class SubResource(msrest.serialization.Model): - """The resource model definition for a nested resource. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(SubResource, self).__init__(**kwargs) - self.id = None - self.name = None - self.type = None - - -class FirewallRule(SubResource): - """Data Lake Store firewall rule information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :vartype start_ip_address: str - :ivar end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :vartype end_ip_address: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'start_ip_address': {'readonly': True}, - 'end_ip_address': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(FirewallRule, self).__init__(**kwargs) - self.start_ip_address = None - self.end_ip_address = None - - -class FirewallRuleListResult(msrest.serialization.Model): - """Data Lake Store firewall rule list information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: The results of the list operation. - :vartype value: list[~azure.mgmt.datalake.store.models.FirewallRule] - :ivar next_link: The link (url) to the next page of results. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[FirewallRule]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(FirewallRuleListResult, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class KeyVaultMetaInfo(msrest.serialization.Model): - """Metadata information used by account encryption. - - All required parameters must be populated in order to send to Azure. - - :param key_vault_resource_id: Required. The resource identifier for the user managed Key Vault - being used to encrypt. - :type key_vault_resource_id: str - :param encryption_key_name: Required. The name of the user managed encryption key. - :type encryption_key_name: str - :param encryption_key_version: Required. The version of the user managed encryption key. - :type encryption_key_version: str - """ - - _validation = { - 'key_vault_resource_id': {'required': True}, - 'encryption_key_name': {'required': True}, - 'encryption_key_version': {'required': True}, - } - - _attribute_map = { - 'key_vault_resource_id': {'key': 'keyVaultResourceId', 'type': 'str'}, - 'encryption_key_name': {'key': 'encryptionKeyName', 'type': 'str'}, - 'encryption_key_version': {'key': 'encryptionKeyVersion', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(KeyVaultMetaInfo, self).__init__(**kwargs) - self.key_vault_resource_id = kwargs['key_vault_resource_id'] - self.encryption_key_name = kwargs['encryption_key_name'] - self.encryption_key_version = kwargs['encryption_key_version'] - - -class NameAvailabilityInformation(msrest.serialization.Model): - """Data Lake Store account name availability result information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name_available: The Boolean value of true or false to indicate whether the Data Lake - Store account name is available or not. - :vartype name_available: bool - :ivar reason: The reason why the Data Lake Store account name is not available, if - nameAvailable is false. - :vartype reason: str - :ivar message: The message describing why the Data Lake Store account name is not available, if - nameAvailable is false. - :vartype message: str - """ - - _validation = { - 'name_available': {'readonly': True}, - 'reason': {'readonly': True}, - 'message': {'readonly': True}, - } - - _attribute_map = { - 'name_available': {'key': 'nameAvailable', 'type': 'bool'}, - 'reason': {'key': 'reason', 'type': 'str'}, - 'message': {'key': 'message', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(NameAvailabilityInformation, self).__init__(**kwargs) - self.name_available = None - self.reason = None - self.message = None - - -class Operation(msrest.serialization.Model): - """An available operation for Data Lake Store. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar name: The name of the operation. - :vartype name: str - :param display: The display information for the operation. - :type display: ~azure.mgmt.datalake.store.models.OperationDisplay - :ivar origin: The intended executor of the operation. Possible values include: "user", - "system", "user,system". - :vartype origin: str or ~azure.mgmt.datalake.store.models.OperationOrigin - """ - - _validation = { - 'name': {'readonly': True}, - 'origin': {'readonly': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'display': {'key': 'display', 'type': 'OperationDisplay'}, - 'origin': {'key': 'origin', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(Operation, self).__init__(**kwargs) - self.name = None - self.display = kwargs.get('display', None) - self.origin = None - - -class OperationDisplay(msrest.serialization.Model): - """The display information for a particular operation. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar provider: The resource provider of the operation. - :vartype provider: str - :ivar resource: The resource type of the operation. - :vartype resource: str - :ivar operation: A friendly name of the operation. - :vartype operation: str - :ivar description: A friendly description of the operation. - :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 - ): - super(OperationDisplay, self).__init__(**kwargs) - self.provider = None - self.resource = None - self.operation = None - self.description = None - - -class OperationListResult(msrest.serialization.Model): - """The list of available operations for Data Lake Store. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: The results of the list operation. - :vartype value: list[~azure.mgmt.datalake.store.models.Operation] - :ivar next_link: The link (url) to the next page of results. - :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 - ): - super(OperationListResult, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class TrustedIdProvider(SubResource): - """Data Lake Store trusted identity provider information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar id_provider: The URL of this trusted identity provider. - :vartype id_provider: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'id_provider': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TrustedIdProvider, self).__init__(**kwargs) - self.id_provider = None - - -class TrustedIdProviderListResult(msrest.serialization.Model): - """Data Lake Store trusted identity provider list information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: The results of the list operation. - :vartype value: list[~azure.mgmt.datalake.store.models.TrustedIdProvider] - :ivar next_link: The link (url) to the next page of results. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[TrustedIdProvider]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(TrustedIdProviderListResult, self).__init__(**kwargs) - self.value = None - self.next_link = None - - -class UpdateDataLakeStoreAccountParameters(msrest.serialization.Model): - """Data Lake Store account information to update. - - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :param default_group: The default owner group for all new folders and files created in the Data - Lake Store account. - :type default_group: str - :param encryption_config: Used for rotation of user managed Key Vault keys. Can only be used to - rotate a user managed encryption Key Vault key. - :type encryption_config: ~azure.mgmt.datalake.store.models.UpdateEncryptionConfig - :param firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :type firewall_rules: - list[~azure.mgmt.datalake.store.models.UpdateFirewallRuleWithAccountParameters] - :param virtual_network_rules: The list of virtual network rules associated with this Data Lake - Store account. - :type virtual_network_rules: - list[~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleWithAccountParameters] - :param firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Disabling the firewall does not remove existing rules, they will just be ignored until - the firewall is re-enabled. Possible values include: "Enabled", "Disabled". - :type firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :param firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :type firewall_allow_azure_ips: str or - ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :param trusted_id_providers: The list of trusted identity providers associated with this Data - Lake Store account. - :type trusted_id_providers: - list[~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderWithAccountParameters] - :param trusted_id_provider_state: The current state of the trusted identity provider feature - for this Data Lake Store account. Disabling trusted identity provider functionality does not - remove the providers, they will just be ignored until this feature is re-enabled. Possible - values include: "Enabled", "Disabled". - :type trusted_id_provider_state: str or - ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :param new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :type new_tier: str or ~azure.mgmt.datalake.store.models.TierType - """ - - _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'UpdateEncryptionConfig'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[UpdateFirewallRuleWithAccountParameters]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[UpdateVirtualNetworkRuleWithAccountParameters]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[UpdateTrustedIdProviderWithAccountParameters]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateDataLakeStoreAccountParameters, self).__init__(**kwargs) - self.tags = kwargs.get('tags', None) - self.default_group = kwargs.get('default_group', None) - self.encryption_config = kwargs.get('encryption_config', None) - self.firewall_rules = kwargs.get('firewall_rules', None) - self.virtual_network_rules = kwargs.get('virtual_network_rules', None) - self.firewall_state = kwargs.get('firewall_state', None) - self.firewall_allow_azure_ips = kwargs.get('firewall_allow_azure_ips', None) - self.trusted_id_providers = kwargs.get('trusted_id_providers', None) - self.trusted_id_provider_state = kwargs.get('trusted_id_provider_state', None) - self.new_tier = kwargs.get('new_tier', None) - - -class UpdateEncryptionConfig(msrest.serialization.Model): - """The encryption configuration used to update a user managed Key Vault key. - - :param key_vault_meta_info: The updated Key Vault key to use in user managed key rotation. - :type key_vault_meta_info: ~azure.mgmt.datalake.store.models.UpdateKeyVaultMetaInfo - """ - - _attribute_map = { - 'key_vault_meta_info': {'key': 'keyVaultMetaInfo', 'type': 'UpdateKeyVaultMetaInfo'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateEncryptionConfig, self).__init__(**kwargs) - self.key_vault_meta_info = kwargs.get('key_vault_meta_info', None) - - -class UpdateFirewallRuleParameters(msrest.serialization.Model): - """The parameters used to update a firewall rule. - - :param start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :type end_ip_address: str - """ - - _attribute_map = { - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateFirewallRuleParameters, self).__init__(**kwargs) - self.start_ip_address = kwargs.get('start_ip_address', None) - self.end_ip_address = kwargs.get('end_ip_address', None) - - -class UpdateFirewallRuleWithAccountParameters(msrest.serialization.Model): - """The parameters used to update a firewall rule while updating a Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the firewall rule to update. - :type name: str - :param start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or - ipv6. Start and End should be in the same protocol. - :type end_ip_address: str - """ - - _validation = { - 'name': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateFirewallRuleWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.start_ip_address = kwargs.get('start_ip_address', None) - self.end_ip_address = kwargs.get('end_ip_address', None) - - -class UpdateKeyVaultMetaInfo(msrest.serialization.Model): - """The Key Vault update information used for user managed key rotation. - - :param encryption_key_version: The version of the user managed encryption key to update through - a key rotation. - :type encryption_key_version: str - """ - - _attribute_map = { - 'encryption_key_version': {'key': 'encryptionKeyVersion', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateKeyVaultMetaInfo, self).__init__(**kwargs) - self.encryption_key_version = kwargs.get('encryption_key_version', None) - - -class UpdateTrustedIdProviderParameters(msrest.serialization.Model): - """The parameters used to update a trusted identity provider. - - :param id_provider: The URL of this trusted identity provider. - :type id_provider: str - """ - - _attribute_map = { - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateTrustedIdProviderParameters, self).__init__(**kwargs) - self.id_provider = kwargs.get('id_provider', None) - - -class UpdateTrustedIdProviderWithAccountParameters(msrest.serialization.Model): - """The parameters used to update a trusted identity provider while updating a Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the trusted identity provider to update. - :type name: str - :param id_provider: The URL of this trusted identity provider. - :type id_provider: str - """ - - _validation = { - 'name': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateTrustedIdProviderWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.id_provider = kwargs.get('id_provider', None) - - -class UpdateVirtualNetworkRuleParameters(msrest.serialization.Model): - """The parameters used to update a virtual network rule. - - :param subnet_id: The resource identifier for the subnet. - :type subnet_id: str - """ - - _attribute_map = { - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateVirtualNetworkRuleParameters, self).__init__(**kwargs) - self.subnet_id = kwargs.get('subnet_id', None) - - -class UpdateVirtualNetworkRuleWithAccountParameters(msrest.serialization.Model): - """The parameters used to update a virtual network rule while updating a Data Lake Store account. - - All required parameters must be populated in order to send to Azure. - - :param name: Required. The unique name of the virtual network rule to update. - :type name: str - :param subnet_id: The resource identifier for the subnet. - :type subnet_id: str - """ - - _validation = { - 'name': {'required': True}, - } - - _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UpdateVirtualNetworkRuleWithAccountParameters, self).__init__(**kwargs) - self.name = kwargs['name'] - self.subnet_id = kwargs.get('subnet_id', None) - - -class Usage(msrest.serialization.Model): - """Describes the Resource Usage. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar unit: Gets the unit of measurement. Possible values include: "Count", "Bytes", "Seconds", - "Percent", "CountsPerSecond", "BytesPerSecond". - :vartype unit: str or ~azure.mgmt.datalake.store.models.UsageUnit - :ivar id: Resource identifier. - :vartype id: str - :ivar current_value: Gets the current count of the allocated resources in the subscription. - :vartype current_value: int - :ivar limit: Gets the maximum count of the resources that can be allocated in the subscription. - :vartype limit: int - :ivar name: Gets the name of the type of usage. - :vartype name: ~azure.mgmt.datalake.store.models.UsageName - """ - - _validation = { - 'unit': {'readonly': True}, - 'id': {'readonly': True}, - 'current_value': {'readonly': True}, - 'limit': {'readonly': True}, - 'name': {'readonly': True}, - } - - _attribute_map = { - 'unit': {'key': 'unit', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'current_value': {'key': 'currentValue', 'type': 'int'}, - 'limit': {'key': 'limit', 'type': 'int'}, - 'name': {'key': 'name', 'type': 'UsageName'}, - } - - def __init__( - self, - **kwargs - ): - super(Usage, self).__init__(**kwargs) - self.unit = None - self.id = None - self.current_value = None - self.limit = None - self.name = None - - -class UsageListResult(msrest.serialization.Model): - """The response from the List Usages operation. - - :param value: Gets or sets the list of Storage Resource Usages. - :type value: list[~azure.mgmt.datalake.store.models.Usage] - """ - - _attribute_map = { - 'value': {'key': 'value', 'type': '[Usage]'}, - } - - def __init__( - self, - **kwargs - ): - super(UsageListResult, self).__init__(**kwargs) - self.value = kwargs.get('value', None) - - -class UsageName(msrest.serialization.Model): - """The usage names that can be used. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: Gets a string describing the resource name. - :vartype value: str - :ivar localized_value: Gets a localized string describing the resource name. - :vartype localized_value: str - """ - - _validation = { - 'value': {'readonly': True}, - 'localized_value': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': 'str'}, - 'localized_value': {'key': 'localizedValue', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(UsageName, self).__init__(**kwargs) - self.value = None - self.localized_value = None - - -class VirtualNetworkRule(SubResource): - """Data Lake Store virtual network rule information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar id: The resource identifier. - :vartype id: str - :ivar name: The resource name. - :vartype name: str - :ivar type: The resource type. - :vartype type: str - :ivar subnet_id: The resource identifier for the subnet. - :vartype subnet_id: str - """ - - _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'subnet_id': {'readonly': True}, - } - - _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(VirtualNetworkRule, self).__init__(**kwargs) - self.subnet_id = None - - -class VirtualNetworkRuleListResult(msrest.serialization.Model): - """Data Lake Store virtual network rule list information. - - Variables are only populated by the server, and will be ignored when sending a request. - - :ivar value: The results of the list operation. - :vartype value: list[~azure.mgmt.datalake.store.models.VirtualNetworkRule] - :ivar next_link: The link (url) to the next page of results. - :vartype next_link: str - """ - - _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, - } - - _attribute_map = { - 'value': {'key': 'value', 'type': '[VirtualNetworkRule]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(VirtualNetworkRuleListResult, self).__init__(**kwargs) - self.value = None - self.next_link = None diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models_py3.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models_py3.py index 9b9fc06e4199..5830a6dae565 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models_py3.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_models_py3.py @@ -1,4 +1,5 @@ # coding=utf-8 +# pylint: disable=too-many-lines # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for license information. @@ -6,22 +7,24 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union -import msrest.serialization +from .. import _serialization -from ._data_lake_store_account_management_client_enums import * +if TYPE_CHECKING: + # pylint: disable=unused-import,ungrouped-imports + from .. import models as _models -class CapabilityInformation(msrest.serialization.Model): +class CapabilityInformation(_serialization.Model): """Subscription-level properties and limits for Data Lake Store. Variables are only populated by the server, and will be ignored when sending a request. :ivar subscription_id: The subscription credentials that uniquely identifies the subscription. :vartype subscription_id: str - :ivar state: The subscription state. Possible values include: "Registered", "Suspended", - "Deleted", "Unregistered", "Warned". + :ivar state: The subscription state. Known values are: "Registered", "Suspended", "Deleted", + "Unregistered", and "Warned". :vartype state: str or ~azure.mgmt.datalake.store.models.SubscriptionState :ivar max_account_count: The maximum supported number of accounts under this subscription. :vartype max_account_count: int @@ -32,26 +35,24 @@ class CapabilityInformation(msrest.serialization.Model): """ _validation = { - 'subscription_id': {'readonly': True}, - 'state': {'readonly': True}, - 'max_account_count': {'readonly': True}, - 'account_count': {'readonly': True}, - 'migration_state': {'readonly': True}, + "subscription_id": {"readonly": True}, + "state": {"readonly": True}, + "max_account_count": {"readonly": True}, + "account_count": {"readonly": True}, + "migration_state": {"readonly": True}, } _attribute_map = { - 'subscription_id': {'key': 'subscriptionId', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'max_account_count': {'key': 'maxAccountCount', 'type': 'int'}, - 'account_count': {'key': 'accountCount', 'type': 'int'}, - 'migration_state': {'key': 'migrationState', 'type': 'bool'}, + "subscription_id": {"key": "subscriptionId", "type": "str"}, + "state": {"key": "state", "type": "str"}, + "max_account_count": {"key": "maxAccountCount", "type": "int"}, + "account_count": {"key": "accountCount", "type": "int"}, + "migration_state": {"key": "migrationState", "type": "bool"}, } - def __init__( - self, - **kwargs - ): - super(CapabilityInformation, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.subscription_id = None self.state = None self.max_account_count = None @@ -59,109 +60,115 @@ def __init__( self.migration_state = None -class CheckNameAvailabilityParameters(msrest.serialization.Model): +class CheckNameAvailabilityParameters(_serialization.Model): """Data Lake Store account name availability check parameters. - 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 Azure. - :param name: Required. The Data Lake Store name to check availability for. - :type name: str - :ivar type: Required. The resource type. Note: This should not be set by the user, as the - constant value is Microsoft.DataLakeStore/accounts. Default value: - "Microsoft.DataLakeStore/accounts". - :vartype type: str + :ivar name: The Data Lake Store name to check availability for. Required. + :vartype name: str + :ivar type: The resource type. Note: This should not be set by the user, as the constant value + is Microsoft.DataLakeStore/accounts. Required. "Microsoft.DataLakeStore/accounts" + :vartype type: str or ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParametersType """ _validation = { - 'name': {'required': True}, - 'type': {'required': True, 'constant': True}, + "name": {"required": True}, + "type": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, } - type = "Microsoft.DataLakeStore/accounts" - def __init__( - self, - *, - name: str, - **kwargs - ): - super(CheckNameAvailabilityParameters, self).__init__(**kwargs) + self, *, name: str, type: Union[str, "_models.CheckNameAvailabilityParametersType"], **kwargs: Any + ) -> None: + """ + :keyword name: The Data Lake Store name to check availability for. Required. + :paramtype name: str + :keyword type: The resource type. Note: This should not be set by the user, as the constant + value is Microsoft.DataLakeStore/accounts. Required. "Microsoft.DataLakeStore/accounts" + :paramtype type: str or ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParametersType + """ + super().__init__(**kwargs) self.name = name + self.type = type -class CreateDataLakeStoreAccountParameters(msrest.serialization.Model): +class CreateDataLakeStoreAccountParameters(_serialization.Model): # pylint: disable=too-many-instance-attributes """CreateDataLakeStoreAccountParameters. All required parameters must be populated in order to send to Azure. - :param location: Required. The resource location. - :type location: str - :param tags: A set of tags. The resource tags. - :type tags: dict[str, str] - :param identity: The Key Vault encryption identity, if any. - :type identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity - :param default_group: The default owner group for all new folders and files created in the Data + :ivar location: The resource location. Required. + :vartype location: str + :ivar tags: The resource tags. + :vartype tags: dict[str, str] + :ivar identity: The Key Vault encryption identity, if any. + :vartype identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity + :ivar default_group: The default owner group for all new folders and files created in the Data Lake Store account. - :type default_group: str - :param encryption_config: The Key Vault encryption configuration. - :type encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :param encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". - :type encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState - :param firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :type firewall_rules: + :vartype default_group: str + :ivar encryption_config: The Key Vault encryption configuration. + :vartype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig + :ivar encryption_state: The current state of encryption for this Data Lake Store account. Known + values are: "Enabled" and "Disabled". + :vartype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState + :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. + :vartype firewall_rules: list[~azure.mgmt.datalake.store.models.CreateFirewallRuleWithAccountParameters] - :param virtual_network_rules: The list of virtual network rules associated with this Data Lake + :ivar virtual_network_rules: The list of virtual network rules associated with this Data Lake Store account. - :type virtual_network_rules: + :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.CreateVirtualNetworkRuleWithAccountParameters] - :param firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". - :type firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :param firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :type firewall_allow_azure_ips: str or + :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store + account. Known values are: "Enabled" and "Disabled". + :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState + :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". + :vartype firewall_allow_azure_ips: str or ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :param trusted_id_providers: The list of trusted identity providers associated with this Data + :ivar trusted_id_providers: The list of trusted identity providers associated with this Data Lake Store account. - :type trusted_id_providers: + :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.CreateTrustedIdProviderWithAccountParameters] - :param trusted_id_provider_state: The current state of the trusted identity provider feature - for this Data Lake Store account. Possible values include: "Enabled", "Disabled". - :type trusted_id_provider_state: str or + :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for + this Data Lake Store account. Known values are: "Enabled" and "Disabled". + :vartype trusted_id_provider_state: str or ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :param new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :type new_tier: str or ~azure.mgmt.datalake.store.models.TierType + :ivar new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". + :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType """ _validation = { - 'location': {'required': True}, + "location": {"required": True}, } _attribute_map = { - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'identity': {'key': 'identity', 'type': 'EncryptionIdentity'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'properties.encryptionState', 'type': 'str'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[CreateFirewallRuleWithAccountParameters]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[CreateVirtualNetworkRuleWithAccountParameters]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[CreateTrustedIdProviderWithAccountParameters]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, + "location": {"key": "location", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "identity": {"key": "identity", "type": "EncryptionIdentity"}, + "default_group": {"key": "properties.defaultGroup", "type": "str"}, + "encryption_config": {"key": "properties.encryptionConfig", "type": "EncryptionConfig"}, + "encryption_state": {"key": "properties.encryptionState", "type": "str"}, + "firewall_rules": {"key": "properties.firewallRules", "type": "[CreateFirewallRuleWithAccountParameters]"}, + "virtual_network_rules": { + "key": "properties.virtualNetworkRules", + "type": "[CreateVirtualNetworkRuleWithAccountParameters]", + }, + "firewall_state": {"key": "properties.firewallState", "type": "str"}, + "firewall_allow_azure_ips": {"key": "properties.firewallAllowAzureIps", "type": "str"}, + "trusted_id_providers": { + "key": "properties.trustedIdProviders", + "type": "[CreateTrustedIdProviderWithAccountParameters]", + }, + "trusted_id_provider_state": {"key": "properties.trustedIdProviderState", "type": "str"}, + "new_tier": {"key": "properties.newTier", "type": "str"}, } def __init__( @@ -169,20 +176,64 @@ def __init__( *, location: str, tags: Optional[Dict[str, str]] = None, - identity: Optional["EncryptionIdentity"] = None, + identity: Optional["_models.EncryptionIdentity"] = None, default_group: Optional[str] = None, - encryption_config: Optional["EncryptionConfig"] = None, - encryption_state: Optional[Union[str, "EncryptionState"]] = None, - firewall_rules: Optional[List["CreateFirewallRuleWithAccountParameters"]] = None, - virtual_network_rules: Optional[List["CreateVirtualNetworkRuleWithAccountParameters"]] = None, - firewall_state: Optional[Union[str, "FirewallState"]] = None, - firewall_allow_azure_ips: Optional[Union[str, "FirewallAllowAzureIpsState"]] = None, - trusted_id_providers: Optional[List["CreateTrustedIdProviderWithAccountParameters"]] = None, - trusted_id_provider_state: Optional[Union[str, "TrustedIdProviderState"]] = None, - new_tier: Optional[Union[str, "TierType"]] = None, - **kwargs - ): - super(CreateDataLakeStoreAccountParameters, self).__init__(**kwargs) + encryption_config: Optional["_models.EncryptionConfig"] = None, + encryption_state: Optional[Union[str, "_models.EncryptionState"]] = None, + firewall_rules: Optional[List["_models.CreateFirewallRuleWithAccountParameters"]] = None, + virtual_network_rules: Optional[List["_models.CreateVirtualNetworkRuleWithAccountParameters"]] = None, + firewall_state: Optional[Union[str, "_models.FirewallState"]] = None, + firewall_allow_azure_ips: Optional[Union[str, "_models.FirewallAllowAzureIpsState"]] = None, + trusted_id_providers: Optional[List["_models.CreateTrustedIdProviderWithAccountParameters"]] = None, + trusted_id_provider_state: Optional[Union[str, "_models.TrustedIdProviderState"]] = None, + new_tier: Optional[Union[str, "_models.TierType"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword location: The resource location. Required. + :paramtype location: str + :keyword tags: The resource tags. + :paramtype tags: dict[str, str] + :keyword identity: The Key Vault encryption identity, if any. + :paramtype identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity + :keyword default_group: The default owner group for all new folders and files created in the + Data Lake Store account. + :paramtype default_group: str + :keyword encryption_config: The Key Vault encryption configuration. + :paramtype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig + :keyword encryption_state: The current state of encryption for this Data Lake Store account. + Known values are: "Enabled" and "Disabled". + :paramtype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState + :keyword firewall_rules: The list of firewall rules associated with this Data Lake Store + account. + :paramtype firewall_rules: + list[~azure.mgmt.datalake.store.models.CreateFirewallRuleWithAccountParameters] + :keyword virtual_network_rules: The list of virtual network rules associated with this Data + Lake Store account. + :paramtype virtual_network_rules: + list[~azure.mgmt.datalake.store.models.CreateVirtualNetworkRuleWithAccountParameters] + :keyword firewall_state: The current state of the IP address firewall for this Data Lake Store + account. Known values are: "Enabled" and "Disabled". + :paramtype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState + :keyword firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". + :paramtype firewall_allow_azure_ips: str or + ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState + :keyword trusted_id_providers: The list of trusted identity providers associated with this Data + Lake Store account. + :paramtype trusted_id_providers: + list[~azure.mgmt.datalake.store.models.CreateTrustedIdProviderWithAccountParameters] + :keyword trusted_id_provider_state: The current state of the trusted identity provider feature + for this Data Lake Store account. Known values are: "Enabled" and "Disabled". + :paramtype trusted_id_provider_state: str or + ~azure.mgmt.datalake.store.models.TrustedIdProviderState + :keyword new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". + :paramtype new_tier: str or ~azure.mgmt.datalake.store.models.TierType + """ + super().__init__(**kwargs) self.location = location self.tags = tags self.identity = identity @@ -198,203 +249,208 @@ def __init__( self.new_tier = new_tier -class CreateFirewallRuleWithAccountParameters(msrest.serialization.Model): +class CreateFirewallRuleWithAccountParameters(_serialization.Model): """The parameters used to create a new firewall rule while creating a new Data Lake Store account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the firewall rule to create. - :type name: str - :param start_ip_address: Required. The start IP address for the firewall rule. This can be - either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: Required. The end IP address for the firewall rule. This can be either - ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str + :ivar name: The unique name of the firewall rule to create. Required. + :vartype name: str + :ivar start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :vartype start_ip_address: str + :ivar end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :vartype end_ip_address: str """ _validation = { - 'name': {'required': True}, - 'start_ip_address': {'required': True}, - 'end_ip_address': {'required': True}, + "name": {"required": True}, + "start_ip_address": {"required": True}, + "end_ip_address": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - *, - name: str, - start_ip_address: str, - end_ip_address: str, - **kwargs - ): - super(CreateFirewallRuleWithAccountParameters, self).__init__(**kwargs) + "name": {"key": "name", "type": "str"}, + "start_ip_address": {"key": "properties.startIpAddress", "type": "str"}, + "end_ip_address": {"key": "properties.endIpAddress", "type": "str"}, + } + + def __init__(self, *, name: str, start_ip_address: str, end_ip_address: str, **kwargs: Any) -> None: + """ + :keyword name: The unique name of the firewall rule to create. Required. + :paramtype name: str + :keyword start_ip_address: The start IP address for the firewall rule. This can be either ipv4 + or ipv6. Start and End should be in the same protocol. Required. + :paramtype start_ip_address: str + :keyword end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :paramtype end_ip_address: str + """ + super().__init__(**kwargs) self.name = name self.start_ip_address = start_ip_address self.end_ip_address = end_ip_address -class CreateOrUpdateFirewallRuleParameters(msrest.serialization.Model): +class CreateOrUpdateFirewallRuleParameters(_serialization.Model): """The parameters used to create a new firewall rule. All required parameters must be populated in order to send to Azure. - :param start_ip_address: Required. The start IP address for the firewall rule. This can be - either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: Required. The end IP address for the firewall rule. This can be either - ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str + :ivar start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :vartype start_ip_address: str + :ivar end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :vartype end_ip_address: str """ _validation = { - 'start_ip_address': {'required': True}, - 'end_ip_address': {'required': True}, + "start_ip_address": {"required": True}, + "end_ip_address": {"required": True}, } _attribute_map = { - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, - } - - def __init__( - self, - *, - start_ip_address: str, - end_ip_address: str, - **kwargs - ): - super(CreateOrUpdateFirewallRuleParameters, self).__init__(**kwargs) + "start_ip_address": {"key": "properties.startIpAddress", "type": "str"}, + "end_ip_address": {"key": "properties.endIpAddress", "type": "str"}, + } + + def __init__(self, *, start_ip_address: str, end_ip_address: str, **kwargs: Any) -> None: + """ + :keyword start_ip_address: The start IP address for the firewall rule. This can be either ipv4 + or ipv6. Start and End should be in the same protocol. Required. + :paramtype start_ip_address: str + :keyword end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. Required. + :paramtype end_ip_address: str + """ + super().__init__(**kwargs) self.start_ip_address = start_ip_address self.end_ip_address = end_ip_address -class CreateOrUpdateTrustedIdProviderParameters(msrest.serialization.Model): +class CreateOrUpdateTrustedIdProviderParameters(_serialization.Model): """The parameters used to create a new trusted identity provider. All required parameters must be populated in order to send to Azure. - :param id_provider: Required. The URL of this trusted identity provider. - :type id_provider: str + :ivar id_provider: The URL of this trusted identity provider. Required. + :vartype id_provider: str """ _validation = { - 'id_provider': {'required': True}, + "id_provider": {"required": True}, } _attribute_map = { - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, + "id_provider": {"key": "properties.idProvider", "type": "str"}, } - def __init__( - self, - *, - id_provider: str, - **kwargs - ): - super(CreateOrUpdateTrustedIdProviderParameters, self).__init__(**kwargs) + def __init__(self, *, id_provider: str, **kwargs: Any) -> None: + """ + :keyword id_provider: The URL of this trusted identity provider. Required. + :paramtype id_provider: str + """ + super().__init__(**kwargs) self.id_provider = id_provider -class CreateOrUpdateVirtualNetworkRuleParameters(msrest.serialization.Model): +class CreateOrUpdateVirtualNetworkRuleParameters(_serialization.Model): """The parameters used to create a new virtual network rule. All required parameters must be populated in order to send to Azure. - :param subnet_id: Required. The resource identifier for the subnet. - :type subnet_id: str + :ivar subnet_id: The resource identifier for the subnet. Required. + :vartype subnet_id: str """ _validation = { - 'subnet_id': {'required': True}, + "subnet_id": {"required": True}, } _attribute_map = { - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, + "subnet_id": {"key": "properties.subnetId", "type": "str"}, } - def __init__( - self, - *, - subnet_id: str, - **kwargs - ): - super(CreateOrUpdateVirtualNetworkRuleParameters, self).__init__(**kwargs) + def __init__(self, *, subnet_id: str, **kwargs: Any) -> None: + """ + :keyword subnet_id: The resource identifier for the subnet. Required. + :paramtype subnet_id: str + """ + super().__init__(**kwargs) self.subnet_id = subnet_id -class CreateTrustedIdProviderWithAccountParameters(msrest.serialization.Model): - """The parameters used to create a new trusted identity provider while creating a new Data Lake Store account. +class CreateTrustedIdProviderWithAccountParameters(_serialization.Model): + """The parameters used to create a new trusted identity provider while creating a new Data Lake + Store account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the trusted identity provider to create. - :type name: str - :param id_provider: Required. The URL of this trusted identity provider. - :type id_provider: str + :ivar name: The unique name of the trusted identity provider to create. Required. + :vartype name: str + :ivar id_provider: The URL of this trusted identity provider. Required. + :vartype id_provider: str """ _validation = { - 'name': {'required': True}, - 'id_provider': {'required': True}, + "name": {"required": True}, + "id_provider": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - *, - name: str, - id_provider: str, - **kwargs - ): - super(CreateTrustedIdProviderWithAccountParameters, self).__init__(**kwargs) + "name": {"key": "name", "type": "str"}, + "id_provider": {"key": "properties.idProvider", "type": "str"}, + } + + def __init__(self, *, name: str, id_provider: str, **kwargs: Any) -> None: + """ + :keyword name: The unique name of the trusted identity provider to create. Required. + :paramtype name: str + :keyword id_provider: The URL of this trusted identity provider. Required. + :paramtype id_provider: str + """ + super().__init__(**kwargs) self.name = name self.id_provider = id_provider -class CreateVirtualNetworkRuleWithAccountParameters(msrest.serialization.Model): - """The parameters used to create a new virtual network rule while creating a new Data Lake Store account. +class CreateVirtualNetworkRuleWithAccountParameters(_serialization.Model): + """The parameters used to create a new virtual network rule while creating a new Data Lake Store + account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the virtual network rule to create. - :type name: str - :param subnet_id: Required. The resource identifier for the subnet. - :type subnet_id: str + :ivar name: The unique name of the virtual network rule to create. Required. + :vartype name: str + :ivar subnet_id: The resource identifier for the subnet. Required. + :vartype subnet_id: str """ _validation = { - 'name': {'required': True}, - 'subnet_id': {'required': True}, + "name": {"required": True}, + "subnet_id": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - *, - name: str, - subnet_id: str, - **kwargs - ): - super(CreateVirtualNetworkRuleWithAccountParameters, self).__init__(**kwargs) + "name": {"key": "name", "type": "str"}, + "subnet_id": {"key": "properties.subnetId", "type": "str"}, + } + + def __init__(self, *, name: str, subnet_id: str, **kwargs: Any) -> None: + """ + :keyword name: The unique name of the virtual network rule to create. Required. + :paramtype name: str + :keyword subnet_id: The resource identifier for the subnet. Required. + :paramtype subnet_id: str + """ + super().__init__(**kwargs) self.name = name self.subnet_id = subnet_id -class Resource(msrest.serialization.Model): +class Resource(_serialization.Model): """The resource model definition. Variables are only populated by the server, and will be ignored when sending a request. @@ -407,31 +463,29 @@ class Resource(msrest.serialization.Model): :vartype type: str :ivar location: The resource location. :vartype location: str - :ivar tags: A set of tags. The resource tags. + :ivar tags: The resource tags. :vartype tags: dict[str, str] """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "location": {"readonly": True}, + "tags": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "location": {"key": "location", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, } - def __init__( - self, - **kwargs - ): - super(Resource, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.id = None self.name = None self.type = None @@ -439,7 +493,7 @@ def __init__( self.tags = None -class DataLakeStoreAccount(Resource): +class DataLakeStoreAccount(Resource): # pylint: disable=too-many-instance-attributes """Data Lake Store account information. Variables are only populated by the server, and will be ignored when sending a request. @@ -452,18 +506,18 @@ class DataLakeStoreAccount(Resource): :vartype type: str :ivar location: The resource location. :vartype location: str - :ivar tags: A set of tags. The resource tags. + :ivar tags: The resource tags. :vartype tags: dict[str, str] :ivar identity: The Key Vault encryption identity, if any. :vartype identity: ~azure.mgmt.datalake.store.models.EncryptionIdentity :ivar account_id: The unique identifier associated with this Data Lake Store account. :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". + :ivar provisioning_state: The provisioning status of the Data Lake Store account. Known values + are: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", "Resuming", + "Deleting", "Deleted", "Undeleting", and "Canceled". :vartype provisioning_state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", + :ivar state: The state of the Data Lake Store account. Known values are: "Active" and "Suspended". :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState :ivar creation_time: The account creation time. @@ -477,11 +531,11 @@ class DataLakeStoreAccount(Resource): :vartype default_group: str :ivar encryption_config: The Key Vault encryption configuration. :vartype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :ivar encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". + :ivar encryption_state: The current state of encryption for this Data Lake Store account. Known + values are: "Enabled" and "Disabled". :vartype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState :ivar encryption_provisioning_state: The current state of encryption provisioning for this Data - Lake Store account. Possible values include: "Creating", "Succeeded". + Lake Store account. Known values are: "Creating" and "Succeeded". :vartype encryption_provisioning_state: str or ~azure.mgmt.datalake.store.models.EncryptionProvisioningState :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. @@ -490,89 +544,87 @@ class DataLakeStoreAccount(Resource): Store account. :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.VirtualNetworkRule] :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". + account. Known values are: "Enabled" and "Disabled". :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". :vartype firewall_allow_azure_ips: str or ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState :ivar trusted_id_providers: The list of trusted identity providers associated with this Data Lake Store account. :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.TrustedIdProvider] :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for - this Data Lake Store account. Possible values include: "Enabled", "Disabled". + this Data Lake Store account. Known values are: "Enabled" and "Disabled". :vartype trusted_id_provider_state: str or ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :ivar new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". + :ivar new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType - :ivar current_tier: The commitment tier in use for the current month. Possible values include: + :ivar current_tier: The commitment tier in use for the current month. Known values are: "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". + "Commitment_1PB", and "Commitment_5PB". :vartype current_tier: str or ~azure.mgmt.datalake.store.models.TierType """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, - 'identity': {'readonly': True}, - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - 'default_group': {'readonly': True}, - 'encryption_config': {'readonly': True}, - 'encryption_state': {'readonly': True}, - 'encryption_provisioning_state': {'readonly': True}, - 'firewall_rules': {'readonly': True}, - 'virtual_network_rules': {'readonly': True}, - 'firewall_state': {'readonly': True}, - 'firewall_allow_azure_ips': {'readonly': True}, - 'trusted_id_providers': {'readonly': True}, - 'trusted_id_provider_state': {'readonly': True}, - 'new_tier': {'readonly': True}, - 'current_tier': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "location": {"readonly": True}, + "tags": {"readonly": True}, + "identity": {"readonly": True}, + "account_id": {"readonly": True}, + "provisioning_state": {"readonly": True}, + "state": {"readonly": True}, + "creation_time": {"readonly": True}, + "last_modified_time": {"readonly": True}, + "endpoint": {"readonly": True}, + "default_group": {"readonly": True}, + "encryption_config": {"readonly": True}, + "encryption_state": {"readonly": True}, + "encryption_provisioning_state": {"readonly": True}, + "firewall_rules": {"readonly": True}, + "virtual_network_rules": {"readonly": True}, + "firewall_state": {"readonly": True}, + "firewall_allow_azure_ips": {"readonly": True}, + "trusted_id_providers": {"readonly": True}, + "trusted_id_provider_state": {"readonly": True}, + "new_tier": {"readonly": True}, + "current_tier": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'identity': {'key': 'identity', 'type': 'EncryptionIdentity'}, - 'account_id': {'key': 'properties.accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'creation_time': {'key': 'properties.creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'properties.lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'properties.endpoint', 'type': 'str'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'properties.encryptionState', 'type': 'str'}, - 'encryption_provisioning_state': {'key': 'properties.encryptionProvisioningState', 'type': 'str'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[FirewallRule]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[VirtualNetworkRule]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[TrustedIdProvider]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, - 'current_tier': {'key': 'properties.currentTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccount, self).__init__(**kwargs) + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "location": {"key": "location", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "identity": {"key": "identity", "type": "EncryptionIdentity"}, + "account_id": {"key": "properties.accountId", "type": "str"}, + "provisioning_state": {"key": "properties.provisioningState", "type": "str"}, + "state": {"key": "properties.state", "type": "str"}, + "creation_time": {"key": "properties.creationTime", "type": "iso-8601"}, + "last_modified_time": {"key": "properties.lastModifiedTime", "type": "iso-8601"}, + "endpoint": {"key": "properties.endpoint", "type": "str"}, + "default_group": {"key": "properties.defaultGroup", "type": "str"}, + "encryption_config": {"key": "properties.encryptionConfig", "type": "EncryptionConfig"}, + "encryption_state": {"key": "properties.encryptionState", "type": "str"}, + "encryption_provisioning_state": {"key": "properties.encryptionProvisioningState", "type": "str"}, + "firewall_rules": {"key": "properties.firewallRules", "type": "[FirewallRule]"}, + "virtual_network_rules": {"key": "properties.virtualNetworkRules", "type": "[VirtualNetworkRule]"}, + "firewall_state": {"key": "properties.firewallState", "type": "str"}, + "firewall_allow_azure_ips": {"key": "properties.firewallAllowAzureIps", "type": "str"}, + "trusted_id_providers": {"key": "properties.trustedIdProviders", "type": "[TrustedIdProvider]"}, + "trusted_id_provider_state": {"key": "properties.trustedIdProviderState", "type": "str"}, + "new_tier": {"key": "properties.newTier", "type": "str"}, + "current_tier": {"key": "properties.currentTier", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: # pylint: disable=too-many-locals + """ """ + super().__init__(**kwargs) self.identity = None self.account_id = None self.provisioning_state = None @@ -594,7 +646,7 @@ def __init__( self.current_tier = None -class DataLakeStoreAccountBasic(Resource): +class DataLakeStoreAccountBasic(Resource): # pylint: disable=too-many-instance-attributes """Basic Data Lake Store account information, returned on list calls. Variables are only populated by the server, and will be ignored when sending a request. @@ -607,16 +659,16 @@ class DataLakeStoreAccountBasic(Resource): :vartype type: str :ivar location: The resource location. :vartype location: str - :ivar tags: A set of tags. The resource tags. + :ivar tags: The resource tags. :vartype tags: dict[str, str] :ivar account_id: The unique identifier associated with this Data Lake Store account. :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". + :ivar provisioning_state: The provisioning status of the Data Lake Store account. Known values + are: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", "Resuming", + "Deleting", "Deleted", "Undeleting", and "Canceled". :vartype provisioning_state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", + :ivar state: The state of the Data Lake Store account. Known values are: "Active" and "Suspended". :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState :ivar creation_time: The account creation time. @@ -628,38 +680,36 @@ class DataLakeStoreAccountBasic(Resource): """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'location': {'readonly': True}, - 'tags': {'readonly': True}, - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "location": {"readonly": True}, + "tags": {"readonly": True}, + "account_id": {"readonly": True}, + "provisioning_state": {"readonly": True}, + "state": {"readonly": True}, + "creation_time": {"readonly": True}, + "last_modified_time": {"readonly": True}, + "endpoint": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'location': {'key': 'location', 'type': 'str'}, - 'tags': {'key': 'tags', 'type': '{str}'}, - 'account_id': {'key': 'properties.accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'properties.provisioningState', 'type': 'str'}, - 'state': {'key': 'properties.state', 'type': 'str'}, - 'creation_time': {'key': 'properties.creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'properties.lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'properties.endpoint', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountBasic, self).__init__(**kwargs) + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "location": {"key": "location", "type": "str"}, + "tags": {"key": "tags", "type": "{str}"}, + "account_id": {"key": "properties.accountId", "type": "str"}, + "provisioning_state": {"key": "properties.provisioningState", "type": "str"}, + "state": {"key": "properties.state", "type": "str"}, + "creation_time": {"key": "properties.creationTime", "type": "iso-8601"}, + "last_modified_time": {"key": "properties.lastModifiedTime", "type": "iso-8601"}, + "endpoint": {"key": "properties.endpoint", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.account_id = None self.provisioning_state = None self.state = None @@ -668,7 +718,7 @@ def __init__( self.endpoint = None -class DataLakeStoreAccountListResult(msrest.serialization.Model): +class DataLakeStoreAccountListResult(_serialization.Model): """Data Lake Store account list information response. Variables are only populated by the server, and will be ignored when sending a request. @@ -680,37 +730,36 @@ class DataLakeStoreAccountListResult(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + "value": {"readonly": True}, + "next_link": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[DataLakeStoreAccountBasic]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + "value": {"key": "value", "type": "[DataLakeStoreAccountBasic]"}, + "next_link": {"key": "nextLink", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountListResult, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.next_link = None -class DataLakeStoreAccountPropertiesBasic(msrest.serialization.Model): - """The basic account specific properties that are associated with an underlying Data Lake Store account. +class DataLakeStoreAccountPropertiesBasic(_serialization.Model): + """The basic account specific properties that are associated with an underlying Data Lake Store + account. Variables are only populated by the server, and will be ignored when sending a request. :ivar account_id: The unique identifier associated with this Data Lake Store account. :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". + :ivar provisioning_state: The provisioning status of the Data Lake Store account. Known values + are: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", "Resuming", + "Deleting", "Deleted", "Undeleting", and "Canceled". :vartype provisioning_state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", + :ivar state: The state of the Data Lake Store account. Known values are: "Active" and "Suspended". :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState :ivar creation_time: The account creation time. @@ -722,28 +771,26 @@ class DataLakeStoreAccountPropertiesBasic(msrest.serialization.Model): """ _validation = { - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, + "account_id": {"readonly": True}, + "provisioning_state": {"readonly": True}, + "state": {"readonly": True}, + "creation_time": {"readonly": True}, + "last_modified_time": {"readonly": True}, + "endpoint": {"readonly": True}, } _attribute_map = { - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'provisioningState', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'creation_time': {'key': 'creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'endpoint', 'type': 'str'}, + "account_id": {"key": "accountId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + "state": {"key": "state", "type": "str"}, + "creation_time": {"key": "creationTime", "type": "iso-8601"}, + "last_modified_time": {"key": "lastModifiedTime", "type": "iso-8601"}, + "endpoint": {"key": "endpoint", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountPropertiesBasic, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.account_id = None self.provisioning_state = None self.state = None @@ -752,19 +799,21 @@ def __init__( self.endpoint = None -class DataLakeStoreAccountProperties(DataLakeStoreAccountPropertiesBasic): +class DataLakeStoreAccountProperties( + DataLakeStoreAccountPropertiesBasic +): # pylint: disable=too-many-instance-attributes """Data Lake Store account properties information. Variables are only populated by the server, and will be ignored when sending a request. :ivar account_id: The unique identifier associated with this Data Lake Store account. :vartype account_id: str - :ivar provisioning_state: The provisioning status of the Data Lake Store account. Possible - values include: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", - "Resuming", "Deleting", "Deleted", "Undeleting", "Canceled". + :ivar provisioning_state: The provisioning status of the Data Lake Store account. Known values + are: "Failed", "Creating", "Running", "Succeeded", "Patching", "Suspending", "Resuming", + "Deleting", "Deleted", "Undeleting", and "Canceled". :vartype provisioning_state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountStatus - :ivar state: The state of the Data Lake Store account. Possible values include: "Active", + :ivar state: The state of the Data Lake Store account. Known values are: "Active" and "Suspended". :vartype state: str or ~azure.mgmt.datalake.store.models.DataLakeStoreAccountState :ivar creation_time: The account creation time. @@ -778,11 +827,11 @@ class DataLakeStoreAccountProperties(DataLakeStoreAccountPropertiesBasic): :vartype default_group: str :ivar encryption_config: The Key Vault encryption configuration. :vartype encryption_config: ~azure.mgmt.datalake.store.models.EncryptionConfig - :ivar encryption_state: The current state of encryption for this Data Lake Store account. - Possible values include: "Enabled", "Disabled". + :ivar encryption_state: The current state of encryption for this Data Lake Store account. Known + values are: "Enabled" and "Disabled". :vartype encryption_state: str or ~azure.mgmt.datalake.store.models.EncryptionState :ivar encryption_provisioning_state: The current state of encryption provisioning for this Data - Lake Store account. Possible values include: "Creating", "Succeeded". + Lake Store account. Known values are: "Creating" and "Succeeded". :vartype encryption_provisioning_state: str or ~azure.mgmt.datalake.store.models.EncryptionProvisioningState :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. @@ -791,77 +840,75 @@ class DataLakeStoreAccountProperties(DataLakeStoreAccountPropertiesBasic): Store account. :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.VirtualNetworkRule] :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store - account. Possible values include: "Enabled", "Disabled". + account. Known values are: "Enabled" and "Disabled". :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". :vartype firewall_allow_azure_ips: str or ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState :ivar trusted_id_providers: The list of trusted identity providers associated with this Data Lake Store account. :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.TrustedIdProvider] :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for - this Data Lake Store account. Possible values include: "Enabled", "Disabled". + this Data Lake Store account. Known values are: "Enabled" and "Disabled". :vartype trusted_id_provider_state: str or ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :ivar new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". + :ivar new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType - :ivar current_tier: The commitment tier in use for the current month. Possible values include: + :ivar current_tier: The commitment tier in use for the current month. Known values are: "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". + "Commitment_1PB", and "Commitment_5PB". :vartype current_tier: str or ~azure.mgmt.datalake.store.models.TierType """ _validation = { - 'account_id': {'readonly': True}, - 'provisioning_state': {'readonly': True}, - 'state': {'readonly': True}, - 'creation_time': {'readonly': True}, - 'last_modified_time': {'readonly': True}, - 'endpoint': {'readonly': True}, - 'default_group': {'readonly': True}, - 'encryption_config': {'readonly': True}, - 'encryption_state': {'readonly': True}, - 'encryption_provisioning_state': {'readonly': True}, - 'firewall_rules': {'readonly': True}, - 'virtual_network_rules': {'readonly': True}, - 'firewall_state': {'readonly': True}, - 'firewall_allow_azure_ips': {'readonly': True}, - 'trusted_id_providers': {'readonly': True}, - 'trusted_id_provider_state': {'readonly': True}, - 'new_tier': {'readonly': True}, - 'current_tier': {'readonly': True}, + "account_id": {"readonly": True}, + "provisioning_state": {"readonly": True}, + "state": {"readonly": True}, + "creation_time": {"readonly": True}, + "last_modified_time": {"readonly": True}, + "endpoint": {"readonly": True}, + "default_group": {"readonly": True}, + "encryption_config": {"readonly": True}, + "encryption_state": {"readonly": True}, + "encryption_provisioning_state": {"readonly": True}, + "firewall_rules": {"readonly": True}, + "virtual_network_rules": {"readonly": True}, + "firewall_state": {"readonly": True}, + "firewall_allow_azure_ips": {"readonly": True}, + "trusted_id_providers": {"readonly": True}, + "trusted_id_provider_state": {"readonly": True}, + "new_tier": {"readonly": True}, + "current_tier": {"readonly": True}, } _attribute_map = { - 'account_id': {'key': 'accountId', 'type': 'str'}, - 'provisioning_state': {'key': 'provisioningState', 'type': 'str'}, - 'state': {'key': 'state', 'type': 'str'}, - 'creation_time': {'key': 'creationTime', 'type': 'iso-8601'}, - 'last_modified_time': {'key': 'lastModifiedTime', 'type': 'iso-8601'}, - 'endpoint': {'key': 'endpoint', 'type': 'str'}, - 'default_group': {'key': 'defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'encryptionConfig', 'type': 'EncryptionConfig'}, - 'encryption_state': {'key': 'encryptionState', 'type': 'str'}, - 'encryption_provisioning_state': {'key': 'encryptionProvisioningState', 'type': 'str'}, - 'firewall_rules': {'key': 'firewallRules', 'type': '[FirewallRule]'}, - 'virtual_network_rules': {'key': 'virtualNetworkRules', 'type': '[VirtualNetworkRule]'}, - 'firewall_state': {'key': 'firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'trustedIdProviders', 'type': '[TrustedIdProvider]'}, - 'trusted_id_provider_state': {'key': 'trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'newTier', 'type': 'str'}, - 'current_tier': {'key': 'currentTier', 'type': 'str'}, - } - - def __init__( - self, - **kwargs - ): - super(DataLakeStoreAccountProperties, self).__init__(**kwargs) + "account_id": {"key": "accountId", "type": "str"}, + "provisioning_state": {"key": "provisioningState", "type": "str"}, + "state": {"key": "state", "type": "str"}, + "creation_time": {"key": "creationTime", "type": "iso-8601"}, + "last_modified_time": {"key": "lastModifiedTime", "type": "iso-8601"}, + "endpoint": {"key": "endpoint", "type": "str"}, + "default_group": {"key": "defaultGroup", "type": "str"}, + "encryption_config": {"key": "encryptionConfig", "type": "EncryptionConfig"}, + "encryption_state": {"key": "encryptionState", "type": "str"}, + "encryption_provisioning_state": {"key": "encryptionProvisioningState", "type": "str"}, + "firewall_rules": {"key": "firewallRules", "type": "[FirewallRule]"}, + "virtual_network_rules": {"key": "virtualNetworkRules", "type": "[VirtualNetworkRule]"}, + "firewall_state": {"key": "firewallState", "type": "str"}, + "firewall_allow_azure_ips": {"key": "firewallAllowAzureIps", "type": "str"}, + "trusted_id_providers": {"key": "trustedIdProviders", "type": "[TrustedIdProvider]"}, + "trusted_id_provider_state": {"key": "trustedIdProviderState", "type": "str"}, + "new_tier": {"key": "newTier", "type": "str"}, + "current_tier": {"key": "currentTier", "type": "str"}, + } + + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.default_group = None self.encryption_config = None self.encryption_state = None @@ -876,50 +923,59 @@ def __init__( self.current_tier = None -class EncryptionConfig(msrest.serialization.Model): +class EncryptionConfig(_serialization.Model): """The encryption configuration for the account. All required parameters must be populated in order to send to Azure. - :param type: Required. The type of encryption configuration being used. Currently the only - supported types are 'UserManaged' and 'ServiceManaged'. Possible values include: "UserManaged", + :ivar type: The type of encryption configuration being used. Currently the only supported types + are 'UserManaged' and 'ServiceManaged'. Required. Known values are: "UserManaged" and "ServiceManaged". - :type type: str or ~azure.mgmt.datalake.store.models.EncryptionConfigType - :param key_vault_meta_info: The Key Vault information for connecting to user managed encryption + :vartype type: str or ~azure.mgmt.datalake.store.models.EncryptionConfigType + :ivar key_vault_meta_info: The Key Vault information for connecting to user managed encryption keys. - :type key_vault_meta_info: ~azure.mgmt.datalake.store.models.KeyVaultMetaInfo + :vartype key_vault_meta_info: ~azure.mgmt.datalake.store.models.KeyVaultMetaInfo """ _validation = { - 'type': {'required': True}, + "type": {"required": True}, } _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'key_vault_meta_info': {'key': 'keyVaultMetaInfo', 'type': 'KeyVaultMetaInfo'}, + "type": {"key": "type", "type": "str"}, + "key_vault_meta_info": {"key": "keyVaultMetaInfo", "type": "KeyVaultMetaInfo"}, } def __init__( self, *, - type: Union[str, "EncryptionConfigType"], - key_vault_meta_info: Optional["KeyVaultMetaInfo"] = None, - **kwargs - ): - super(EncryptionConfig, self).__init__(**kwargs) + type: Union[str, "_models.EncryptionConfigType"], + key_vault_meta_info: Optional["_models.KeyVaultMetaInfo"] = None, + **kwargs: Any + ) -> None: + """ + :keyword type: The type of encryption configuration being used. Currently the only supported + types are 'UserManaged' and 'ServiceManaged'. Required. Known values are: "UserManaged" and + "ServiceManaged". + :paramtype type: str or ~azure.mgmt.datalake.store.models.EncryptionConfigType + :keyword key_vault_meta_info: The Key Vault information for connecting to user managed + encryption keys. + :paramtype key_vault_meta_info: ~azure.mgmt.datalake.store.models.KeyVaultMetaInfo + """ + super().__init__(**kwargs) self.type = type self.key_vault_meta_info = key_vault_meta_info -class EncryptionIdentity(msrest.serialization.Model): +class EncryptionIdentity(_serialization.Model): """The encryption identity properties. 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 Azure. - :ivar type: Required. The type of encryption being used. Currently the only supported type is - 'SystemAssigned'. Default value: "SystemAssigned". + :ivar type: The type of encryption being used. Currently the only supported type is + 'SystemAssigned'. Required. Default value is "SystemAssigned". :vartype type: str :ivar principal_id: The principal identifier associated with the encryption. :vartype principal_id: str @@ -928,29 +984,27 @@ class EncryptionIdentity(msrest.serialization.Model): """ _validation = { - 'type': {'required': True, 'constant': True}, - 'principal_id': {'readonly': True}, - 'tenant_id': {'readonly': True}, + "type": {"required": True, "constant": True}, + "principal_id": {"readonly": True}, + "tenant_id": {"readonly": True}, } _attribute_map = { - 'type': {'key': 'type', 'type': 'str'}, - 'principal_id': {'key': 'principalId', 'type': 'str'}, - 'tenant_id': {'key': 'tenantId', 'type': 'str'}, + "type": {"key": "type", "type": "str"}, + "principal_id": {"key": "principalId", "type": "str"}, + "tenant_id": {"key": "tenantId", "type": "str"}, } type = "SystemAssigned" - def __init__( - self, - **kwargs - ): - super(EncryptionIdentity, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.principal_id = None self.tenant_id = None -class SubResource(msrest.serialization.Model): +class SubResource(_serialization.Model): """The resource model definition for a nested resource. Variables are only populated by the server, and will be ignored when sending a request. @@ -964,22 +1018,20 @@ class SubResource(msrest.serialization.Model): """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(SubResource, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.id = None self.name = None self.type = None @@ -1005,31 +1057,29 @@ class FirewallRule(SubResource): """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'start_ip_address': {'readonly': True}, - 'end_ip_address': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "start_ip_address": {"readonly": True}, + "end_ip_address": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "start_ip_address": {"key": "properties.startIpAddress", "type": "str"}, + "end_ip_address": {"key": "properties.endIpAddress", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(FirewallRule, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.start_ip_address = None self.end_ip_address = None -class FirewallRuleListResult(msrest.serialization.Model): +class FirewallRuleListResult(_serialization.Model): """Data Lake Store firewall rule list information. Variables are only populated by the server, and will be ignored when sending a request. @@ -1041,65 +1091,67 @@ class FirewallRuleListResult(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + "value": {"readonly": True}, + "next_link": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[FirewallRule]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + "value": {"key": "value", "type": "[FirewallRule]"}, + "next_link": {"key": "nextLink", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(FirewallRuleListResult, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.next_link = None -class KeyVaultMetaInfo(msrest.serialization.Model): +class KeyVaultMetaInfo(_serialization.Model): """Metadata information used by account encryption. All required parameters must be populated in order to send to Azure. - :param key_vault_resource_id: Required. The resource identifier for the user managed Key Vault - being used to encrypt. - :type key_vault_resource_id: str - :param encryption_key_name: Required. The name of the user managed encryption key. - :type encryption_key_name: str - :param encryption_key_version: Required. The version of the user managed encryption key. - :type encryption_key_version: str + :ivar key_vault_resource_id: The resource identifier for the user managed Key Vault being used + to encrypt. Required. + :vartype key_vault_resource_id: str + :ivar encryption_key_name: The name of the user managed encryption key. Required. + :vartype encryption_key_name: str + :ivar encryption_key_version: The version of the user managed encryption key. Required. + :vartype encryption_key_version: str """ _validation = { - 'key_vault_resource_id': {'required': True}, - 'encryption_key_name': {'required': True}, - 'encryption_key_version': {'required': True}, + "key_vault_resource_id": {"required": True}, + "encryption_key_name": {"required": True}, + "encryption_key_version": {"required": True}, } _attribute_map = { - 'key_vault_resource_id': {'key': 'keyVaultResourceId', 'type': 'str'}, - 'encryption_key_name': {'key': 'encryptionKeyName', 'type': 'str'}, - 'encryption_key_version': {'key': 'encryptionKeyVersion', 'type': 'str'}, + "key_vault_resource_id": {"key": "keyVaultResourceId", "type": "str"}, + "encryption_key_name": {"key": "encryptionKeyName", "type": "str"}, + "encryption_key_version": {"key": "encryptionKeyVersion", "type": "str"}, } def __init__( - self, - *, - key_vault_resource_id: str, - encryption_key_name: str, - encryption_key_version: str, - **kwargs - ): - super(KeyVaultMetaInfo, self).__init__(**kwargs) + self, *, key_vault_resource_id: str, encryption_key_name: str, encryption_key_version: str, **kwargs: Any + ) -> None: + """ + :keyword key_vault_resource_id: The resource identifier for the user managed Key Vault being + used to encrypt. Required. + :paramtype key_vault_resource_id: str + :keyword encryption_key_name: The name of the user managed encryption key. Required. + :paramtype encryption_key_name: str + :keyword encryption_key_version: The version of the user managed encryption key. Required. + :paramtype encryption_key_version: str + """ + super().__init__(**kwargs) self.key_vault_resource_id = key_vault_resource_id self.encryption_key_name = encryption_key_name self.encryption_key_version = encryption_key_version -class NameAvailabilityInformation(msrest.serialization.Model): +class NameAvailabilityInformation(_serialization.Model): """Data Lake Store account name availability result information. Variables are only populated by the server, and will be ignored when sending a request. @@ -1116,65 +1168,62 @@ class NameAvailabilityInformation(msrest.serialization.Model): """ _validation = { - 'name_available': {'readonly': True}, - 'reason': {'readonly': True}, - 'message': {'readonly': True}, + "name_available": {"readonly": True}, + "reason": {"readonly": True}, + "message": {"readonly": True}, } _attribute_map = { - 'name_available': {'key': 'nameAvailable', 'type': 'bool'}, - 'reason': {'key': 'reason', 'type': 'str'}, - 'message': {'key': 'message', 'type': 'str'}, + "name_available": {"key": "nameAvailable", "type": "bool"}, + "reason": {"key": "reason", "type": "str"}, + "message": {"key": "message", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(NameAvailabilityInformation, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.name_available = None self.reason = None self.message = None -class Operation(msrest.serialization.Model): +class Operation(_serialization.Model): """An available operation for Data Lake Store. Variables are only populated by the server, and will be ignored when sending a request. :ivar name: The name of the operation. :vartype name: str - :param display: The display information for the operation. - :type display: ~azure.mgmt.datalake.store.models.OperationDisplay - :ivar origin: The intended executor of the operation. Possible values include: "user", - "system", "user,system". + :ivar display: The display information for the operation. + :vartype display: ~azure.mgmt.datalake.store.models.OperationDisplay + :ivar origin: The intended executor of the operation. Known values are: "user", "system", and + "user,system". :vartype origin: str or ~azure.mgmt.datalake.store.models.OperationOrigin """ _validation = { - 'name': {'readonly': True}, - 'origin': {'readonly': True}, + "name": {"readonly": True}, + "origin": {"readonly": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'display': {'key': 'display', 'type': 'OperationDisplay'}, - 'origin': {'key': 'origin', 'type': 'str'}, + "name": {"key": "name", "type": "str"}, + "display": {"key": "display", "type": "OperationDisplay"}, + "origin": {"key": "origin", "type": "str"}, } - def __init__( - self, - *, - display: Optional["OperationDisplay"] = None, - **kwargs - ): - super(Operation, self).__init__(**kwargs) + def __init__(self, *, display: Optional["_models.OperationDisplay"] = None, **kwargs: Any) -> None: + """ + :keyword display: The display information for the operation. + :paramtype display: ~azure.mgmt.datalake.store.models.OperationDisplay + """ + super().__init__(**kwargs) self.name = None self.display = display self.origin = None -class OperationDisplay(msrest.serialization.Model): +class OperationDisplay(_serialization.Model): """The display information for a particular operation. Variables are only populated by the server, and will be ignored when sending a request. @@ -1190,31 +1239,29 @@ class OperationDisplay(msrest.serialization.Model): """ _validation = { - 'provider': {'readonly': True}, - 'resource': {'readonly': True}, - 'operation': {'readonly': True}, - 'description': {'readonly': True}, + "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'}, + "provider": {"key": "provider", "type": "str"}, + "resource": {"key": "resource", "type": "str"}, + "operation": {"key": "operation", "type": "str"}, + "description": {"key": "description", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(OperationDisplay, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.provider = None self.resource = None self.operation = None self.description = None -class OperationListResult(msrest.serialization.Model): +class OperationListResult(_serialization.Model): """The list of available operations for Data Lake Store. Variables are only populated by the server, and will be ignored when sending a request. @@ -1226,20 +1273,18 @@ class OperationListResult(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + "value": {"readonly": True}, + "next_link": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[Operation]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + "value": {"key": "value", "type": "[Operation]"}, + "next_link": {"key": "nextLink", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(OperationListResult, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.next_link = None @@ -1260,28 +1305,26 @@ class TrustedIdProvider(SubResource): """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'id_provider': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "id_provider": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "id_provider": {"key": "properties.idProvider", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(TrustedIdProvider, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.id_provider = None -class TrustedIdProviderListResult(msrest.serialization.Model): +class TrustedIdProviderListResult(_serialization.Model): """Data Lake Store trusted identity provider list information. Variables are only populated by the server, and will be ignored when sending a request. @@ -1293,78 +1336,82 @@ class TrustedIdProviderListResult(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + "value": {"readonly": True}, + "next_link": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[TrustedIdProvider]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + "value": {"key": "value", "type": "[TrustedIdProvider]"}, + "next_link": {"key": "nextLink", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(TrustedIdProviderListResult, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.next_link = None -class UpdateDataLakeStoreAccountParameters(msrest.serialization.Model): +class UpdateDataLakeStoreAccountParameters(_serialization.Model): """Data Lake Store account information to update. - :param tags: A set of tags. Resource tags. - :type tags: dict[str, str] - :param default_group: The default owner group for all new folders and files created in the Data + :ivar tags: Resource tags. + :vartype tags: dict[str, str] + :ivar default_group: The default owner group for all new folders and files created in the Data Lake Store account. - :type default_group: str - :param encryption_config: Used for rotation of user managed Key Vault keys. Can only be used to + :vartype default_group: str + :ivar encryption_config: Used for rotation of user managed Key Vault keys. Can only be used to rotate a user managed encryption Key Vault key. - :type encryption_config: ~azure.mgmt.datalake.store.models.UpdateEncryptionConfig - :param firewall_rules: The list of firewall rules associated with this Data Lake Store account. - :type firewall_rules: + :vartype encryption_config: ~azure.mgmt.datalake.store.models.UpdateEncryptionConfig + :ivar firewall_rules: The list of firewall rules associated with this Data Lake Store account. + :vartype firewall_rules: list[~azure.mgmt.datalake.store.models.UpdateFirewallRuleWithAccountParameters] - :param virtual_network_rules: The list of virtual network rules associated with this Data Lake + :ivar virtual_network_rules: The list of virtual network rules associated with this Data Lake Store account. - :type virtual_network_rules: + :vartype virtual_network_rules: list[~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleWithAccountParameters] - :param firewall_state: The current state of the IP address firewall for this Data Lake Store + :ivar firewall_state: The current state of the IP address firewall for this Data Lake Store account. Disabling the firewall does not remove existing rules, they will just be ignored until - the firewall is re-enabled. Possible values include: "Enabled", "Disabled". - :type firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState - :param firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating - within Azure through the firewall. If the firewall is disabled, this is not enforced. Possible - values include: "Enabled", "Disabled". - :type firewall_allow_azure_ips: str or + the firewall is re-enabled. Known values are: "Enabled" and "Disabled". + :vartype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState + :ivar firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". + :vartype firewall_allow_azure_ips: str or ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState - :param trusted_id_providers: The list of trusted identity providers associated with this Data + :ivar trusted_id_providers: The list of trusted identity providers associated with this Data Lake Store account. - :type trusted_id_providers: + :vartype trusted_id_providers: list[~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderWithAccountParameters] - :param trusted_id_provider_state: The current state of the trusted identity provider feature - for this Data Lake Store account. Disabling trusted identity provider functionality does not - remove the providers, they will just be ignored until this feature is re-enabled. Possible - values include: "Enabled", "Disabled". - :type trusted_id_provider_state: str or + :ivar trusted_id_provider_state: The current state of the trusted identity provider feature for + this Data Lake Store account. Disabling trusted identity provider functionality does not remove + the providers, they will just be ignored until this feature is re-enabled. Known values are: + "Enabled" and "Disabled". + :vartype trusted_id_provider_state: str or ~azure.mgmt.datalake.store.models.TrustedIdProviderState - :param new_tier: The commitment tier to use for next month. Possible values include: - "Consumption", "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", - "Commitment_1PB", "Commitment_5PB". - :type new_tier: str or ~azure.mgmt.datalake.store.models.TierType + :ivar new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". + :vartype new_tier: str or ~azure.mgmt.datalake.store.models.TierType """ _attribute_map = { - 'tags': {'key': 'tags', 'type': '{str}'}, - 'default_group': {'key': 'properties.defaultGroup', 'type': 'str'}, - 'encryption_config': {'key': 'properties.encryptionConfig', 'type': 'UpdateEncryptionConfig'}, - 'firewall_rules': {'key': 'properties.firewallRules', 'type': '[UpdateFirewallRuleWithAccountParameters]'}, - 'virtual_network_rules': {'key': 'properties.virtualNetworkRules', 'type': '[UpdateVirtualNetworkRuleWithAccountParameters]'}, - 'firewall_state': {'key': 'properties.firewallState', 'type': 'str'}, - 'firewall_allow_azure_ips': {'key': 'properties.firewallAllowAzureIps', 'type': 'str'}, - 'trusted_id_providers': {'key': 'properties.trustedIdProviders', 'type': '[UpdateTrustedIdProviderWithAccountParameters]'}, - 'trusted_id_provider_state': {'key': 'properties.trustedIdProviderState', 'type': 'str'}, - 'new_tier': {'key': 'properties.newTier', 'type': 'str'}, + "tags": {"key": "tags", "type": "{str}"}, + "default_group": {"key": "properties.defaultGroup", "type": "str"}, + "encryption_config": {"key": "properties.encryptionConfig", "type": "UpdateEncryptionConfig"}, + "firewall_rules": {"key": "properties.firewallRules", "type": "[UpdateFirewallRuleWithAccountParameters]"}, + "virtual_network_rules": { + "key": "properties.virtualNetworkRules", + "type": "[UpdateVirtualNetworkRuleWithAccountParameters]", + }, + "firewall_state": {"key": "properties.firewallState", "type": "str"}, + "firewall_allow_azure_ips": {"key": "properties.firewallAllowAzureIps", "type": "str"}, + "trusted_id_providers": { + "key": "properties.trustedIdProviders", + "type": "[UpdateTrustedIdProviderWithAccountParameters]", + }, + "trusted_id_provider_state": {"key": "properties.trustedIdProviderState", "type": "str"}, + "new_tier": {"key": "properties.newTier", "type": "str"}, } def __init__( @@ -1372,17 +1419,58 @@ def __init__( *, tags: Optional[Dict[str, str]] = None, default_group: Optional[str] = None, - encryption_config: Optional["UpdateEncryptionConfig"] = None, - firewall_rules: Optional[List["UpdateFirewallRuleWithAccountParameters"]] = None, - virtual_network_rules: Optional[List["UpdateVirtualNetworkRuleWithAccountParameters"]] = None, - firewall_state: Optional[Union[str, "FirewallState"]] = None, - firewall_allow_azure_ips: Optional[Union[str, "FirewallAllowAzureIpsState"]] = None, - trusted_id_providers: Optional[List["UpdateTrustedIdProviderWithAccountParameters"]] = None, - trusted_id_provider_state: Optional[Union[str, "TrustedIdProviderState"]] = None, - new_tier: Optional[Union[str, "TierType"]] = None, - **kwargs - ): - super(UpdateDataLakeStoreAccountParameters, self).__init__(**kwargs) + encryption_config: Optional["_models.UpdateEncryptionConfig"] = None, + firewall_rules: Optional[List["_models.UpdateFirewallRuleWithAccountParameters"]] = None, + virtual_network_rules: Optional[List["_models.UpdateVirtualNetworkRuleWithAccountParameters"]] = None, + firewall_state: Optional[Union[str, "_models.FirewallState"]] = None, + firewall_allow_azure_ips: Optional[Union[str, "_models.FirewallAllowAzureIpsState"]] = None, + trusted_id_providers: Optional[List["_models.UpdateTrustedIdProviderWithAccountParameters"]] = None, + trusted_id_provider_state: Optional[Union[str, "_models.TrustedIdProviderState"]] = None, + new_tier: Optional[Union[str, "_models.TierType"]] = None, + **kwargs: Any + ) -> None: + """ + :keyword tags: Resource tags. + :paramtype tags: dict[str, str] + :keyword default_group: The default owner group for all new folders and files created in the + Data Lake Store account. + :paramtype default_group: str + :keyword encryption_config: Used for rotation of user managed Key Vault keys. Can only be used + to rotate a user managed encryption Key Vault key. + :paramtype encryption_config: ~azure.mgmt.datalake.store.models.UpdateEncryptionConfig + :keyword firewall_rules: The list of firewall rules associated with this Data Lake Store + account. + :paramtype firewall_rules: + list[~azure.mgmt.datalake.store.models.UpdateFirewallRuleWithAccountParameters] + :keyword virtual_network_rules: The list of virtual network rules associated with this Data + Lake Store account. + :paramtype virtual_network_rules: + list[~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleWithAccountParameters] + :keyword firewall_state: The current state of the IP address firewall for this Data Lake Store + account. Disabling the firewall does not remove existing rules, they will just be ignored until + the firewall is re-enabled. Known values are: "Enabled" and "Disabled". + :paramtype firewall_state: str or ~azure.mgmt.datalake.store.models.FirewallState + :keyword firewall_allow_azure_ips: The current state of allowing or disallowing IPs originating + within Azure through the firewall. If the firewall is disabled, this is not enforced. Known + values are: "Enabled" and "Disabled". + :paramtype firewall_allow_azure_ips: str or + ~azure.mgmt.datalake.store.models.FirewallAllowAzureIpsState + :keyword trusted_id_providers: The list of trusted identity providers associated with this Data + Lake Store account. + :paramtype trusted_id_providers: + list[~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderWithAccountParameters] + :keyword trusted_id_provider_state: The current state of the trusted identity provider feature + for this Data Lake Store account. Disabling trusted identity provider functionality does not + remove the providers, they will just be ignored until this feature is re-enabled. Known values + are: "Enabled" and "Disabled". + :paramtype trusted_id_provider_state: str or + ~azure.mgmt.datalake.store.models.TrustedIdProviderState + :keyword new_tier: The commitment tier to use for next month. Known values are: "Consumption", + "Commitment_1TB", "Commitment_10TB", "Commitment_100TB", "Commitment_500TB", "Commitment_1PB", + and "Commitment_5PB". + :paramtype new_tier: str or ~azure.mgmt.datalake.store.models.TierType + """ + super().__init__(**kwargs) self.tags = tags self.default_group = default_group self.encryption_config = encryption_config @@ -1395,229 +1483,238 @@ def __init__( self.new_tier = new_tier -class UpdateEncryptionConfig(msrest.serialization.Model): +class UpdateEncryptionConfig(_serialization.Model): """The encryption configuration used to update a user managed Key Vault key. - :param key_vault_meta_info: The updated Key Vault key to use in user managed key rotation. - :type key_vault_meta_info: ~azure.mgmt.datalake.store.models.UpdateKeyVaultMetaInfo + :ivar key_vault_meta_info: The updated Key Vault key to use in user managed key rotation. + :vartype key_vault_meta_info: ~azure.mgmt.datalake.store.models.UpdateKeyVaultMetaInfo """ _attribute_map = { - 'key_vault_meta_info': {'key': 'keyVaultMetaInfo', 'type': 'UpdateKeyVaultMetaInfo'}, + "key_vault_meta_info": {"key": "keyVaultMetaInfo", "type": "UpdateKeyVaultMetaInfo"}, } def __init__( - self, - *, - key_vault_meta_info: Optional["UpdateKeyVaultMetaInfo"] = None, - **kwargs - ): - super(UpdateEncryptionConfig, self).__init__(**kwargs) + self, *, key_vault_meta_info: Optional["_models.UpdateKeyVaultMetaInfo"] = None, **kwargs: Any + ) -> None: + """ + :keyword key_vault_meta_info: The updated Key Vault key to use in user managed key rotation. + :paramtype key_vault_meta_info: ~azure.mgmt.datalake.store.models.UpdateKeyVaultMetaInfo + """ + super().__init__(**kwargs) self.key_vault_meta_info = key_vault_meta_info -class UpdateFirewallRuleParameters(msrest.serialization.Model): +class UpdateFirewallRuleParameters(_serialization.Model): """The parameters used to update a firewall rule. - :param start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or + :ivar start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + :vartype start_ip_address: str + :ivar end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str + :vartype end_ip_address: str """ _attribute_map = { - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, + "start_ip_address": {"key": "properties.startIpAddress", "type": "str"}, + "end_ip_address": {"key": "properties.endIpAddress", "type": "str"}, } def __init__( - self, - *, - start_ip_address: Optional[str] = None, - end_ip_address: Optional[str] = None, - **kwargs - ): - super(UpdateFirewallRuleParameters, self).__init__(**kwargs) + self, *, start_ip_address: Optional[str] = None, end_ip_address: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword start_ip_address: The start IP address for the firewall rule. This can be either ipv4 + or ipv6. Start and End should be in the same protocol. + :paramtype start_ip_address: str + :keyword end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. + :paramtype end_ip_address: str + """ + super().__init__(**kwargs) self.start_ip_address = start_ip_address self.end_ip_address = end_ip_address -class UpdateFirewallRuleWithAccountParameters(msrest.serialization.Model): +class UpdateFirewallRuleWithAccountParameters(_serialization.Model): """The parameters used to update a firewall rule while updating a Data Lake Store account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the firewall rule to update. - :type name: str - :param start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or + :ivar name: The unique name of the firewall rule to update. Required. + :vartype name: str + :ivar start_ip_address: The start IP address for the firewall rule. This can be either ipv4 or ipv6. Start and End should be in the same protocol. - :type start_ip_address: str - :param end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + :vartype start_ip_address: str + :ivar end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or ipv6. Start and End should be in the same protocol. - :type end_ip_address: str + :vartype end_ip_address: str """ _validation = { - 'name': {'required': True}, + "name": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'start_ip_address': {'key': 'properties.startIpAddress', 'type': 'str'}, - 'end_ip_address': {'key': 'properties.endIpAddress', 'type': 'str'}, + "name": {"key": "name", "type": "str"}, + "start_ip_address": {"key": "properties.startIpAddress", "type": "str"}, + "end_ip_address": {"key": "properties.endIpAddress", "type": "str"}, } def __init__( - self, - *, - name: str, - start_ip_address: Optional[str] = None, - end_ip_address: Optional[str] = None, - **kwargs - ): - super(UpdateFirewallRuleWithAccountParameters, self).__init__(**kwargs) + self, *, name: str, start_ip_address: Optional[str] = None, end_ip_address: Optional[str] = None, **kwargs: Any + ) -> None: + """ + :keyword name: The unique name of the firewall rule to update. Required. + :paramtype name: str + :keyword start_ip_address: The start IP address for the firewall rule. This can be either ipv4 + or ipv6. Start and End should be in the same protocol. + :paramtype start_ip_address: str + :keyword end_ip_address: The end IP address for the firewall rule. This can be either ipv4 or + ipv6. Start and End should be in the same protocol. + :paramtype end_ip_address: str + """ + super().__init__(**kwargs) self.name = name self.start_ip_address = start_ip_address self.end_ip_address = end_ip_address -class UpdateKeyVaultMetaInfo(msrest.serialization.Model): +class UpdateKeyVaultMetaInfo(_serialization.Model): """The Key Vault update information used for user managed key rotation. - :param encryption_key_version: The version of the user managed encryption key to update through + :ivar encryption_key_version: The version of the user managed encryption key to update through a key rotation. - :type encryption_key_version: str + :vartype encryption_key_version: str """ _attribute_map = { - 'encryption_key_version': {'key': 'encryptionKeyVersion', 'type': 'str'}, + "encryption_key_version": {"key": "encryptionKeyVersion", "type": "str"}, } - def __init__( - self, - *, - encryption_key_version: Optional[str] = None, - **kwargs - ): - super(UpdateKeyVaultMetaInfo, self).__init__(**kwargs) + def __init__(self, *, encryption_key_version: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword encryption_key_version: The version of the user managed encryption key to update + through a key rotation. + :paramtype encryption_key_version: str + """ + super().__init__(**kwargs) self.encryption_key_version = encryption_key_version -class UpdateTrustedIdProviderParameters(msrest.serialization.Model): +class UpdateTrustedIdProviderParameters(_serialization.Model): """The parameters used to update a trusted identity provider. - :param id_provider: The URL of this trusted identity provider. - :type id_provider: str + :ivar id_provider: The URL of this trusted identity provider. + :vartype id_provider: str """ _attribute_map = { - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, + "id_provider": {"key": "properties.idProvider", "type": "str"}, } - def __init__( - self, - *, - id_provider: Optional[str] = None, - **kwargs - ): - super(UpdateTrustedIdProviderParameters, self).__init__(**kwargs) + def __init__(self, *, id_provider: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword id_provider: The URL of this trusted identity provider. + :paramtype id_provider: str + """ + super().__init__(**kwargs) self.id_provider = id_provider -class UpdateTrustedIdProviderWithAccountParameters(msrest.serialization.Model): - """The parameters used to update a trusted identity provider while updating a Data Lake Store account. +class UpdateTrustedIdProviderWithAccountParameters(_serialization.Model): + """The parameters used to update a trusted identity provider while updating a Data Lake Store + account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the trusted identity provider to update. - :type name: str - :param id_provider: The URL of this trusted identity provider. - :type id_provider: str + :ivar name: The unique name of the trusted identity provider to update. Required. + :vartype name: str + :ivar id_provider: The URL of this trusted identity provider. + :vartype id_provider: str """ _validation = { - 'name': {'required': True}, + "name": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'id_provider': {'key': 'properties.idProvider', 'type': 'str'}, - } - - def __init__( - self, - *, - name: str, - id_provider: Optional[str] = None, - **kwargs - ): - super(UpdateTrustedIdProviderWithAccountParameters, self).__init__(**kwargs) + "name": {"key": "name", "type": "str"}, + "id_provider": {"key": "properties.idProvider", "type": "str"}, + } + + def __init__(self, *, name: str, id_provider: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword name: The unique name of the trusted identity provider to update. Required. + :paramtype name: str + :keyword id_provider: The URL of this trusted identity provider. + :paramtype id_provider: str + """ + super().__init__(**kwargs) self.name = name self.id_provider = id_provider -class UpdateVirtualNetworkRuleParameters(msrest.serialization.Model): +class UpdateVirtualNetworkRuleParameters(_serialization.Model): """The parameters used to update a virtual network rule. - :param subnet_id: The resource identifier for the subnet. - :type subnet_id: str + :ivar subnet_id: The resource identifier for the subnet. + :vartype subnet_id: str """ _attribute_map = { - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, + "subnet_id": {"key": "properties.subnetId", "type": "str"}, } - def __init__( - self, - *, - subnet_id: Optional[str] = None, - **kwargs - ): - super(UpdateVirtualNetworkRuleParameters, self).__init__(**kwargs) + def __init__(self, *, subnet_id: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword subnet_id: The resource identifier for the subnet. + :paramtype subnet_id: str + """ + super().__init__(**kwargs) self.subnet_id = subnet_id -class UpdateVirtualNetworkRuleWithAccountParameters(msrest.serialization.Model): +class UpdateVirtualNetworkRuleWithAccountParameters(_serialization.Model): """The parameters used to update a virtual network rule while updating a Data Lake Store account. All required parameters must be populated in order to send to Azure. - :param name: Required. The unique name of the virtual network rule to update. - :type name: str - :param subnet_id: The resource identifier for the subnet. - :type subnet_id: str + :ivar name: The unique name of the virtual network rule to update. Required. + :vartype name: str + :ivar subnet_id: The resource identifier for the subnet. + :vartype subnet_id: str """ _validation = { - 'name': {'required': True}, + "name": {"required": True}, } _attribute_map = { - 'name': {'key': 'name', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, - } - - def __init__( - self, - *, - name: str, - subnet_id: Optional[str] = None, - **kwargs - ): - super(UpdateVirtualNetworkRuleWithAccountParameters, self).__init__(**kwargs) + "name": {"key": "name", "type": "str"}, + "subnet_id": {"key": "properties.subnetId", "type": "str"}, + } + + def __init__(self, *, name: str, subnet_id: Optional[str] = None, **kwargs: Any) -> None: + """ + :keyword name: The unique name of the virtual network rule to update. Required. + :paramtype name: str + :keyword subnet_id: The resource identifier for the subnet. + :paramtype subnet_id: str + """ + super().__init__(**kwargs) self.name = name self.subnet_id = subnet_id -class Usage(msrest.serialization.Model): +class Usage(_serialization.Model): """Describes the Resource Usage. Variables are only populated by the server, and will be ignored when sending a request. - :ivar unit: Gets the unit of measurement. Possible values include: "Count", "Bytes", "Seconds", - "Percent", "CountsPerSecond", "BytesPerSecond". + :ivar unit: Gets the unit of measurement. Known values are: "Count", "Bytes", "Seconds", + "Percent", "CountsPerSecond", and "BytesPerSecond". :vartype unit: str or ~azure.mgmt.datalake.store.models.UsageUnit :ivar id: Resource identifier. :vartype id: str @@ -1630,26 +1727,24 @@ class Usage(msrest.serialization.Model): """ _validation = { - 'unit': {'readonly': True}, - 'id': {'readonly': True}, - 'current_value': {'readonly': True}, - 'limit': {'readonly': True}, - 'name': {'readonly': True}, + "unit": {"readonly": True}, + "id": {"readonly": True}, + "current_value": {"readonly": True}, + "limit": {"readonly": True}, + "name": {"readonly": True}, } _attribute_map = { - 'unit': {'key': 'unit', 'type': 'str'}, - 'id': {'key': 'id', 'type': 'str'}, - 'current_value': {'key': 'currentValue', 'type': 'int'}, - 'limit': {'key': 'limit', 'type': 'int'}, - 'name': {'key': 'name', 'type': 'UsageName'}, + "unit": {"key": "unit", "type": "str"}, + "id": {"key": "id", "type": "str"}, + "current_value": {"key": "currentValue", "type": "int"}, + "limit": {"key": "limit", "type": "int"}, + "name": {"key": "name", "type": "UsageName"}, } - def __init__( - self, - **kwargs - ): - super(Usage, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.unit = None self.id = None self.current_value = None @@ -1657,28 +1752,27 @@ def __init__( self.name = None -class UsageListResult(msrest.serialization.Model): +class UsageListResult(_serialization.Model): """The response from the List Usages operation. - :param value: Gets or sets the list of Storage Resource Usages. - :type value: list[~azure.mgmt.datalake.store.models.Usage] + :ivar value: Gets or sets the list of Storage Resource Usages. + :vartype value: list[~azure.mgmt.datalake.store.models.Usage] """ _attribute_map = { - 'value': {'key': 'value', 'type': '[Usage]'}, + "value": {"key": "value", "type": "[Usage]"}, } - def __init__( - self, - *, - value: Optional[List["Usage"]] = None, - **kwargs - ): - super(UsageListResult, self).__init__(**kwargs) + def __init__(self, *, value: Optional[List["_models.Usage"]] = None, **kwargs: Any) -> None: + """ + :keyword value: Gets or sets the list of Storage Resource Usages. + :paramtype value: list[~azure.mgmt.datalake.store.models.Usage] + """ + super().__init__(**kwargs) self.value = value -class UsageName(msrest.serialization.Model): +class UsageName(_serialization.Model): """The usage names that can be used. Variables are only populated by the server, and will be ignored when sending a request. @@ -1690,20 +1784,18 @@ class UsageName(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'localized_value': {'readonly': True}, + "value": {"readonly": True}, + "localized_value": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': 'str'}, - 'localized_value': {'key': 'localizedValue', 'type': 'str'}, + "value": {"key": "value", "type": "str"}, + "localized_value": {"key": "localizedValue", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(UsageName, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.localized_value = None @@ -1724,28 +1816,26 @@ class VirtualNetworkRule(SubResource): """ _validation = { - 'id': {'readonly': True}, - 'name': {'readonly': True}, - 'type': {'readonly': True}, - 'subnet_id': {'readonly': True}, + "id": {"readonly": True}, + "name": {"readonly": True}, + "type": {"readonly": True}, + "subnet_id": {"readonly": True}, } _attribute_map = { - 'id': {'key': 'id', 'type': 'str'}, - 'name': {'key': 'name', 'type': 'str'}, - 'type': {'key': 'type', 'type': 'str'}, - 'subnet_id': {'key': 'properties.subnetId', 'type': 'str'}, + "id": {"key": "id", "type": "str"}, + "name": {"key": "name", "type": "str"}, + "type": {"key": "type", "type": "str"}, + "subnet_id": {"key": "properties.subnetId", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(VirtualNetworkRule, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.subnet_id = None -class VirtualNetworkRuleListResult(msrest.serialization.Model): +class VirtualNetworkRuleListResult(_serialization.Model): """Data Lake Store virtual network rule list information. Variables are only populated by the server, and will be ignored when sending a request. @@ -1757,19 +1847,17 @@ class VirtualNetworkRuleListResult(msrest.serialization.Model): """ _validation = { - 'value': {'readonly': True}, - 'next_link': {'readonly': True}, + "value": {"readonly": True}, + "next_link": {"readonly": True}, } _attribute_map = { - 'value': {'key': 'value', 'type': '[VirtualNetworkRule]'}, - 'next_link': {'key': 'nextLink', 'type': 'str'}, + "value": {"key": "value", "type": "[VirtualNetworkRule]"}, + "next_link": {"key": "nextLink", "type": "str"}, } - def __init__( - self, - **kwargs - ): - super(VirtualNetworkRuleListResult, self).__init__(**kwargs) + def __init__(self, **kwargs: Any) -> None: + """ """ + super().__init__(**kwargs) self.value = None self.next_link = None diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_patch.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/__init__.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/__init__.py index ad9f979648cf..1189bb013e90 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/__init__.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/__init__.py @@ -13,11 +13,17 @@ from ._operations import Operations from ._locations_operations import LocationsOperations +from ._patch import __all__ as _patch_all +from ._patch import * # pylint: disable=unused-wildcard-import +from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'AccountsOperations', - 'FirewallRulesOperations', - 'VirtualNetworkRulesOperations', - 'TrustedIdProvidersOperations', - 'Operations', - 'LocationsOperations', + "AccountsOperations", + "FirewallRulesOperations", + "VirtualNetworkRulesOperations", + "TrustedIdProvidersOperations", + "Operations", + "LocationsOperations", ] +__all__.extend([p for p in _patch_all if p not in __all__]) +_patch_sdk() diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_accounts_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_accounts_operations.py index 6a2d9e73c5ad..3090b580fdfe 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_accounts_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_accounts_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,137 +6,442 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, cast, overload +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.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from azure.mgmt.core.polling.arm_polling import ARMPolling from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request, _format_url_section + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +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( + subscription_id: str, + *, + filter: Optional[str] = None, + top: Optional[int] = None, + skip: Optional[int] = None, + select: Optional[str] = None, + orderby: Optional[str] = None, + count: Optional[bool] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/accounts") + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + if filter is not None: + _params["$filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["$top"] = _SERIALIZER.query("top", top, "int", minimum=1) + if skip is not None: + _params["$skip"] = _SERIALIZER.query("skip", skip, "int", minimum=1) + if select is not None: + _params["$select"] = _SERIALIZER.query("select", select, "str") + if orderby is not None: + _params["$orderby"] = _SERIALIZER.query("orderby", orderby, "str") + if count is not None: + _params["$count"] = _SERIALIZER.query("count", count, "bool") + _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, + *, + filter: Optional[str] = None, + top: Optional[int] = None, + skip: Optional[int] = None, + select: Optional[str] = None, + orderby: Optional[str] = None, + count: Optional[bool] = None, + **kwargs: Any +) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + if filter is not None: + _params["$filter"] = _SERIALIZER.query("filter", filter, "str") + if top is not None: + _params["$top"] = _SERIALIZER.query("top", top, "int", minimum=1) + if skip is not None: + _params["$skip"] = _SERIALIZER.query("skip", skip, "int", minimum=1) + if select is not None: + _params["$select"] = _SERIALIZER.query("select", select, "str") + if orderby is not None: + _params["$orderby"] = _SERIALIZER.query("orderby", orderby, "str") + if count is not None: + _params["$count"] = _SERIALIZER.query("count", count, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_create_request( + resource_group_name: str, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_request(resource_group_name: str, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **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, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **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_delete_request( + resource_group_name: str, account_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +def build_enable_key_vault_request( + resource_group_name: str, account_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/enableKeyVault", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") -class AccountsOperations(object): - """AccountsOperations operations. + return HttpRequest(method="POST", url=_url, params=_params, **kwargs) - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. +def build_check_name_availability_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/checkNameAvailability", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "location": _SERIALIZER.url("location", location, "str"), + } + + _url: str = _format_url_section(_url, **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 AccountsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`accounts` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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, - filter=None, # type: Optional[str] - top=None, # type: Optional[int] - skip=None, # type: Optional[int] - select=None, # type: Optional[str] - orderby=None, # type: Optional[str] - count=None, # type: Optional[bool] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.DataLakeStoreAccountListResult"] + filter: Optional[str] = None, + top: Optional[int] = None, + skip: Optional[int] = None, + select: Optional[str] = None, + orderby: Optional[str] = None, + count: Optional[bool] = None, + **kwargs: Any + ) -> Iterable["_models.DataLakeStoreAccountBasic"]: """Lists the Data Lake Store accounts within the subscription. The response includes a link to the next page of results, if any. - :param filter: OData filter. Optional. + :param filter: OData filter. Optional. Default value is None. :type filter: str - :param top: The number of items to return. Optional. + :param top: The number of items to return. Optional. Default value is None. :type top: int - :param skip: The number of items to skip over before returning elements. Optional. + :param skip: The number of items to skip over before returning elements. Optional. Default + value is None. :type skip: int :param select: OData Select statement. Limits the properties on each entry to just those - requested, e.g. Categories?$select=CategoryName,Description. Optional. + requested, e.g. Categories?$select=CategoryName,Description. Optional. Default value is None. :type select: str :param orderby: OrderBy clause. One or more comma-separated expressions with an optional "asc" (the default) or "desc" depending on the order you'd like the values sorted, e.g. - Categories?$orderby=CategoryName desc. Optional. + Categories?$orderby=CategoryName desc. Optional. Default value is None. :type orderby: str :param count: The Boolean value of true or false to request a count of the matching resources - included with the resources in the response, e.g. Categories?$count=true. Optional. + included with the resources in the response, e.g. Categories?$count=true. Optional. Default + value is None. :type count: bool :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DataLakeStoreAccountListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either DataLakeStoreAccountBasic or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountBasic] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccountListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccountListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int', minimum=1) - if skip is not None: - query_parameters['$skip'] = self._serialize.query("skip", skip, 'int', minimum=1) - if select is not None: - query_parameters['$select'] = self._serialize.query("select", select, 'str') - if orderby is not None: - query_parameters['$orderby'] = self._serialize.query("orderby", orderby, 'str') - if count is not None: - query_parameters['$count'] = self._serialize.query("count", count, 'bool') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_request( + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + skip=skip, + select=select, + orderby=orderby, + count=count, + api_version=api_version, + template_url=self.list.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccountListResult', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccountListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -144,104 +450,119 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - list.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/accounts'} # type: ignore + return ItemPaged(get_next, extract_data) + + list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/accounts"} + @distributed_trace def list_by_resource_group( self, - resource_group_name, # type: str - filter=None, # type: Optional[str] - top=None, # type: Optional[int] - skip=None, # type: Optional[int] - select=None, # type: Optional[str] - orderby=None, # type: Optional[str] - count=None, # type: Optional[bool] - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.DataLakeStoreAccountListResult"] + resource_group_name: str, + filter: Optional[str] = None, + top: Optional[int] = None, + skip: Optional[int] = None, + select: Optional[str] = None, + orderby: Optional[str] = None, + count: Optional[bool] = None, + **kwargs: Any + ) -> Iterable["_models.DataLakeStoreAccountBasic"]: """Lists the Data Lake Store accounts within a specific resource group. The response includes a link to the next page of results, if any. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param filter: OData filter. Optional. + :param filter: OData filter. Optional. Default value is None. :type filter: str - :param top: The number of items to return. Optional. + :param top: The number of items to return. Optional. Default value is None. :type top: int - :param skip: The number of items to skip over before returning elements. Optional. + :param skip: The number of items to skip over before returning elements. Optional. Default + value is None. :type skip: int :param select: OData Select statement. Limits the properties on each entry to just those - requested, e.g. Categories?$select=CategoryName,Description. Optional. + requested, e.g. Categories?$select=CategoryName,Description. Optional. Default value is None. :type select: str :param orderby: OrderBy clause. One or more comma-separated expressions with an optional "asc" (the default) or "desc" depending on the order you'd like the values sorted, e.g. - Categories?$orderby=CategoryName desc. Optional. + Categories?$orderby=CategoryName desc. Optional. Default value is None. :type orderby: str :param count: A Boolean value of true or false to request a count of the matching resources - included with the resources in the response, e.g. Categories?$count=true. Optional. + included with the resources in the response, e.g. Categories?$count=true. Optional. Default + value is None. :type count: bool :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either DataLakeStoreAccountListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either DataLakeStoreAccountBasic or the result of + cls(response) + :rtype: + ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.DataLakeStoreAccountBasic] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccountListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccountListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_resource_group.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - if filter is not None: - query_parameters['$filter'] = self._serialize.query("filter", filter, 'str') - if top is not None: - query_parameters['$top'] = self._serialize.query("top", top, 'int', minimum=1) - if skip is not None: - query_parameters['$skip'] = self._serialize.query("skip", skip, 'int', minimum=1) - if select is not None: - query_parameters['$select'] = self._serialize.query("select", select, 'str') - if orderby is not None: - query_parameters['$orderby'] = self._serialize.query("orderby", orderby, 'str') - if count is not None: - query_parameters['$count'] = self._serialize.query("count", count, 'bool') - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_resource_group_request( + resource_group_name=resource_group_name, + subscription_id=self._config.subscription_id, + filter=filter, + top=top, + skip=skip, + select=select, + orderby=orderby, + count=count, + api_version=api_version, + template_url=self.list_by_resource_group.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccountListResult', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccountListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -250,51 +571,63 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - list_by_resource_group.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts'} # type: ignore + return ItemPaged(get_next, extract_data) + + list_by_resource_group.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts" + } def _create_initial( self, - resource_group_name, # type: str - account_name, # type: str - parameters, # type: "_models.CreateDataLakeStoreAccountParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.DataLakeStoreAccount" - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] + resource_group_name: str, + account_name: str, + parameters: Union[_models.CreateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> _models.DataLakeStoreAccount: error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._create_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateDataLakeStoreAccountParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.DataLakeStoreAccount] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateDataLakeStoreAccountParameters") + + request = build_create_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self._create_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200, 201]: @@ -302,188 +635,286 @@ def _create_initial( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - _create_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return deserialized # type: ignore + + _create_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + @overload def begin_create( self, - resource_group_name, # type: str - account_name, # type: str - parameters, # type: "_models.CreateDataLakeStoreAccountParameters" - **kwargs # type: Any - ): - # type: (...) -> LROPoller["_models.DataLakeStoreAccount"] + resource_group_name: str, + account_name: str, + parameters: _models.CreateDataLakeStoreAccountParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: """Creates the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param parameters: Parameters supplied to create the Data Lake Store account. + :param parameters: Parameters supplied to create the Data Lake Store account. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateDataLakeStoreAccountParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. - :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of cls(response) + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] :raises ~azure.core.exceptions.HttpResponseError: """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + + @overload + def begin_create( + self, + resource_group_name: str, + account_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: + """Creates the specified Data Lake Store account. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to create the Data Lake Store account. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. + :paramtype polling: bool or ~azure.core.polling.PollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_create( + self, + resource_group_name: str, + account_name: str, + parameters: Union[_models.CreateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: + """Creates the specified Data Lake Store account. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to create the Data Lake Store account. Is either a + CreateDataLakeStoreAccountParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateDataLakeStoreAccountParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. + :paramtype polling: bool or ~azure.core.polling.PollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.DataLakeStoreAccount] = 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, account_name=account_name, parameters=parameters, - cls=lambda x,y,z: x, + api_version=api_version, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) - - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) - + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = NoPolling() - else: polling_method = polling + if polling is True: + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_create.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - def get( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.DataLakeStoreAccount" + begin_create.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @distributed_trace + def get(self, resource_group_name: str, account_name: str, **kwargs: Any) -> _models.DataLakeStoreAccount: """Gets the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: DataLakeStoreAccount, or the result of cls(response) + :return: DataLakeStoreAccount or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.DataLakeStoreAccount - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.DataLakeStoreAccount] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } def _update_initial( self, - resource_group_name, # type: str - account_name, # type: str - parameters, # type: "_models.UpdateDataLakeStoreAccountParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.DataLakeStoreAccount" - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] + resource_group_name: str, + account_name: str, + parameters: Union[_models.UpdateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> _models.DataLakeStoreAccount: error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self._update_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'UpdateDataLakeStoreAccountParameters') - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.DataLakeStoreAccount] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "UpdateDataLakeStoreAccountParameters") + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self._update_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200, 201, 202]: @@ -491,124 +922,211 @@ def _update_initial( raise HttpResponseError(response=response, error_format=ARMErrorFormat) if response.status_code == 200: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 201: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if response.status_code == 202: - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - _update_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return deserialized # type: ignore + _update_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @overload def begin_update( self, - resource_group_name, # type: str - account_name, # type: str - parameters, # type: "_models.UpdateDataLakeStoreAccountParameters" - **kwargs # type: Any - ): - # type: (...) -> LROPoller["_models.DataLakeStoreAccount"] + resource_group_name: str, + account_name: str, + parameters: _models.UpdateDataLakeStoreAccountParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: """Updates the specified Data Lake Store account information. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param parameters: Parameters supplied to update the Data Lake Store account. + :param parameters: Parameters supplied to update the Data Lake Store account. Required. :type parameters: ~azure.mgmt.datalake.store.models.UpdateDataLakeStoreAccountParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. - :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of cls(response) + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] :raises ~azure.core.exceptions.HttpResponseError: """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType["_models.DataLakeStoreAccount"] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + + @overload + def begin_update( + self, + resource_group_name: str, + account_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: + """Updates the specified Data Lake Store account information. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to update the Data Lake Store account. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. + :paramtype polling: bool or ~azure.core.polling.PollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def begin_update( + self, + resource_group_name: str, + account_name: str, + parameters: Union[_models.UpdateDataLakeStoreAccountParameters, IO], + **kwargs: Any + ) -> LROPoller[_models.DataLakeStoreAccount]: + """Updates the specified Data Lake Store account information. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param parameters: Parameters supplied to update the Data Lake Store account. Is either a + UpdateDataLakeStoreAccountParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateDataLakeStoreAccountParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :keyword str continuation_token: A continuation token to restart a poller from a saved state. + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. + :paramtype polling: bool or ~azure.core.polling.PollingMethod + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. + :return: An instance of LROPoller that returns either DataLakeStoreAccount or the result of + cls(response) + :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.datalake.store.models.DataLakeStoreAccount] + :raises ~azure.core.exceptions.HttpResponseError: + """ + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.DataLakeStoreAccount] = 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, account_name=account_name, parameters=parameters, - cls=lambda x,y,z: x, + api_version=api_version, + content_type=content_type, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) - - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize('DataLakeStoreAccount', pipeline_response) - + deserialized = self._deserialize("DataLakeStoreAccount", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = NoPolling() - else: polling_method = polling + if polling is True: + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - def _delete_initial( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None - cls = kwargs.pop('cls', None) # type: ClsType[None] + begin_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + def _delete_initial( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> None: error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self._delete_initial.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self._delete_initial.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 202, 204]: @@ -618,117 +1136,123 @@ def _delete_initial( if cls: return cls(pipeline_response, None, {}) - _delete_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + _delete_initial.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } - def begin_delete( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> LROPoller[None] + @distributed_trace + def begin_delete(self, resource_group_name: str, account_name: str, **kwargs: Any) -> LROPoller[None]: """Deletes the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: True for ARMPolling, False for no polling, or a - polling object for personal polling strategy + :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this + operation to not poll, or pass in your own initialized polling object for a personal polling + strategy. :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present. + :keyword int polling_interval: Default waiting time between two polls for LRO operations if no + Retry-After header is present. :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: """ - polling = kwargs.pop('polling', True) # type: Union[bool, PollingMethod] - cls = kwargs.pop('cls', None) # type: ClsType[None] - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) ) - cont_token = kwargs.pop('continuation_token', None) # type: Optional[str] + 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( + raw_result = self._delete_initial( # type: ignore resource_group_name=resource_group_name, account_name=account_name, - cls=lambda x,y,z: x, + api_version=api_version, + cls=lambda x, y, z: x, + headers=_headers, + params=_params, **kwargs ) + kwargs.pop("error_map", None) - kwargs.pop('error_map', None) - kwargs.pop('content_type', None) - - def get_long_running_output(pipeline_response): + def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: return cls(pipeline_response, None, {}) - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - - if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments, **kwargs) - elif polling is False: polling_method = NoPolling() - else: polling_method = polling + if polling is True: + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller.from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) - else: - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) - begin_delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}'} # type: ignore + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - def enable_key_vault( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + begin_delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}" + } + + @distributed_trace + def enable_key_vault( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> None: """Attempts to enable a user managed Key Vault for encryption of the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.enable_key_vault.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_enable_key_vault_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.enable_key_vault.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.post(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: @@ -738,67 +1262,130 @@ def enable_key_vault( if cls: return cls(pipeline_response, None, {}) - enable_key_vault.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/enableKeyVault'} # type: ignore + enable_key_vault.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/enableKeyVault" + } + @overload def check_name_availability( self, - location, # type: str - parameters, # type: "_models.CheckNameAvailabilityParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.NameAvailabilityInformation" + location: str, + parameters: _models.CheckNameAvailabilityParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.NameAvailabilityInformation: """Checks whether the specified account name is available or taken. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Required. :type parameters: ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: NameAvailabilityInformation, or the result of cls(response) + :return: NameAvailabilityInformation or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def check_name_availability( + self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + ) -> _models.NameAvailabilityInformation: + """Checks whether the specified account name is available or taken. + + :param location: The resource location without whitespace. Required. + :type location: str + :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: NameAvailabilityInformation or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def check_name_availability( + self, location: str, parameters: Union[_models.CheckNameAvailabilityParameters, IO], **kwargs: Any + ) -> _models.NameAvailabilityInformation: + """Checks whether the specified account name is available or taken. + + :param location: The resource location without whitespace. Required. + :type location: str + :param parameters: Parameters supplied to check the Data Lake Store account name availability. + Is either a CheckNameAvailabilityParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CheckNameAvailabilityParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: NameAvailabilityInformation or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.NameAvailabilityInformation + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.NameAvailabilityInformation"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.check_name_availability.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CheckNameAvailabilityParameters') - body_content_kwargs['content'] = body_content - request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.NameAvailabilityInformation] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CheckNameAvailabilityParameters") + + request = build_check_name_availability_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.check_name_availability.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('NameAvailabilityInformation', pipeline_response) + deserialized = self._deserialize("NameAvailabilityInformation", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - check_name_availability.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/checkNameAvailability'} # type: ignore + + check_name_availability.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/checkNameAvailability" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_firewall_rules_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_firewall_rules_operations.py index e0c429462009..bbe50f3b96fa 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_firewall_rules_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_firewall_rules_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,108 +6,296 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload +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.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request, _format_url_section + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_account_request( + resource_group_name: str, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_create_or_update_request( + resource_group_name: str, account_name: str, firewall_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "firewallRuleName": _SERIALIZER.url("firewall_rule_name", firewall_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_request( + resource_group_name: str, account_name: str, firewall_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "firewallRuleName": _SERIALIZER.url("firewall_rule_name", firewall_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **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, account_name: str, firewall_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "firewallRuleName": _SERIALIZER.url("firewall_rule_name", firewall_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + # 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") -class FirewallRulesOperations(object): - """FirewallRulesOperations operations. + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_delete_request( + resource_group_name: str, account_name: str, firewall_rule_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "firewallRuleName": _SERIALIZER.url("firewall_rule_name", firewall_rule_name, "str"), + } - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +class FirewallRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`firewall_rules` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.FirewallRuleListResult"] + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> Iterable["_models.FirewallRule"]: """Lists the Data Lake Store firewall rules within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either FirewallRuleListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.FirewallRuleListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either FirewallRule or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.FirewallRule] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRuleListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.FirewallRuleListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('FirewallRuleListResult', pipeline_response) + deserialized = self._deserialize("FirewallRuleListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -115,267 +304,435 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules'} # type: ignore + return ItemPaged(get_next, extract_data) + + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules" + } + @overload def create_or_update( self, - resource_group_name, # type: str - account_name, # type: str - firewall_rule_name, # type: str - parameters, # type: "_models.CreateOrUpdateFirewallRuleParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.FirewallRule" + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: _models.CreateOrUpdateFirewallRuleParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: """Creates or updates the specified firewall rule. During update, the firewall rule with the specified name will be replaced with this new firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to create or update. + :param firewall_rule_name: The name of the firewall rule to create or update. Required. :type firewall_rule_name: str - :param parameters: Parameters supplied to create or update the firewall rule. + :param parameters: Parameters supplied to create or update the firewall rule. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateFirewallRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: + """Creates or updates the specified firewall rule. During update, the firewall rule with the + specified name will be replaced with this new firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to create or update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to create or update the firewall rule. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Union[_models.CreateOrUpdateFirewallRuleParameters, IO], + **kwargs: Any + ) -> _models.FirewallRule: + """Creates or updates the specified firewall rule. During update, the firewall rule with the + specified name will be replaced with this new firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to create or update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to create or update the firewall rule. Is either a + CreateOrUpdateFirewallRuleParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateFirewallRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateFirewallRuleParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.FirewallRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateFirewallRuleParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @distributed_trace def get( - self, - resource_group_name, # type: str - account_name, # type: str - firewall_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.FirewallRule" + self, resource_group_name: str, account_name: str, firewall_rule_name: str, **kwargs: Any + ) -> _models.FirewallRule: """Gets the specified Data Lake Store firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to retrieve. + :param firewall_rule_name: The name of the firewall rule to retrieve. Required. :type firewall_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.FirewallRule] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @overload def update( self, - resource_group_name, # type: str - account_name, # type: str - firewall_rule_name, # type: str - parameters=None, # type: Optional["_models.UpdateFirewallRuleParameters"] - **kwargs # type: Any - ): - # type: (...) -> "_models.FirewallRule" + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Optional[_models.UpdateFirewallRuleParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: """Updates the specified firewall rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to update. + :param firewall_rule_name: The name of the firewall rule to update. Required. :type firewall_rule_name: str - :param parameters: Parameters supplied to update the firewall rule. + :param parameters: Parameters supplied to update the firewall rule. Default value is None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateFirewallRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: FirewallRule, or the result of cls(response) + :return: FirewallRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.FirewallRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.FirewallRule: + """Updates the specified firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to update the firewall rule. Default value is None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + account_name: str, + firewall_rule_name: str, + parameters: Optional[Union[_models.UpdateFirewallRuleParameters, IO]] = None, + **kwargs: Any + ) -> _models.FirewallRule: + """Updates the specified firewall rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param firewall_rule_name: The name of the firewall rule to update. Required. + :type firewall_rule_name: str + :param parameters: Parameters supplied to update the firewall rule. Is either a + UpdateFirewallRuleParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateFirewallRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: FirewallRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.FirewallRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.FirewallRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateFirewallRuleParameters') + api_version: Literal["2016-11-01"] = 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.FirewallRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateFirewallRuleParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('FirewallRule', pipeline_response) + deserialized = self._deserialize("FirewallRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore - def delete( - self, - resource_group_name, # type: str - account_name, # type: str - firewall_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, firewall_rule_name: str, **kwargs: Any + ) -> None: """Deletes the specified firewall rule from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param firewall_rule_name: The name of the firewall rule to delete. + :param firewall_rule_name: The name of the firewall rule to delete. Required. :type firewall_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'firewallRuleName': self._serialize.url("firewall_rule_name", firewall_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + firewall_rule_name=firewall_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -385,4 +742,6 @@ def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/firewallRules/{firewallRuleName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_locations_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_locations_operations.py index 4a1679b8c521..1778b7aba36d 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_locations_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_locations_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,88 +6,158 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings +import sys +from typing import Any, Callable, Dict, Iterable, Optional, TypeVar +import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +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.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request, _format_url_section -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False -class LocationsOperations(object): - """LocationsOperations operations. - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. +def build_get_capability_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 {}) - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/capability", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "location": _SERIALIZER.url("location", location, "str"), + } + + _url: str = _format_url_section(_url, **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_get_usage_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/usages" + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "location": _SERIALIZER.url("location", location, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class LocationsOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`locations` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config - - def get_capability( - self, - location, # type: str - **kwargs # type: Any - ): - # type: (...) -> Optional["_models.CapabilityInformation"] + 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 get_capability(self, location: str, **kwargs: Any) -> Optional[_models.CapabilityInformation]: """Gets subscription-level properties and limits for Data Lake Store specified by resource location. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: CapabilityInformation, or the result of cls(response) + :return: CapabilityInformation or None or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.CapabilityInformation or None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[Optional["_models.CapabilityInformation"]] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get_capability.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[Optional[_models.CapabilityInformation]] = kwargs.pop("cls", None) + + request = build_get_capability_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get_capability.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 404]: @@ -95,73 +166,90 @@ def get_capability( deserialized = None if response.status_code == 200: - deserialized = self._deserialize('CapabilityInformation', pipeline_response) + deserialized = self._deserialize("CapabilityInformation", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get_capability.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/capability'} # type: ignore - - def get_usage( - self, - location, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.UsageListResult"] + + get_capability.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/capability" + } + + @distributed_trace + def get_usage(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: """Gets the current usage count and the limit for the resources of the location under the subscription. - :param location: The resource location without whitespace. + :param location: The resource location without whitespace. Required. :type location: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either UsageListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.UsageListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either Usage or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.Usage] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.UsageListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.UsageListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.get_usage.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'location': self._serialize.url("location", location, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_get_usage_request( + location=location, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get_usage.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('UsageListResult', pipeline_response) + deserialized = self._deserialize("UsageListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + list_of_elem = cls(list_of_elem) # type: ignore return None, iter(list_of_elem) def get_next(next_link=None): request = prepare_request(next_link) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -170,7 +258,8 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - get_usage.metadata = {'url': '/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/usages'} # type: ignore + return ItemPaged(get_next, extract_data) + + get_usage.metadata = { + "url": "/subscriptions/{subscriptionId}/providers/Microsoft.DataLakeStore/locations/{location}/usages" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_operations.py index 525307da8bba..4eaedbf31306 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,88 +6,126 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, Optional, TypeVar + +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse -from azure.core.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +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(**kwargs: Any) -> HttpRequest: + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Optional, TypeVar + # Construct URL + _url = kwargs.pop("template_url", "/providers/Microsoft.DataLakeStore/operations") - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") -class Operations(object): - """Operations operations. + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + +class Operations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`operations` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config - - def list( - self, - **kwargs # type: Any - ): - # type: (...) -> "_models.OperationListResult" + 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, **kwargs: Any) -> _models.OperationListResult: """Lists all of the available Data Lake Store REST API operations. :keyword callable cls: A custom type or function that will be passed the direct response - :return: OperationListResult, or the result of cls(response) + :return: OperationListResult or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.OperationListResult - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.OperationListResult"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct URL - url = self.list.metadata['url'] # type: ignore + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.OperationListResult] = kwargs.pop("cls", None) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + request = build_list_request( + api_version=api_version, + template_url=self.list.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('OperationListResult', pipeline_response) + deserialized = self._deserialize("OperationListResult", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - list.metadata = {'url': '/providers/Microsoft.DataLakeStore/operations'} # type: ignore + + list.metadata = {"url": "/providers/Microsoft.DataLakeStore/operations"} diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_patch.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_patch.py new file mode 100644 index 000000000000..f7dd32510333 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_patch.py @@ -0,0 +1,20 @@ +# ------------------------------------ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +# ------------------------------------ +"""Customize generated code here. + +Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize +""" +from typing import List + +__all__: List[str] = [] # Add all objects you want publicly available to users at this package level + + +def patch_sdk(): + """Do not remove from this file. + + `patch_sdk` is a last resort escape hatch that allows you to do customizations + you can't accomplish using the techniques described in + https://aka.ms/azsdk/python/dpcodegen/python/customize + """ diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_trusted_id_providers_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_trusted_id_providers_operations.py index c2f933549325..413b3f7988e9 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_trusted_id_providers_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_trusted_id_providers_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,109 +6,297 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload +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.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request, _format_url_section + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_account_request( + resource_group_name: str, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_create_or_update_request( + resource_group_name: str, account_name: str, trusted_id_provider_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "trustedIdProviderName": _SERIALIZER.url("trusted_id_provider_name", trusted_id_provider_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_request( + resource_group_name: str, account_name: str, trusted_id_provider_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "trustedIdProviderName": _SERIALIZER.url("trusted_id_provider_name", trusted_id_provider_name, "str"), + } + + _url: str = _format_url_section(_url, **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, account_name: str, trusted_id_provider_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "trustedIdProviderName": _SERIALIZER.url("trusted_id_provider_name", trusted_id_provider_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + # 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") -class TrustedIdProvidersOperations(object): - """TrustedIdProvidersOperations operations. + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_delete_request( + resource_group_name: str, account_name: str, trusted_id_provider_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "trustedIdProviderName": _SERIALIZER.url("trusted_id_provider_name", trusted_id_provider_name, "str"), + } - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +class TrustedIdProvidersOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`trusted_id_providers` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.TrustedIdProviderListResult"] + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> Iterable["_models.TrustedIdProvider"]: """Lists the Data Lake Store trusted identity providers within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either TrustedIdProviderListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.TrustedIdProviderListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either TrustedIdProvider or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.TrustedIdProvider] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProviderListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.TrustedIdProviderListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('TrustedIdProviderListResult', pipeline_response) + deserialized = self._deserialize("TrustedIdProviderListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -116,269 +305,447 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders'} # type: ignore + return ItemPaged(get_next, extract_data) + + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders" + } + @overload def create_or_update( self, - resource_group_name, # type: str - account_name, # type: str - trusted_id_provider_name, # type: str - parameters, # type: "_models.CreateOrUpdateTrustedIdProviderParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.TrustedIdProvider" + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: _models.CreateOrUpdateTrustedIdProviderParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: """Creates or updates the specified trusted identity provider. During update, the trusted identity provider with the specified name will be replaced with this new provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider. This is used for - differentiation of providers in the account. + differentiation of providers in the account. Required. :type trusted_id_provider_name: str :param parameters: Parameters supplied to create or replace the trusted identity provider. + Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateTrustedIdProviderParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Creates or updates the specified trusted identity provider. During update, the trusted identity + provider with the specified name will be replaced with this new provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to create or replace the trusted identity provider. + Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Union[_models.CreateOrUpdateTrustedIdProviderParameters, IO], + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Creates or updates the specified trusted identity provider. During update, the trusted identity + provider with the specified name will be replaced with this new provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to create or replace the trusted identity provider. Is + either a CreateOrUpdateTrustedIdProviderParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateTrustedIdProviderParameters + or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateTrustedIdProviderParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.TrustedIdProvider] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateTrustedIdProviderParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @distributed_trace def get( - self, - resource_group_name, # type: str - account_name, # type: str - trusted_id_provider_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.TrustedIdProvider" + self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, **kwargs: Any + ) -> _models.TrustedIdProvider: """Gets the specified Data Lake Store trusted identity provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider to retrieve. + Required. :type trusted_id_provider_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.TrustedIdProvider] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @overload def update( self, - resource_group_name, # type: str - account_name, # type: str - trusted_id_provider_name, # type: str - parameters=None, # type: Optional["_models.UpdateTrustedIdProviderParameters"] - **kwargs # type: Any - ): - # type: (...) -> "_models.TrustedIdProvider" + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Optional[_models.UpdateTrustedIdProviderParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: """Updates the specified trusted identity provider. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param trusted_id_provider_name: The name of the trusted identity provider. This is used for - differentiation of providers in the account. + differentiation of providers in the account. Required. :type trusted_id_provider_name: str - :param parameters: Parameters supplied to update the trusted identity provider. + :param parameters: Parameters supplied to update the trusted identity provider. Default value + is None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Updates the specified trusted identity provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to update the trusted identity provider. Default value + is None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: TrustedIdProvider or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + account_name: str, + trusted_id_provider_name: str, + parameters: Optional[Union[_models.UpdateTrustedIdProviderParameters, IO]] = None, + **kwargs: Any + ) -> _models.TrustedIdProvider: + """Updates the specified trusted identity provider. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param trusted_id_provider_name: The name of the trusted identity provider. This is used for + differentiation of providers in the account. Required. + :type trusted_id_provider_name: str + :param parameters: Parameters supplied to update the trusted identity provider. Is either a + UpdateTrustedIdProviderParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateTrustedIdProviderParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: TrustedIdProvider, or the result of cls(response) + :return: TrustedIdProvider or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.TrustedIdProvider - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.TrustedIdProvider"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateTrustedIdProviderParameters') + api_version: Literal["2016-11-01"] = 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.TrustedIdProvider] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateTrustedIdProviderParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('TrustedIdProvider', pipeline_response) + deserialized = self._deserialize("TrustedIdProvider", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore - def delete( - self, - resource_group_name, # type: str - account_name, # type: str - trusted_id_provider_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, trusted_id_provider_name: str, **kwargs: Any + ) -> None: """Deletes the specified trusted identity provider from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param trusted_id_provider_name: The name of the trusted identity provider to delete. + :param trusted_id_provider_name: The name of the trusted identity provider to delete. Required. :type trusted_id_provider_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'trustedIdProviderName': self._serialize.url("trusted_id_provider_name", trusted_id_provider_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + trusted_id_provider_name=trusted_id_provider_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -388,4 +755,6 @@ def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/trustedIdProviders/{trustedIdProviderName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_virtual_network_rules_operations.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_virtual_network_rules_operations.py index 7d69985d66ac..3225a0f0d8fa 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_virtual_network_rules_operations.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/operations/_virtual_network_rules_operations.py @@ -1,3 +1,4 @@ +# pylint: disable=too-many-lines # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -5,108 +6,296 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import TYPE_CHECKING -import warnings - -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error +import sys +from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload +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.pipeline.transport import HttpRequest, HttpResponse +from azure.core.pipeline.transport import HttpResponse +from azure.core.rest import HttpRequest +from azure.core.tracing.decorator import distributed_trace +from azure.core.utils import case_insensitive_dict from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models +from .._serialization import Serializer +from .._vendor import _convert_request, _format_url_section + +if sys.version_info >= (3, 8): + from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports +else: + from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports +T = TypeVar("T") +ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + +_SERIALIZER = Serializer() +_SERIALIZER.client_side_validation = False + + +def build_list_by_account_request( + resource_group_name: str, account_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_create_or_update_request( + resource_group_name: str, account_name: str, virtual_network_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "virtualNetworkRuleName": _SERIALIZER.url("virtual_network_rule_name", virtual_network_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore + + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + # Construct headers + if content_type is not None: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_get_request( + resource_group_name: str, account_name: str, virtual_network_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "virtualNetworkRuleName": _SERIALIZER.url("virtual_network_rule_name", virtual_network_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **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, account_name: str, virtual_network_rule_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: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") + + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "virtualNetworkRuleName": _SERIALIZER.url("virtual_network_rule_name", virtual_network_rule_name, "str"), + } + + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore -if TYPE_CHECKING: - # pylint: disable=unused-import,ungrouped-imports - from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") - T = TypeVar('T') - ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] + # 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") -class VirtualNetworkRulesOperations(object): - """VirtualNetworkRulesOperations operations. + return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) + + +def build_delete_request( + resource_group_name: str, account_name: str, virtual_network_rule_name: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop("api_version", _params.pop("api-version", "2016-11-01")) + # Construct URL + _url = kwargs.pop( + "template_url", + "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}", + ) # pylint: disable=line-too-long + path_format_arguments = { + "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, "str"), + "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, "str"), + "accountName": _SERIALIZER.url("account_name", account_name, "str"), + "virtualNetworkRuleName": _SERIALIZER.url("virtual_network_rule_name", virtual_network_rule_name, "str"), + } - You should not instantiate this class directly. Instead, you should create a Client instance that - instantiates it for you and attaches it as an attribute. + _url: str = _format_url_section(_url, **path_format_arguments) # type: ignore - :ivar models: Alias to model classes used in this operation group. - :type models: ~azure.mgmt.datalake.store.models - :param client: Client for service requests. - :param config: Configuration of service client. - :param serializer: An object model serializer. - :param deserializer: An object model deserializer. + # Construct parameters + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + + return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) + + +class VirtualNetworkRulesOperations: + """ + .. warning:: + **DO NOT** instantiate this class directly. + + Instead, you should access the following operations through + :class:`~azure.mgmt.datalake.store.DataLakeStoreAccountManagementClient`'s + :attr:`virtual_network_rules` attribute. """ models = _models - def __init__(self, client, config, serializer, deserializer): - self._client = client - self._serialize = serializer - self._deserialize = deserializer - self._config = config + 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_by_account( - self, - resource_group_name, # type: str - account_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> Iterable["_models.VirtualNetworkRuleListResult"] + self, resource_group_name: str, account_name: str, **kwargs: Any + ) -> Iterable["_models.VirtualNetworkRule"]: """Lists the Data Lake Store virtual network rules within the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: An iterator like instance of either VirtualNetworkRuleListResult or the result of cls(response) - :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.VirtualNetworkRuleListResult] - :raises: ~azure.core.exceptions.HttpResponseError + :return: An iterator like instance of either VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.datalake.store.models.VirtualNetworkRule] + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRuleListResult"] + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) + + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.VirtualNetworkRuleListResult] = kwargs.pop("cls", None) + error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" + error_map.update(kwargs.pop("error_map", {}) or {}) def prepare_request(next_link=None): - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - if not next_link: - # Construct URL - url = self.list_by_account.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - request = self._client.get(url, query_parameters, header_parameters) + + request = build_list_by_account_request( + resource_group_name=resource_group_name, + account_name=account_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.list_by_account.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + else: - url = next_link - query_parameters = {} # type: Dict[str, Any] - request = self._client.get(url, query_parameters, header_parameters) + # make call to next link with the client's api-version + _parsed_next_link = urllib.parse.urlparse(next_link) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) + _next_request_params["api-version"] = self._config.api_version + request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + request.method = "GET" return request def extract_data(pipeline_response): - deserialized = self._deserialize('VirtualNetworkRuleListResult', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRuleListResult", pipeline_response) list_of_elem = deserialized.value if cls: - list_of_elem = cls(list_of_elem) + 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) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) response = pipeline_response.http_response if response.status_code not in [200]: @@ -115,267 +304,441 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged( - get_next, extract_data - ) - list_by_account.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules'} # type: ignore + return ItemPaged(get_next, extract_data) + list_by_account.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules" + } + + @overload def create_or_update( self, - resource_group_name, # type: str - account_name, # type: str - virtual_network_rule_name, # type: str - parameters, # type: "_models.CreateOrUpdateVirtualNetworkRuleParameters" - **kwargs # type: Any - ): - # type: (...) -> "_models.VirtualNetworkRule" + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: _models.CreateOrUpdateVirtualNetworkRuleParameters, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: """Creates or updates the specified virtual network rule. During update, the virtual network rule with the specified name will be replaced with this new virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. :type virtual_network_rule_name: str - :param parameters: Parameters supplied to create or update the virtual network rule. + :param parameters: Parameters supplied to create or update the virtual network rule. Required. :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateVirtualNetworkRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def create_or_update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: IO, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Creates or updates the specified virtual network rule. During update, the virtual network rule + with the specified name will be replaced with this new virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to create or update the virtual network rule. Required. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def create_or_update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Union[_models.CreateOrUpdateVirtualNetworkRuleParameters, IO], + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Creates or updates the specified virtual network rule. During update, the virtual network rule + with the specified name will be replaced with this new virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to create or update. + Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to create or update the virtual network rule. Is either + a CreateOrUpdateVirtualNetworkRuleParameters type or a IO type. Required. + :type parameters: ~azure.mgmt.datalake.store.models.CreateOrUpdateVirtualNetworkRuleParameters + or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.create_or_update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') - - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') - - body_content_kwargs = {} # type: Dict[str, Any] - body_content = self._serialize.body(parameters, 'CreateOrUpdateVirtualNetworkRuleParameters') - body_content_kwargs['content'] = body_content - request = self._client.put(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + 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: Literal["2016-11-01"] = 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.VirtualNetworkRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters + else: + _json = self._serialize.body(parameters, "CreateOrUpdateVirtualNetworkRuleParameters") + + request = build_create_or_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.create_or_update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + create_or_update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @distributed_trace def get( - self, - resource_group_name, # type: str - account_name, # type: str - virtual_network_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> "_models.VirtualNetworkRule" + self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, **kwargs: Any + ) -> _models.VirtualNetworkRule: """Gets the specified Data Lake Store virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to retrieve. + :param virtual_network_rule_name: The name of the virtual network rule to retrieve. Required. :type virtual_network_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - accept = "application/json" - - # Construct URL - url = self.get.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[_models.VirtualNetworkRule] = kwargs.pop("cls", None) + + request = build_get_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.get.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.get(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + get.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @overload def update( self, - resource_group_name, # type: str - account_name, # type: str - virtual_network_rule_name, # type: str - parameters=None, # type: Optional["_models.UpdateVirtualNetworkRuleParameters"] - **kwargs # type: Any - ): - # type: (...) -> "_models.VirtualNetworkRule" + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Optional[_models.UpdateVirtualNetworkRuleParameters] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: """Updates the specified virtual network rule. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to update. + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. :type virtual_network_rule_name: str - :param parameters: Parameters supplied to update the virtual network rule. + :param parameters: Parameters supplied to update the virtual network rule. Default value is + None. :type parameters: ~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleParameters + :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. + Default value is "application/json". + :paramtype content_type: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: VirtualNetworkRule, or the result of cls(response) + :return: VirtualNetworkRule or the result of cls(response) :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @overload + def update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Optional[IO] = None, + *, + content_type: str = "application/json", + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Updates the specified virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to update the virtual network rule. Default value is + None. + :type parameters: IO + :keyword content_type: Body Parameter content-type. Content type parameter for binary body. + Default value is "application/json". + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: + """ + + @distributed_trace + def update( + self, + resource_group_name: str, + account_name: str, + virtual_network_rule_name: str, + parameters: Optional[Union[_models.UpdateVirtualNetworkRuleParameters, IO]] = None, + **kwargs: Any + ) -> _models.VirtualNetworkRule: + """Updates the specified virtual network rule. + + :param resource_group_name: The name of the Azure resource group. Required. + :type resource_group_name: str + :param account_name: The name of the Data Lake Store account. Required. + :type account_name: str + :param virtual_network_rule_name: The name of the virtual network rule to update. Required. + :type virtual_network_rule_name: str + :param parameters: Parameters supplied to update the virtual network rule. Is either a + UpdateVirtualNetworkRuleParameters type or a IO type. Default value is None. + :type parameters: ~azure.mgmt.datalake.store.models.UpdateVirtualNetworkRuleParameters or IO + :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + Default value is None. + :paramtype content_type: str + :keyword callable cls: A custom type or function that will be passed the direct response + :return: VirtualNetworkRule or the result of cls(response) + :rtype: ~azure.mgmt.datalake.store.models.VirtualNetworkRule + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType["_models.VirtualNetworkRule"] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - content_type = kwargs.pop("content_type", "application/json") - accept = "application/json" - - # Construct URL - url = self.update.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), - } - url = self._client.format_url(url, **path_format_arguments) - - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + error_map.update(kwargs.pop("error_map", {}) or {}) - # Construct headers - header_parameters = {} # type: Dict[str, Any] - header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str') - header_parameters['Accept'] = self._serialize.header("accept", accept, 'str') + _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - body_content_kwargs = {} # type: Dict[str, Any] - if parameters is not None: - body_content = self._serialize.body(parameters, 'UpdateVirtualNetworkRuleParameters') + api_version: Literal["2016-11-01"] = 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.VirtualNetworkRule] = kwargs.pop("cls", None) + + content_type = content_type or "application/json" + _json = None + _content = None + if isinstance(parameters, (IO, bytes)): + _content = parameters else: - body_content = None - body_content_kwargs['content'] = body_content - request = self._client.patch(url, query_parameters, header_parameters, **body_content_kwargs) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) + if parameters is not None: + _json = self._serialize.body(parameters, "UpdateVirtualNetworkRuleParameters") + else: + _json = None + + request = build_update_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + content_type=content_type, + json=_json, + content=_content, + template_url=self.update.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) + + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) + response = pipeline_response.http_response if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response, error_format=ARMErrorFormat) - deserialized = self._deserialize('VirtualNetworkRule', pipeline_response) + deserialized = self._deserialize("VirtualNetworkRule", pipeline_response) if cls: return cls(pipeline_response, deserialized, {}) return deserialized - update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore - def delete( - self, - resource_group_name, # type: str - account_name, # type: str - virtual_network_rule_name, # type: str - **kwargs # type: Any - ): - # type: (...) -> None + update.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } + + @distributed_trace + def delete( # pylint: disable=inconsistent-return-statements + self, resource_group_name: str, account_name: str, virtual_network_rule_name: str, **kwargs: Any + ) -> None: """Deletes the specified virtual network rule from the specified Data Lake Store account. - :param resource_group_name: The name of the Azure resource group. + :param resource_group_name: The name of the Azure resource group. Required. :type resource_group_name: str - :param account_name: The name of the Data Lake Store account. + :param account_name: The name of the Data Lake Store account. Required. :type account_name: str - :param virtual_network_rule_name: The name of the virtual network rule to delete. + :param virtual_network_rule_name: The name of the virtual network rule to delete. Required. :type virtual_network_rule_name: str :keyword callable cls: A custom type or function that will be passed the direct response - :return: None, or the result of cls(response) + :return: None or the result of cls(response) :rtype: None - :raises: ~azure.core.exceptions.HttpResponseError + :raises ~azure.core.exceptions.HttpResponseError: """ - cls = kwargs.pop('cls', None) # type: ClsType[None] error_map = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError - } - error_map.update(kwargs.pop('error_map', {})) - api_version = "2016-11-01" - - # Construct URL - url = self.delete.metadata['url'] # type: ignore - path_format_arguments = { - 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'), - 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'), - 'accountName': self._serialize.url("account_name", account_name, 'str'), - 'virtualNetworkRuleName': self._serialize.url("virtual_network_rule_name", virtual_network_rule_name, 'str'), + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - url = self._client.format_url(url, **path_format_arguments) + error_map.update(kwargs.pop("error_map", {}) or {}) + + _headers = kwargs.pop("headers", {}) or {} + _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - # Construct parameters - query_parameters = {} # type: Dict[str, Any] - query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str') + api_version: Literal["2016-11-01"] = kwargs.pop( + "api_version", _params.pop("api-version", self._config.api_version) + ) + cls: ClsType[None] = kwargs.pop("cls", None) + + request = build_delete_request( + resource_group_name=resource_group_name, + account_name=account_name, + virtual_network_rule_name=virtual_network_rule_name, + subscription_id=self._config.subscription_id, + api_version=api_version, + template_url=self.delete.metadata["url"], + headers=_headers, + params=_params, + ) + request = _convert_request(request) + request.url = self._client.format_url(request.url) - # Construct headers - header_parameters = {} # type: Dict[str, Any] + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + request, stream=False, **kwargs + ) - request = self._client.delete(url, query_parameters, header_parameters) - pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs) response = pipeline_response.http_response if response.status_code not in [200, 204]: @@ -385,4 +748,6 @@ def delete( if cls: return cls(pipeline_response, None, {}) - delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}'} # type: ignore + delete.metadata = { + "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DataLakeStore/accounts/{accountName}/virtualNetworkRules/{virtualNetworkRuleName}" + } diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_check_name_availability.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_check_name_availability.py new file mode 100644 index 000000000000..36a66f396e45 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_check_name_availability.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_check_name_availability.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.check_name_availability( + location="EastUS2", + parameters={"name": "contosoadla", "type": "Microsoft.DataLakeStore/accounts"}, + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_CheckNameAvailability.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_create.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_create.py new file mode 100644 index 000000000000..87ae55332181 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_create.py @@ -0,0 +1,71 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_create.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.begin_create( + resource_group_name="contosorg", + account_name="contosoadla", + parameters={ + "identity": {"type": "SystemAssigned"}, + "location": "eastus2", + "properties": { + "defaultGroup": "test_default_group", + "encryptionConfig": { + "keyVaultMetaInfo": { + "encryptionKeyName": "test_encryption_key_name", + "encryptionKeyVersion": "encryption_key_version", + "keyVaultResourceId": "34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + }, + "type": "UserManaged", + }, + "encryptionState": "Enabled", + "firewallAllowAzureIps": "Enabled", + "firewallRules": [ + {"name": "test_rule", "properties": {"endIpAddress": "2.2.2.2", "startIpAddress": "1.1.1.1"}} + ], + "firewallState": "Enabled", + "newTier": "Consumption", + "trustedIdProviderState": "Enabled", + "trustedIdProviders": [ + { + "name": "test_trusted_id_provider_name", + "properties": {"idProvider": "https://sts.windows.net/ea9ec534-a3e3-4e45-ad36-3afc5bb291c1"}, + } + ], + }, + "tags": {"test_key": "test_value"}, + }, + ).result() + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_Create.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_delete.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_delete.py new file mode 100644 index 000000000000..ba57e67aeca1 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_delete.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_delete.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.begin_delete( + resource_group_name="contosorg", + account_name="contosoadla", + ).result() + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_Delete.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_enable_key_vault.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_enable_key_vault.py new file mode 100644 index 000000000000..3d1471978bc5 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_enable_key_vault.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_enable_key_vault.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.enable_key_vault( + resource_group_name="contosorg", + account_name="contosoadla", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_EnableKeyVault.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_get.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_get.py new file mode 100644 index 000000000000..7380eae00a03 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_get.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_get.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.get( + resource_group_name="contosorg", + account_name="contosoadla", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_Get.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list.py new file mode 100644 index 000000000000..70c2f18ffd4d --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list.py @@ -0,0 +1,39 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.list() + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_List.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list_by_resource_group.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list_by_resource_group.py new file mode 100644 index 000000000000..ad043afa1b3f --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_list_by_resource_group.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_list_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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.list_by_resource_group( + resource_group_name="contosorg", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_ListByResourceGroup.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_update.py new file mode 100644 index 000000000000..693ff0ef200f --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/accounts_update.py @@ -0,0 +1,52 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python accounts_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.accounts.begin_update( + resource_group_name="contosorg", + account_name="contosoadla", + parameters={ + "properties": { + "defaultGroup": "test_default_group", + "encryptionConfig": {"keyVaultMetaInfo": {"encryptionKeyVersion": "encryption_key_version"}}, + "firewallAllowAzureIps": "Enabled", + "firewallState": "Enabled", + "newTier": "Consumption", + "trustedIdProviderState": "Enabled", + }, + "tags": {"test_key": "test_value"}, + }, + ).result() + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Accounts_Update.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_create_or_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_create_or_update.py new file mode 100644 index 000000000000..92d68deeb7d8 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_create_or_update.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python firewall_rules_create_or_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.firewall_rules.create_or_update( + resource_group_name="contosorg", + account_name="contosoadla", + firewall_rule_name="test_rule", + parameters={"properties": {"endIpAddress": "2.2.2.2", "startIpAddress": "1.1.1.1"}}, + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/FirewallRules_CreateOrUpdate.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_delete.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_delete.py new file mode 100644 index 000000000000..c1873b37f7df --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_delete.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python firewall_rules_delete.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.firewall_rules.delete( + resource_group_name="contosorg", + account_name="contosoadla", + firewall_rule_name="test_rule", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/FirewallRules_Delete.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_get.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_get.py new file mode 100644 index 000000000000..6da8491436da --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_get.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python firewall_rules_get.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.firewall_rules.get( + resource_group_name="contosorg", + account_name="contosoadla", + firewall_rule_name="test_rule", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/FirewallRules_Get.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_list_by_account.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_list_by_account.py new file mode 100644 index 000000000000..2f51c4e876d6 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_list_by_account.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python firewall_rules_list_by_account.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.firewall_rules.list_by_account( + resource_group_name="contosorg", + account_name="contosoadla", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/FirewallRules_ListByAccount.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_update.py new file mode 100644 index 000000000000..6a4ce2878bed --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/firewall_rules_update.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python firewall_rules_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.firewall_rules.update( + resource_group_name="contosorg", + account_name="contosoadla", + firewall_rule_name="test_rule", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/FirewallRules_Update.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_capability.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_capability.py new file mode 100644 index 000000000000..3e9e83f271f6 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_capability.py @@ -0,0 +1,40 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python locations_get_capability.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.locations.get_capability( + location="EastUS2", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Locations_GetCapability.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_usage.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_usage.py new file mode 100644 index 000000000000..5ea3baef2308 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/locations_get_usage.py @@ -0,0 +1,41 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python locations_get_usage.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="{subscription-id}", + ) + + response = client.locations.get_usage( + location="WestUS", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Locations_GetUsage.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/operations_list.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/operations_list.py new file mode 100644 index 000000000000..dcabd7c974e8 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/operations_list.py @@ -0,0 +1,38 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# 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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="SUBSCRIPTION_ID", + ) + + response = client.operations.list() + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/Operations_List.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_create_or_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_create_or_update.py new file mode 100644 index 000000000000..b400398714db --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_create_or_update.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python trusted_id_providers_create_or_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.trusted_id_providers.create_or_update( + resource_group_name="contosorg", + account_name="contosoadla", + trusted_id_provider_name="test_trusted_id_provider_name", + parameters={"properties": {"idProvider": "https://sts.windows.net/ea9ec534-a3e3-4e45-ad36-3afc5bb291c1"}}, + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/TrustedIdProviders_CreateOrUpdate.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_delete.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_delete.py new file mode 100644 index 000000000000..c3175d0a28d8 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_delete.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python trusted_id_providers_delete.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.trusted_id_providers.delete( + resource_group_name="contosorg", + account_name="contosoadla", + trusted_id_provider_name="test_trusted_id_provider_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/TrustedIdProviders_Delete.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_get.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_get.py new file mode 100644 index 000000000000..86bb522bf0e3 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_get.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python trusted_id_providers_get.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.trusted_id_providers.get( + resource_group_name="contosorg", + account_name="contosoadla", + trusted_id_provider_name="test_trusted_id_provider_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/TrustedIdProviders_Get.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_list_by_account.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_list_by_account.py new file mode 100644 index 000000000000..4549742b043c --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_list_by_account.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python trusted_id_providers_list_by_account.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.trusted_id_providers.list_by_account( + resource_group_name="contosorg", + account_name="contosoadla", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/TrustedIdProviders_ListByAccount.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_update.py new file mode 100644 index 000000000000..1fda5357e6ed --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/trusted_id_providers_update.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python trusted_id_providers_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.trusted_id_providers.update( + resource_group_name="contosorg", + account_name="contosoadla", + trusted_id_provider_name="test_trusted_id_provider_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/TrustedIdProviders_Update.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_create_or_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_create_or_update.py new file mode 100644 index 000000000000..b2dc7f7bbb50 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_create_or_update.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python virtual_network_rules_create_or_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.virtual_network_rules.create_or_update( + resource_group_name="contosorg", + account_name="contosoadla", + virtual_network_rule_name="test_virtual_network_rules_name", + parameters={"properties": {"subnetId": "test_subnetId"}}, + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/VirtualNetworkRules_CreateOrUpdate.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_delete.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_delete.py new file mode 100644 index 000000000000..0485c5280b3d --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_delete.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python virtual_network_rules_delete.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.virtual_network_rules.delete( + resource_group_name="contosorg", + account_name="contosoadla", + virtual_network_rule_name="test_virtual_network_rules_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/VirtualNetworkRules_Delete.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_get.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_get.py new file mode 100644 index 000000000000..2f3ba1f2c207 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_get.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python virtual_network_rules_get.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.virtual_network_rules.get( + resource_group_name="contosorg", + account_name="contosoadla", + virtual_network_rule_name="test_virtual_network_rules_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/VirtualNetworkRules_Get.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_list_by_account.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_list_by_account.py new file mode 100644 index 000000000000..a542d1913ab5 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_list_by_account.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python virtual_network_rules_list_by_account.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.virtual_network_rules.list_by_account( + resource_group_name="contosorg", + account_name="contosoadla", + ) + for item in response: + print(item) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/VirtualNetworkRules_ListByAccount.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_update.py b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_update.py new file mode 100644 index 000000000000..b5b46090a0a7 --- /dev/null +++ b/sdk/datalake/azure-mgmt-datalake-store/generated_samples/virtual_network_rules_update.py @@ -0,0 +1,42 @@ +# 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.datalake.store import DataLakeStoreAccountManagementClient + +""" +# PREREQUISITES + pip install azure-identity + pip install azure-mgmt-datalake-store +# USAGE + python virtual_network_rules_update.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 = DataLakeStoreAccountManagementClient( + credential=DefaultAzureCredential(), + subscription_id="34adfa4f-cedf-4dc0-ba29-b6d1a69ab345", + ) + + response = client.virtual_network_rules.update( + resource_group_name="contosorg", + account_name="contosoadla", + virtual_network_rule_name="test_virtual_network_rules_name", + ) + print(response) + + +# x-ms-original-file: specification/datalake-store/resource-manager/Microsoft.DataLakeStore/stable/2016-11-01/examples/VirtualNetworkRules_Update.json +if __name__ == "__main__": + main() diff --git a/sdk/datalake/azure-mgmt-datalake-store/setup.py b/sdk/datalake/azure-mgmt-datalake-store/setup.py index f92f80361f11..27f92d4cc064 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/setup.py +++ b/sdk/datalake/azure-mgmt-datalake-store/setup.py @@ -55,6 +55,7 @@ 'Programming Language :: Python :: 3.8', 'Programming Language :: Python :: 3.9', 'Programming Language :: Python :: 3.10', + 'Programming Language :: Python :: 3.11', 'License :: OSI Approved :: MIT License', ], zip_safe=False, @@ -70,9 +71,10 @@ 'pytyped': ['py.typed'], }, install_requires=[ - 'msrest>=0.7.1', - 'azure-common~=1.1', - 'azure-mgmt-core>=1.3.2,<2.0.0', + "msrest>=0.7.1", + "azure-common~=1.1", + "azure-mgmt-core>=1.3.2,<2.0.0", + "typing-extensions>=4.3.0; python_version<'3.8.0'", ], python_requires=">=3.7" ) diff --git a/shared_requirements.txt b/shared_requirements.txt index ffcc98217b0c..bed78ab6b80d 100644 --- a/shared_requirements.txt +++ b/shared_requirements.txt @@ -184,6 +184,7 @@ yarl<2.0,>=1.0 #override azure-mgmt-powerbidedicated msrest>=0.7.1 #override azure-mgmt-eventhub typing-extensions>=4.3.0; python_version<'3.8.0' #override azure-mgmt-nginx typing-extensions>=4.3.0; python_version<'3.8.0' +#override azure-mgmt-datalake-store typing-extensions>=4.3.0; python_version<'3.8.0' #override azure-mgmt-portal msrest>=0.7.1 #override azure-search-documents azure-core<2.0.0,>=1.24.0 #override azure-mgmt-managementgroups typing-extensions>=4.3.0; python_version<'3.8.0' From 95b0798b0e4204b0b515254ec76c0b9aa64fbc24 Mon Sep 17 00:00:00 2001 From: Yuchao Yan Date: Thu, 16 Feb 2023 16:47:52 +0800 Subject: [PATCH 2/3] Update CHANGELOG.md --- sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md b/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md index e965dda625ea..60f16e20aec2 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md +++ b/sdk/datalake/azure-mgmt-datalake-store/CHANGELOG.md @@ -1,6 +1,6 @@ # Release History -## 1.1.0 (2023-02-16) +## 1.1.0b1 (2023-02-16) ### Features Added From af5a765f23fad964b5804b285659fd857880550b Mon Sep 17 00:00:00 2001 From: Yuchao Yan Date: Thu, 16 Feb 2023 16:48:25 +0800 Subject: [PATCH 3/3] Update _version.py --- .../azure/mgmt/datalake/store/_version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py index 59deb8c7263b..653b73a4a199 100644 --- a/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py +++ b/sdk/datalake/azure-mgmt-datalake-store/azure/mgmt/datalake/store/_version.py @@ -6,4 +6,4 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -VERSION = "1.1.0" +VERSION = "1.1.0b1"