[AutoPR cognitiveservices/resource-manager] Add CheckDomainAvailability resource type.

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

@@ -9,10 +9,11 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-from .cognitive_services_management_client import CognitiveServicesManagementClient
-from .version import VERSION
+from ._configuration import CognitiveServicesManagementClientConfiguration
+from ._cognitive_services_management_client import CognitiveServicesManagementClient
+__all__ = ['CognitiveServicesManagementClient', 'CognitiveServicesManagementClientConfiguration']
 
-__all__ = ['CognitiveServicesManagementClient']
+from .version import VERSION
 
 __version__ = VERSION
 

sdk/cognitiveservices/azure-mgmt-cognitiveservices/azure/mgmt/cognitiveservices/cognitive_services_management_client.py → sdk/cognitiveservices/azure-mgmt-cognitiveservices/azure/mgmt/cognitiveservices/_cognitive_services_management_client.py

@@ -11,48 +11,17 @@
 
 from msrest.service_client import SDKClient
 from msrest import Serializer, Deserializer
-from msrestazure import AzureConfiguration
-from .version import VERSION
-from .operations.accounts_operations import AccountsOperations
-from .operations.resource_skus_operations import ResourceSkusOperations
-from .operations.operations import Operations
-from .operations.check_sku_availability_operations import CheckSkuAvailabilityOperations
-from . import models
-
-
-class CognitiveServicesManagementClientConfiguration(AzureConfiguration):
-    """Configuration for CognitiveServicesManagementClient
-    Note that all parameters used to create this instance are saved as instance
-    attributes.
-
-    :param credentials: Credentials needed for the client to connect to Azure.
-    :type credentials: :mod:`A msrestazure Credentials
-     object<msrestazure.azure_active_directory>`
-    :param subscription_id: Azure Subscription ID.
-    :type subscription_id: str
-    :param str base_url: Service URL
-    """
-
-    def __init__(
-            self, credentials, subscription_id, base_url=None):
 
-        if credentials is None:
-            raise ValueError("Parameter 'credentials' must not be None.")
-        if subscription_id is None:
-            raise ValueError("Parameter 'subscription_id' must not be None.")
-        if not base_url:
-            base_url = 'https://management.azure.com'
-
-        super(CognitiveServicesManagementClientConfiguration, self).__init__(base_url)
-
-        self.add_user_agent('azure-mgmt-cognitiveservices/{}'.format(VERSION))
-        self.add_user_agent('Azure-SDK-For-Python')
-
-        self.credentials = credentials
-        self.subscription_id = subscription_id
+from ._configuration import CognitiveServicesManagementClientConfiguration
+from .operations import CognitiveServicesManagementClientOperationsMixin
+from .operations import AccountsOperations
+from .operations import ResourceSkusOperations
+from .operations import Operations
+from .operations import CheckSkuAvailabilityOperations
+from . import models
 
 
-class CognitiveServicesManagementClient(SDKClient):
+class CognitiveServicesManagementClient(CognitiveServicesManagementClientOperationsMixin, SDKClient):
     """Cognitive Services Management Client
 
     :ivar config: Configuration for client.

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

@@ -0,0 +1,48 @@
+# 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 msrestazure import AzureConfiguration
+
+from .version import VERSION
+
+
+class CognitiveServicesManagementClientConfiguration(AzureConfiguration):
+    """Configuration for CognitiveServicesManagementClient
+    Note that all parameters used to create this instance are saved as instance
+    attributes.
+
+    :param credentials: Credentials needed for the client to connect to Azure.
+    :type credentials: :mod:`A msrestazure Credentials
+     object<msrestazure.azure_active_directory>`
+    :param subscription_id: Azure Subscription ID.
+    :type subscription_id: str
+    :param str base_url: Service URL
+    """
+
+    def __init__(
+            self, credentials, subscription_id, base_url=None):
+
+        if credentials is None:
+            raise ValueError("Parameter 'credentials' must not be None.")
+        if subscription_id is None:
+            raise ValueError("Parameter 'subscription_id' must not be None.")
+        if not base_url:
+            base_url = 'https://management.azure.com'
+
+        super(CognitiveServicesManagementClientConfiguration, self).__init__(base_url)
+
+        # Starting Autorest.Python 4.0.64, make connection pool activated by default
+        self.keep_alive = True
+
+        self.add_user_agent('azure-mgmt-cognitiveservices/{}'.format(VERSION))
+        self.add_user_agent('Azure-SDK-For-Python')
+
+        self.credentials = credentials
+        self.subscription_id = subscription_id

sdk/cognitiveservices/azure-mgmt-cognitiveservices/azure/mgmt/cognitiveservices/models/__init__.py

@@ -10,55 +10,67 @@
 # --------------------------------------------------------------------------
 
 try:
-    from .sku_py3 import Sku
-    from .cognitive_services_account_create_parameters_py3 import CognitiveServicesAccountCreateParameters
-    from .cognitive_services_account_update_parameters_py3 import CognitiveServicesAccountUpdateParameters
-    from .cognitive_services_account_py3 import CognitiveServicesAccount
-    from .cognitive_services_account_keys_py3 import CognitiveServicesAccountKeys
-    from .regenerate_key_parameters_py3 import RegenerateKeyParameters
-    from .cognitive_services_resource_and_sku_py3 import CognitiveServicesResourceAndSku
-    from .cognitive_services_account_enumerate_skus_result_py3 import CognitiveServicesAccountEnumerateSkusResult
-    from .metric_name_py3 import MetricName
-    from .usage_py3 import Usage
-    from .usages_result_py3 import UsagesResult
-    from .error_body_py3 import ErrorBody
-    from .error_py3 import Error, ErrorException
-    from .operation_display_info_py3 import OperationDisplayInfo
-    from .operation_entity_py3 import OperationEntity
-    from .check_sku_availability_parameter_py3 import CheckSkuAvailabilityParameter
-    from .check_sku_availability_result_py3 import CheckSkuAvailabilityResult
-    from .check_sku_availability_result_list_py3 import CheckSkuAvailabilityResultList
-    from .resource_sku_restriction_info_py3 import ResourceSkuRestrictionInfo
-    from .resource_sku_restrictions_py3 import ResourceSkuRestrictions
-    from .resource_sku_py3 import ResourceSku
+    from ._models_py3 import CheckDomainAvailabilityParameter
+    from ._models_py3 import CheckDomainAvailabilityResult
+    from ._models_py3 import CheckSkuAvailabilityParameter
+    from ._models_py3 import CheckSkuAvailabilityResult
+    from ._models_py3 import CheckSkuAvailabilityResultList
+    from ._models_py3 import CognitiveServicesAccount
+    from ._models_py3 import CognitiveServicesAccountCreateParameters
+    from ._models_py3 import CognitiveServicesAccountEnumerateSkusResult
+    from ._models_py3 import CognitiveServicesAccountKeys
+    from ._models_py3 import CognitiveServicesAccountUpdateParameters
+    from ._models_py3 import CognitiveServicesResourceAndSku
+    from ._models_py3 import Error, ErrorException
+    from ._models_py3 import ErrorBody
+    from ._models_py3 import IpRule
+    from ._models_py3 import MetricName
+    from ._models_py3 import NetworkRuleSet
+    from ._models_py3 import OperationDisplayInfo
+    from ._models_py3 import OperationEntity
+    from ._models_py3 import RegenerateKeyParameters
+    from ._models_py3 import ResourceSku
+    from ._models_py3 import ResourceSkuRestrictionInfo
+    from ._models_py3 import ResourceSkuRestrictions
+    from ._models_py3 import Sku
+    from ._models_py3 import Usage
+    from ._models_py3 import UsagesResult
+    from ._models_py3 import VirtualNetworkRule
 except (SyntaxError, ImportError):
-    from .sku import Sku
-    from .cognitive_services_account_create_parameters import CognitiveServicesAccountCreateParameters
-    from .cognitive_services_account_update_parameters import CognitiveServicesAccountUpdateParameters
-    from .cognitive_services_account import CognitiveServicesAccount
-    from .cognitive_services_account_keys import CognitiveServicesAccountKeys
-    from .regenerate_key_parameters import RegenerateKeyParameters
-    from .cognitive_services_resource_and_sku import CognitiveServicesResourceAndSku
-    from .cognitive_services_account_enumerate_skus_result import CognitiveServicesAccountEnumerateSkusResult
-    from .metric_name import MetricName
-    from .usage import Usage
-    from .usages_result import UsagesResult
-    from .error_body import ErrorBody
-    from .error import Error, ErrorException
-    from .operation_display_info import OperationDisplayInfo
-    from .operation_entity import OperationEntity
-    from .check_sku_availability_parameter import CheckSkuAvailabilityParameter
-    from .check_sku_availability_result import CheckSkuAvailabilityResult
-    from .check_sku_availability_result_list import CheckSkuAvailabilityResultList
-    from .resource_sku_restriction_info import ResourceSkuRestrictionInfo
-    from .resource_sku_restrictions import ResourceSkuRestrictions
-    from .resource_sku import ResourceSku
-from .cognitive_services_account_paged import CognitiveServicesAccountPaged
-from .resource_sku_paged import ResourceSkuPaged
-from .operation_entity_paged import OperationEntityPaged
-from .cognitive_services_management_client_enums import (
+    from ._models import CheckDomainAvailabilityParameter
+    from ._models import CheckDomainAvailabilityResult
+    from ._models import CheckSkuAvailabilityParameter
+    from ._models import CheckSkuAvailabilityResult
+    from ._models import CheckSkuAvailabilityResultList
+    from ._models import CognitiveServicesAccount
+    from ._models import CognitiveServicesAccountCreateParameters
+    from ._models import CognitiveServicesAccountEnumerateSkusResult
+    from ._models import CognitiveServicesAccountKeys
+    from ._models import CognitiveServicesAccountUpdateParameters
+    from ._models import CognitiveServicesResourceAndSku
+    from ._models import Error, ErrorException
+    from ._models import ErrorBody
+    from ._models import IpRule
+    from ._models import MetricName
+    from ._models import NetworkRuleSet
+    from ._models import OperationDisplayInfo
+    from ._models import OperationEntity
+    from ._models import RegenerateKeyParameters
+    from ._models import ResourceSku
+    from ._models import ResourceSkuRestrictionInfo
+    from ._models import ResourceSkuRestrictions
+    from ._models import Sku
+    from ._models import Usage
+    from ._models import UsagesResult
+    from ._models import VirtualNetworkRule
+from ._paged_models import CognitiveServicesAccountPaged
+from ._paged_models import OperationEntityPaged
+from ._paged_models import ResourceSkuPaged
+from ._cognitive_services_management_client_enums import (
     SkuTier,
     ProvisioningState,
+    NetworkRuleBypassOptions,
+    NetworkRuleAction,
     KeyName,
     UnitType,
     QuotaUsageStatus,
@@ -67,32 +79,39 @@
 )
 
 __all__ = [
-    'Sku',
-    'CognitiveServicesAccountCreateParameters',
-    'CognitiveServicesAccountUpdateParameters',
+    'CheckDomainAvailabilityParameter',
+    'CheckDomainAvailabilityResult',
+    'CheckSkuAvailabilityParameter',
+    'CheckSkuAvailabilityResult',
+    'CheckSkuAvailabilityResultList',
     'CognitiveServicesAccount',
+    'CognitiveServicesAccountCreateParameters',
+    'CognitiveServicesAccountEnumerateSkusResult',
     'CognitiveServicesAccountKeys',
-    'RegenerateKeyParameters',
+    'CognitiveServicesAccountUpdateParameters',
     'CognitiveServicesResourceAndSku',
-    'CognitiveServicesAccountEnumerateSkusResult',
-    'MetricName',
-    'Usage',
-    'UsagesResult',
-    'ErrorBody',
     'Error', 'ErrorException',
+    'ErrorBody',
+    'IpRule',
+    'MetricName',
+    'NetworkRuleSet',
     'OperationDisplayInfo',
     'OperationEntity',
-    'CheckSkuAvailabilityParameter',
-    'CheckSkuAvailabilityResult',
-    'CheckSkuAvailabilityResultList',
+    'RegenerateKeyParameters',
+    'ResourceSku',
     'ResourceSkuRestrictionInfo',
     'ResourceSkuRestrictions',
-    'ResourceSku',
+    'Sku',
+    'Usage',
+    'UsagesResult',
+    'VirtualNetworkRule',
     'CognitiveServicesAccountPaged',
     'ResourceSkuPaged',
     'OperationEntityPaged',
     'SkuTier',
     'ProvisioningState',
+    'NetworkRuleBypassOptions',
+    'NetworkRuleAction',
     'KeyName',
     'UnitType',
     'QuotaUsageStatus',

sdk/cognitiveservices/azure-mgmt-cognitiveservices/azure/mgmt/cognitiveservices/models/cognitive_services_management_client_enums.py → sdk/cognitiveservices/azure-mgmt-cognitiveservices/azure/mgmt/cognitiveservices/models/_cognitive_services_management_client_enums.py

@@ -29,6 +29,18 @@ class ProvisioningState(str, Enum):
     failed = "Failed"
 
 
+class NetworkRuleBypassOptions(str, Enum):
+
+    azure_services = "AzureServices"
+    none = "None"
+
+
+class NetworkRuleAction(str, Enum):
+
+    allow = "Allow"
+    deny = "Deny"
+
+
 class KeyName(str, Enum):
 
     key1 = "Key1"