[AutoPR track2_azure-mgmt-databox] added new API version in preview, new SKU named Customer Disk, new API for the SKU and additional models required for new SKU

sdk/databox/azure-mgmt-databox/MANIFEST.in

@@ -1,3 +1,4 @@
+include _meta.json
 recursive-include tests *.py *.yaml
 include *.md
 include azure/__init__.py

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

@@ -0,0 +1,11 @@
+{
+  "autorest": "3.4.5",
+  "use": [
+    "@autorest/[email protected]",
+    "@autorest/[email protected]"
+  ],
+  "commit": "e954ef3aaafb2800f318adea5e41f32b95710ce9",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "autorest_command": "autorest specification/databox/resource-manager/readme.md --multiapi --python --python-mode=update --python-sdks-folder=/home/vsts/work/1/s/azure-sdk-for-python/sdk --track2 --use=@autorest/[email protected] --use=@autorest/[email protected] --version=3.4.5",
+  "readme": "specification/databox/resource-manager/readme.md"
+}

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

@@ -8,14 +8,19 @@
 # Changes may cause incorrect behavior and will be lost if the code is
 # regenerated.
 # --------------------------------------------------------------------------
-from typing import Any
+from typing import TYPE_CHECKING
 
 from azure.core.configuration import Configuration
 from azure.core.pipeline import policies
 from azure.mgmt.core.policies import ARMHttpLoggingPolicy
 
 from ._version import VERSION
 
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any
+
+    from azure.core.credentials import TokenCredential
 
 class DataBoxManagementClientConfiguration(Configuration):
     """Configuration for DataBoxManagementClient.

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

@@ -9,12 +9,22 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-from azure.mgmt.core import ARMPipelineClient
-from msrest import Serializer, Deserializer
+from typing import TYPE_CHECKING
 
+from azure.mgmt.core import ARMPipelineClient
 from azure.profiles import KnownProfiles, ProfileDefinition
 from azure.profiles.multiapiclient import MultiApiClientMixin
+from msrest import Deserializer, Serializer
+
 from ._configuration import DataBoxManagementClientConfiguration
+from ._operations_mixin import DataBoxManagementClientOperationsMixin
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any, Optional
+
+    from azure.core.credentials import TokenCredential
+    from azure.core.pipeline.transport import HttpRequest, HttpResponse
 
 class _SDKClient(object):
     def __init__(self, *args, **kwargs):
@@ -23,7 +33,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
@@ -38,15 +48,16 @@ class DataBoxManagementClient(MultiApiClientMixin, _SDKClient):
     :type credential: ~azure.core.credentials.TokenCredential
     :param subscription_id: The Subscription Id.
     :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 = '2021-05-01'
     _PROFILE_TAG = "azure.mgmt.databox.DataBoxManagementClient"
     LATEST_PROFILE = ProfileDefinition({
         _PROFILE_TAG: {
@@ -59,9 +70,9 @@ def __init__(
         self,
         credential,  # type: "TokenCredential"
         subscription_id,  # type: str
-        api_version=None,
-        base_url=None,
-        profile=KnownProfiles.default,
+        api_version=None, # type: Optional[str]
+        base_url=None,  # type: Optional[str]
+        profile=KnownProfiles.default, # type: KnownProfiles
         **kwargs  # type: Any
     ):
         if not base_url:
@@ -85,6 +96,9 @@ 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>`
         """
         if api_version == '2018-01-01':
             from .v2018_01_01 import models
@@ -98,6 +112,15 @@ 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
         raise ValueError("API version {} is not available".format(api_version))
 
     @property
@@ -108,6 +131,9 @@ 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>`
         """
         api_version = self._get_api_version('jobs')
         if api_version == '2018-01-01':
@@ -118,6 +144,12 @@ 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
         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 +162,9 @@ 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>`
         """
         api_version = self._get_api_version('operations')
         if api_version == '2018-01-01':
@@ -140,6 +175,12 @@ 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
         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 +193,9 @@ 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>`
         """
         api_version = self._get_api_version('service')
         if api_version == '2018-01-01':
@@ -162,6 +206,12 @@ 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
         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,63 @@
+# 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 msrest import Serializer, Deserializer
+from typing import TYPE_CHECKING
+import warnings
+
+from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.pipeline import PipelineResponse
+from azure.core.pipeline.transport import HttpRequest, HttpResponse
+from azure.mgmt.core.exceptions import ARMErrorFormat
+
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from typing import Any, Callable, Dict, Generic, Optional, TypeVar
+
+
+class DataBoxManagementClientOperationsMixin(object):
+
+    def mitigate(
+        self,
+        job_name,  # type: str
+        resource_group_name,  # type: str
+        mitigate_job_request,  # type: "_models.MitigateJobRequest"
+        **kwargs  # type: Any
+    ):
+        """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.
+        :type job_name: str
+        :param resource_group_name: The Resource Group Name.
+        :type resource_group_name: str
+        :param mitigate_job_request: Mitigation Request.
+        :type mitigate_job_request: ~azure.mgmt.databox.v2021_05_01.models.MitigateJobRequest
+        :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
+        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)

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

@@ -8,14 +8,17 @@
 # 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 .._version import VERSION
 
+if TYPE_CHECKING:
+    # pylint: disable=unused-import,ungrouped-imports
+    from azure.core.credentials_async import AsyncTokenCredential
 
 class DataBoxManagementClientConfiguration(Configuration):
     """Configuration for DataBoxManagementClient.
@@ -31,8 +34,8 @@ class DataBoxManagementClientConfiguration(Configuration):
 
     def __init__(
         self,
-        credential,  # type: "AsyncTokenCredential"
-        subscription_id,  # type: str
+        credential: "AsyncTokenCredential",
+        subscription_id: str,
         **kwargs  # type: Any
     ) -> None:
         if credential is None: