CodeGen from PR 20717 in Azure/azure-rest-api-specs

Merge 78d59478564b815cc41b223b3ad4e521ae121f37 into 7f08477115a1c32910bc0eda047fe0471285a392

sdk/databox/azure-mgmt-databox/_meta.json

@@ -0,0 +1,11 @@
+{
+  "autorest": "3.8.4",
+  "use": [
+    "@autorest/python@6.1.5",
+    "@autorest/modelerfour@4.23.5"
+  ],
+  "commit": "96ee8589906e33a5068de351671ba21757d49bc4",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "autorest_command": "autorest specification/databox/resource-manager/readme.md --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/python@6.1.5 --use=@autorest/modelerfour@4.23.5 --version=3.8.4 --version-tolerant=False",
+  "readme": "specification/databox/resource-manager/readme.md"
+}

sdk/databox/azure-mgmt-databox/azure/mgmt/databox/__init__.py

@@ -14,3 +14,7 @@
     patch_sdk()
 except ImportError:
     pass
+
+from ._version import VERSION
+
+__version__ = VERSION

sdk/databox/azure-mgmt-databox/azure/mgmt/databox/_configuration.py

@@ -8,31 +8,34 @@
 # Changes may cause incorrect behavior and will be lost if the code is
 # regenerated.
 # --------------------------------------------------------------------------
-from typing import Any
+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 TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from azure.core.credentials import TokenCredential
 
 class DataBoxManagementClientConfiguration(Configuration):
     """Configuration for DataBoxManagementClient.
 
     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: The Subscription Id.
+    :param subscription_id: The Subscription Id. Required.
     :type subscription_id: str
     """
 
     def __init__(
         self,
-        credential,  # type: "TokenCredential"
-        subscription_id,  # type: str
+        credential: "TokenCredential",
+        subscription_id: str,
         **kwargs  # type: Any
     ):
         # type: (...) -> None
@@ -63,4 +66,4 @@ def _configure(
         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)

sdk/databox/azure-mgmt-databox/azure/mgmt/databox/_data_box_management_client.py

@@ -9,12 +9,19 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-from azure.mgmt.core import ARMPipelineClient
-from msrest import Serializer, Deserializer
+from typing import Any, Optional, TYPE_CHECKING
 
+from azure.mgmt.core import ARMPipelineClient
 from azure.profiles import KnownProfiles, ProfileDefinition
 from azure.profiles.multiapiclient import MultiApiClientMixin
+
 from ._configuration import DataBoxManagementClientConfiguration
+from ._operations_mixin import DataBoxManagementClientOperationsMixin
+from ._serialization import Deserializer, Serializer
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from azure.core.credentials import TokenCredential
 
 class _SDKClient(object):
     def __init__(self, *args, **kwargs):
@@ -23,7 +30,7 @@ def __init__(self, *args, **kwargs):
         """
         pass
 
-class DataBoxManagementClient(MultiApiClientMixin, _SDKClient):
+class DataBoxManagementClient(DataBoxManagementClientOperationsMixin, MultiApiClientMixin, _SDKClient):
     """The DataBox Client.
 
     This ready contains multiple API versions, to help you deal with all of the Azure clouds
@@ -34,19 +41,20 @@ class DataBoxManagementClient(MultiApiClientMixin, _SDKClient):
     The api-version parameter sets the default API version if the operation
     group is not described in the profile.
 
-    :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: The Subscription Id.
+    :param subscription_id: The Subscription Id. Required.
     :type subscription_id: str
-    :param str api_version: API version to use if no profile is provided, or if
-     missing in profile.
-    :param str base_url: Service URL
+    :param api_version: API version to use if no profile is provided, or if missing in profile.
+    :type api_version: str
+    :param base_url: Service URL
+    :type base_url: str
     :param profile: A profile definition, from KnownProfiles to dict.
     :type profile: azure.profiles.KnownProfiles
     :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
     """
 
-    DEFAULT_API_VERSION = '2020-11-01'
+    DEFAULT_API_VERSION = '2022-09-01'
     _PROFILE_TAG = "azure.mgmt.databox.DataBoxManagementClient"
     LATEST_PROFILE = ProfileDefinition({
         _PROFILE_TAG: {
@@ -57,15 +65,13 @@ class DataBoxManagementClient(MultiApiClientMixin, _SDKClient):
 
     def __init__(
         self,
-        credential,  # type: "TokenCredential"
-        subscription_id,  # type: str
-        api_version=None,
-        base_url=None,
-        profile=KnownProfiles.default,
+        credential: "TokenCredential",
+        subscription_id: str,
+        api_version=None, # type: Optional[str]
+        base_url: str = "https://management.azure.com",
+        profile=KnownProfiles.default, # type: KnownProfiles
         **kwargs  # type: Any
     ):
-        if not base_url:
-            base_url = 'https://management.azure.com'
         self._config = DataBoxManagementClientConfiguration(credential, subscription_id, **kwargs)
         self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs)
         super(DataBoxManagementClient, self).__init__(
@@ -85,6 +91,12 @@ def models(cls, api_version=DEFAULT_API_VERSION):
            * 2019-09-01: :mod:`v2019_09_01.models<azure.mgmt.databox.v2019_09_01.models>`
            * 2020-04-01: :mod:`v2020_04_01.models<azure.mgmt.databox.v2020_04_01.models>`
            * 2020-11-01: :mod:`v2020_11_01.models<azure.mgmt.databox.v2020_11_01.models>`
+           * 2021-03-01: :mod:`v2021_03_01.models<azure.mgmt.databox.v2021_03_01.models>`
+           * 2021-05-01: :mod:`v2021_05_01.models<azure.mgmt.databox.v2021_05_01.models>`
+           * 2021-08-01-preview: :mod:`v2021_08_01_preview.models<azure.mgmt.databox.v2021_08_01_preview.models>`
+           * 2021-12-01: :mod:`v2021_12_01.models<azure.mgmt.databox.v2021_12_01.models>`
+           * 2022-02-01: :mod:`v2022_02_01.models<azure.mgmt.databox.v2022_02_01.models>`
+           * 2022-09-01: :mod:`v2022_09_01.models<azure.mgmt.databox.v2022_09_01.models>`
         """
         if api_version == '2018-01-01':
             from .v2018_01_01 import models
@@ -98,6 +110,24 @@ def models(cls, api_version=DEFAULT_API_VERSION):
         elif api_version == '2020-11-01':
             from .v2020_11_01 import models
             return models
+        elif api_version == '2021-03-01':
+            from .v2021_03_01 import models
+            return models
+        elif api_version == '2021-05-01':
+            from .v2021_05_01 import models
+            return models
+        elif api_version == '2021-08-01-preview':
+            from .v2021_08_01_preview import models
+            return models
+        elif api_version == '2021-12-01':
+            from .v2021_12_01 import models
+            return models
+        elif api_version == '2022-02-01':
+            from .v2022_02_01 import models
+            return models
+        elif api_version == '2022-09-01':
+            from .v2022_09_01 import models
+            return models
         raise ValueError("API version {} is not available".format(api_version))
 
     @property
@@ -108,6 +138,12 @@ def jobs(self):
            * 2019-09-01: :class:`JobsOperations<azure.mgmt.databox.v2019_09_01.operations.JobsOperations>`
            * 2020-04-01: :class:`JobsOperations<azure.mgmt.databox.v2020_04_01.operations.JobsOperations>`
            * 2020-11-01: :class:`JobsOperations<azure.mgmt.databox.v2020_11_01.operations.JobsOperations>`
+           * 2021-03-01: :class:`JobsOperations<azure.mgmt.databox.v2021_03_01.operations.JobsOperations>`
+           * 2021-05-01: :class:`JobsOperations<azure.mgmt.databox.v2021_05_01.operations.JobsOperations>`
+           * 2021-08-01-preview: :class:`JobsOperations<azure.mgmt.databox.v2021_08_01_preview.operations.JobsOperations>`
+           * 2021-12-01: :class:`JobsOperations<azure.mgmt.databox.v2021_12_01.operations.JobsOperations>`
+           * 2022-02-01: :class:`JobsOperations<azure.mgmt.databox.v2022_02_01.operations.JobsOperations>`
+           * 2022-09-01: :class:`JobsOperations<azure.mgmt.databox.v2022_09_01.operations.JobsOperations>`
         """
         api_version = self._get_api_version('jobs')
         if api_version == '2018-01-01':
@@ -118,6 +154,18 @@ def jobs(self):
             from .v2020_04_01.operations import JobsOperations as OperationClass
         elif api_version == '2020-11-01':
             from .v2020_11_01.operations import JobsOperations as OperationClass
+        elif api_version == '2021-03-01':
+            from .v2021_03_01.operations import JobsOperations as OperationClass
+        elif api_version == '2021-05-01':
+            from .v2021_05_01.operations import JobsOperations as OperationClass
+        elif api_version == '2021-08-01-preview':
+            from .v2021_08_01_preview.operations import JobsOperations as OperationClass
+        elif api_version == '2021-12-01':
+            from .v2021_12_01.operations import JobsOperations as OperationClass
+        elif api_version == '2022-02-01':
+            from .v2022_02_01.operations import JobsOperations as OperationClass
+        elif api_version == '2022-09-01':
+            from .v2022_09_01.operations import JobsOperations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'jobs'".format(api_version))
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))
@@ -130,6 +178,12 @@ def operations(self):
            * 2019-09-01: :class:`Operations<azure.mgmt.databox.v2019_09_01.operations.Operations>`
            * 2020-04-01: :class:`Operations<azure.mgmt.databox.v2020_04_01.operations.Operations>`
            * 2020-11-01: :class:`Operations<azure.mgmt.databox.v2020_11_01.operations.Operations>`
+           * 2021-03-01: :class:`Operations<azure.mgmt.databox.v2021_03_01.operations.Operations>`
+           * 2021-05-01: :class:`Operations<azure.mgmt.databox.v2021_05_01.operations.Operations>`
+           * 2021-08-01-preview: :class:`Operations<azure.mgmt.databox.v2021_08_01_preview.operations.Operations>`
+           * 2021-12-01: :class:`Operations<azure.mgmt.databox.v2021_12_01.operations.Operations>`
+           * 2022-02-01: :class:`Operations<azure.mgmt.databox.v2022_02_01.operations.Operations>`
+           * 2022-09-01: :class:`Operations<azure.mgmt.databox.v2022_09_01.operations.Operations>`
         """
         api_version = self._get_api_version('operations')
         if api_version == '2018-01-01':
@@ -140,6 +194,18 @@ def operations(self):
             from .v2020_04_01.operations import Operations as OperationClass
         elif api_version == '2020-11-01':
             from .v2020_11_01.operations import Operations as OperationClass
+        elif api_version == '2021-03-01':
+            from .v2021_03_01.operations import Operations as OperationClass
+        elif api_version == '2021-05-01':
+            from .v2021_05_01.operations import Operations as OperationClass
+        elif api_version == '2021-08-01-preview':
+            from .v2021_08_01_preview.operations import Operations as OperationClass
+        elif api_version == '2021-12-01':
+            from .v2021_12_01.operations import Operations as OperationClass
+        elif api_version == '2022-02-01':
+            from .v2022_02_01.operations import Operations as OperationClass
+        elif api_version == '2022-09-01':
+            from .v2022_09_01.operations import Operations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'operations'".format(api_version))
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))
@@ -152,6 +218,12 @@ def service(self):
            * 2019-09-01: :class:`ServiceOperations<azure.mgmt.databox.v2019_09_01.operations.ServiceOperations>`
            * 2020-04-01: :class:`ServiceOperations<azure.mgmt.databox.v2020_04_01.operations.ServiceOperations>`
            * 2020-11-01: :class:`ServiceOperations<azure.mgmt.databox.v2020_11_01.operations.ServiceOperations>`
+           * 2021-03-01: :class:`ServiceOperations<azure.mgmt.databox.v2021_03_01.operations.ServiceOperations>`
+           * 2021-05-01: :class:`ServiceOperations<azure.mgmt.databox.v2021_05_01.operations.ServiceOperations>`
+           * 2021-08-01-preview: :class:`ServiceOperations<azure.mgmt.databox.v2021_08_01_preview.operations.ServiceOperations>`
+           * 2021-12-01: :class:`ServiceOperations<azure.mgmt.databox.v2021_12_01.operations.ServiceOperations>`
+           * 2022-02-01: :class:`ServiceOperations<azure.mgmt.databox.v2022_02_01.operations.ServiceOperations>`
+           * 2022-09-01: :class:`ServiceOperations<azure.mgmt.databox.v2022_09_01.operations.ServiceOperations>`
         """
         api_version = self._get_api_version('service')
         if api_version == '2018-01-01':
@@ -162,6 +234,18 @@ def service(self):
             from .v2020_04_01.operations import ServiceOperations as OperationClass
         elif api_version == '2020-11-01':
             from .v2020_11_01.operations import ServiceOperations as OperationClass
+        elif api_version == '2021-03-01':
+            from .v2021_03_01.operations import ServiceOperations as OperationClass
+        elif api_version == '2021-05-01':
+            from .v2021_05_01.operations import ServiceOperations as OperationClass
+        elif api_version == '2021-08-01-preview':
+            from .v2021_08_01_preview.operations import ServiceOperations as OperationClass
+        elif api_version == '2021-12-01':
+            from .v2021_12_01.operations import ServiceOperations as OperationClass
+        elif api_version == '2022-02-01':
+            from .v2022_02_01.operations import ServiceOperations as OperationClass
+        elif api_version == '2022-09-01':
+            from .v2022_09_01.operations import ServiceOperations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'service'".format(api_version))
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))

sdk/databox/azure-mgmt-databox/azure/mgmt/databox/_operations_mixin.py

@@ -0,0 +1,65 @@
+# 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 ._serialization import Serializer, Deserializer
+from typing import Any, IO, Optional, Union
+
+from . import models as _models
+
+
+class DataBoxManagementClientOperationsMixin(object):
+
+    def mitigate(  # pylint: disable=inconsistent-return-statements
+        self,
+        job_name: str,
+        resource_group_name: str,
+        mitigate_job_request: Union[_models.MitigateJobRequest, IO],
+        **kwargs: Any
+    ) -> None:
+        """Request to mitigate for a given job.
+
+        :param job_name: The name of the job Resource within the specified resource group. job names
+         must be between 3 and 24 characters in length and use any alphanumeric and underscore only.
+         Required.
+        :type job_name: str
+        :param resource_group_name: The Resource Group Name. Required.
+        :type resource_group_name: str
+        :param mitigate_job_request: Mitigation Request. Is either a model type or a IO type. Required.
+        :type mitigate_job_request: ~azure.mgmt.databox.v2022_09_01.models.MitigateJobRequest 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: None or the result of cls(response)
+        :rtype: None
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        api_version = self._get_api_version('mitigate')
+        if api_version == '2021-03-01':
+            from .v2021_03_01.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        elif api_version == '2021-05-01':
+            from .v2021_05_01.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        elif api_version == '2021-08-01-preview':
+            from .v2021_08_01_preview.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        elif api_version == '2021-12-01':
+            from .v2021_12_01.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        elif api_version == '2022-02-01':
+            from .v2022_02_01.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        elif api_version == '2022-09-01':
+            from .v2022_09_01.operations import DataBoxManagementClientOperationsMixin as OperationClass
+        else:
+            raise ValueError("API version {} does not have operation 'mitigate'".format(api_version))
+        mixin_instance = OperationClass()
+        mixin_instance._client = self._client
+        mixin_instance._config = self._config
+        mixin_instance._serialize = Serializer(self._models_dict(api_version))
+        mixin_instance._serialize.client_side_validation = False
+        mixin_instance._deserialize = Deserializer(self._models_dict(api_version))
+        return mixin_instance.mitigate(job_name, resource_group_name, mitigate_job_request, **kwargs)