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

Merge 1e21dda0817b7942e031a2241bce81354576f926 into 8cf609cb9d734c394306297f6b7847c2752ab103

Author: SDKAuto

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

@@ -0,0 +1,11 @@
+{
+  "commit": "7789cd7e5b1a8f9ebeb80a004788f8df1fe597a7",
+  "repository_url": "https://github.com/Azure/azure-rest-api-specs",
+  "autorest": "3.9.2",
+  "use": [
+    "@autorest/python@6.2.16",
+    "@autorest/modelerfour@4.24.3"
+  ],
+  "autorest_command": "autorest specification/databox/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/mnt/vss/_work/1/s/azure-sdk-for-python/sdk --use=@autorest/python@6.2.16 --use=@autorest/modelerfour@4.24.3 --version=3.9.2 --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,34 +8,36 @@
 # 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
-        **kwargs  # type: Any
+        credential: "TokenCredential",
+        subscription_id: str,
+        **kwargs: Any
     ):
-        # type: (...) -> None
         if credential is None:
             raise ValueError("Parameter 'credential' must not be None.")
         if subscription_id is None:
@@ -50,9 +52,8 @@ def __init__(
 
     def _configure(
         self,
-        **kwargs  # type: Any
+        **kwargs: 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)
@@ -63,4 +64,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-10-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,
-        **kwargs  # type: Any
+        credential: "TokenCredential",
+        subscription_id: str,
+        api_version: Optional[str]=None,
+        base_url: str = "https://management.azure.com",
+        profile: KnownProfiles=KnownProfiles.default,
+        **kwargs: 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,13 @@ 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>`
+           * 2022-10-01: :mod:`v2022_10_01.models<azure.mgmt.databox.v2022_10_01.models>`
         """
         if api_version == '2018-01-01':
             from .v2018_01_01 import models
@@ -98,6 +111,27 @@ 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
+        elif api_version == '2022-10-01':
+            from .v2022_10_01 import models
+            return models
         raise ValueError("API version {} is not available".format(api_version))
 
     @property
@@ -108,6 +142,13 @@ 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>`
+           * 2022-10-01: :class:`JobsOperations<azure.mgmt.databox.v2022_10_01.operations.JobsOperations>`
         """
         api_version = self._get_api_version('jobs')
         if api_version == '2018-01-01':
@@ -118,8 +159,23 @@ 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
+        elif api_version == '2022-10-01':
+            from .v2022_10_01.operations import JobsOperations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'jobs'".format(api_version))
+        self._config.api_version = api_version
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))
 
     @property
@@ -130,6 +186,13 @@ 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>`
+           * 2022-10-01: :class:`Operations<azure.mgmt.databox.v2022_10_01.operations.Operations>`
         """
         api_version = self._get_api_version('operations')
         if api_version == '2018-01-01':
@@ -140,8 +203,23 @@ 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
+        elif api_version == '2022-10-01':
+            from .v2022_10_01.operations import Operations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'operations'".format(api_version))
+        self._config.api_version = api_version
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))
 
     @property
@@ -152,6 +230,13 @@ 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>`
+           * 2022-10-01: :class:`ServiceOperations<azure.mgmt.databox.v2022_10_01.operations.ServiceOperations>`
         """
         api_version = self._get_api_version('service')
         if api_version == '2018-01-01':
@@ -162,8 +247,23 @@ 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
+        elif api_version == '2022-10-01':
+            from .v2022_10_01.operations import ServiceOperations as OperationClass
         else:
             raise ValueError("API version {} does not have operation group 'service'".format(api_version))
+        self._config.api_version = api_version
         return OperationClass(self._client, self._config, Serializer(self._models_dict(api_version)), Deserializer(self._models_dict(api_version)))
 
     def close(self):