{Packaging} Vendor azure-mgmt-datalake-store (#29275)

src/azure-cli-core/azure/cli/core/profiles/_shared.py

@@ -101,7 +101,7 @@ class ResourceType(Enum):  # pylint: disable=too-few-public-methods
     MGMT_CONTAINERINSTANCE = ('azure.mgmt.containerinstance', None)
     MGMT_COSMOSDB = ('azure.mgmt.cosmosdb', None)
     MGMT_DATALAKE_ANALYTICS = ('azure.cli.command_modules.dla.vendored_sdks.azure_mgmt_datalake_analytics', None)
-    MGMT_DATALAKE_STORE = ('azure.mgmt.datalake.store', None)
+    MGMT_DATALAKE_STORE = ('azure.cli.command_modules.dls.vendored_sdks.azure_mgmt_datalake_store', None)
     MGMT_DATAMIGRATION = ('azure.mgmt.datamigration', None)
     MGMT_EVENTGRID = ('azure.mgmt.eventgrid', None)
     MGMT_DEVTESTLABS = ('azure.mgmt.devtestlabs', None)

src/azure-cli/azure/cli/command_modules/dls/_client_factory.py

@@ -6,25 +6,25 @@
 
 def cf_dls_account(cli_ctx, _):
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
-    from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
+    from .vendored_sdks.azure_mgmt_datalake_store import DataLakeStoreAccountManagementClient
     return get_mgmt_service_client(cli_ctx, DataLakeStoreAccountManagementClient).accounts
 
 
 def cf_dls_account_firewall(cli_ctx, _):
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
-    from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
+    from .vendored_sdks.azure_mgmt_datalake_store import DataLakeStoreAccountManagementClient
     return get_mgmt_service_client(cli_ctx, DataLakeStoreAccountManagementClient).firewall_rules
 
 
 def cf_dls_account_virtual_network(cli_ctx, _):
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
-    from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
+    from .vendored_sdks.azure_mgmt_datalake_store import DataLakeStoreAccountManagementClient
     return get_mgmt_service_client(cli_ctx, DataLakeStoreAccountManagementClient).virtual_network_rules
 
 
 def cf_dls_account_trusted_provider(cli_ctx, _):
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
-    from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
+    from .vendored_sdks.azure_mgmt_datalake_store import DataLakeStoreAccountManagementClient
     return get_mgmt_service_client(cli_ctx, DataLakeStoreAccountManagementClient).trusted_id_providers
 
 

src/azure-cli/azure/cli/command_modules/dls/_params.py

@@ -9,13 +9,13 @@
     tags_type, get_resource_name_completion_list, resource_group_name_type, get_enum_type)
 
 from azure.cli.command_modules.dls._validators import validate_resource_group_name
-from azure.mgmt.datalake.store.models import (
+from .vendored_sdks.azure_mgmt_datalake_store.models import (
     FirewallState,
     TrustedIdProviderState,
     TierType,
     FirewallAllowAzureIpsState)
 
-from azure.mgmt.datalake.store.models import EncryptionConfigType
+from .vendored_sdks.azure_mgmt_datalake_store.models import EncryptionConfigType
 
 
 # pylint: disable=line-too-long, too-many-statements

src/azure-cli/azure/cli/command_modules/dls/_validators.py

@@ -29,7 +29,7 @@ def _get_resource_group_from_account_name(client, account_name):
 
 # COMMAND NAMESPACE VALIDATORS
 def validate_resource_group_name(cmd, ns):
-    from azure.mgmt.datalake.store import DataLakeStoreAccountManagementClient
+    from .vendored_sdks.azure_mgmt_datalake_store import DataLakeStoreAccountManagementClient
     if not ns.resource_group_name:
         try:
             account_name = ns.name

src/azure-cli/azure/cli/command_modules/dls/commands.py

@@ -17,7 +17,7 @@ def load_command_table(self, _):
     from ._validators import (
         validate_subnet
     )
-    adls_format_path = 'azure.mgmt.datalake.store.operations.{}#{}.{{}}'
+    adls_format_path = 'azure.cli.command_modules.dls.vendored_sdks.azure_mgmt_datalake_store.operations.{}#{}.{{}}'
 
     dls_custom = CliCommandType(operations_tmpl='azure.cli.command_modules.dls.custom#{}')
 

src/azure-cli/azure/cli/command_modules/dls/custom.py

@@ -6,7 +6,7 @@
 from knack.log import get_logger
 from knack.util import CLIError
 
-from azure.mgmt.datalake.store.models import (
+from .vendored_sdks.azure_mgmt_datalake_store.models import (
     UpdateDataLakeStoreAccountParameters,
     CreateDataLakeStoreAccountParameters,
     EncryptionConfigType,

src/azure-cli/azure/cli/command_modules/dls/vendored_sdks/__init__.py

@@ -0,0 +1,6 @@
+# --------------------------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------------------------
+# pylint: skip-file
+# flake8: noqa

src/azure-cli/azure/cli/command_modules/dls/vendored_sdks/azure_mgmt_datalake_store/__init__.py

@@ -0,0 +1,18 @@
+# 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 .data_lake_store_account_management_client import DataLakeStoreAccountManagementClient
+from .version import VERSION
+
+__all__ = ['DataLakeStoreAccountManagementClient']
+
+__version__ = VERSION
+

src/azure-cli/azure/cli/command_modules/dls/vendored_sdks/azure_mgmt_datalake_store/data_lake_store_account_management_client.py

@@ -0,0 +1,110 @@
+# 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.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.firewall_rules_operations import FirewallRulesOperations
+from .operations.virtual_network_rules_operations import VirtualNetworkRulesOperations
+from .operations.trusted_id_providers_operations import TrustedIdProvidersOperations
+from .operations.operations import Operations
+from .operations.locations_operations import LocationsOperations
+from . import models
+
+
+class DataLakeStoreAccountManagementClientConfiguration(AzureConfiguration):
+    """Configuration for DataLakeStoreAccountManagementClient
+    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: Gets subscription credentials which uniquely
+     identify Microsoft Azure subscription. The subscription ID forms part of
+     the URI for every service call.
+    :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(DataLakeStoreAccountManagementClientConfiguration, self).__init__(base_url)
+
+        self.add_user_agent('azure-mgmt-datalake-store/{}'.format(VERSION))
+        self.add_user_agent('Azure-SDK-For-Python')
+
+        self.credentials = credentials
+        self.subscription_id = subscription_id
+
+
+class DataLakeStoreAccountManagementClient(SDKClient):
+    """Creates an Azure Data Lake Store account management client.
+
+    :ivar config: Configuration for client.
+    :vartype config: DataLakeStoreAccountManagementClientConfiguration
+
+    :ivar accounts: Accounts operations
+    :vartype accounts: azure.mgmt.datalake.store.operations.AccountsOperations
+    :ivar firewall_rules: FirewallRules operations
+    :vartype firewall_rules: azure.mgmt.datalake.store.operations.FirewallRulesOperations
+    :ivar virtual_network_rules: VirtualNetworkRules operations
+    :vartype virtual_network_rules: azure.mgmt.datalake.store.operations.VirtualNetworkRulesOperations
+    :ivar trusted_id_providers: TrustedIdProviders operations
+    :vartype trusted_id_providers: azure.mgmt.datalake.store.operations.TrustedIdProvidersOperations
+    :ivar operations: Operations operations
+    :vartype operations: azure.mgmt.datalake.store.operations.Operations
+    :ivar locations: Locations operations
+    :vartype locations: azure.mgmt.datalake.store.operations.LocationsOperations
+
+    :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: Gets subscription credentials which uniquely
+     identify Microsoft Azure subscription. The subscription ID forms part of
+     the URI for every service call.
+    :type subscription_id: str
+    :param str base_url: Service URL
+    """
+
+    def __init__(
+            self, credentials, subscription_id, base_url=None):
+
+        self.config = DataLakeStoreAccountManagementClientConfiguration(credentials, subscription_id, base_url)
+        super(DataLakeStoreAccountManagementClient, self).__init__(self.config.credentials, self.config)
+
+        client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
+        self.api_version = '2016-11-01'
+        self._serialize = Serializer(client_models)
+        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.virtual_network_rules = VirtualNetworkRulesOperations(
+            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)

src/azure-cli/azure/cli/command_modules/dls/vendored_sdks/azure_mgmt_datalake_store/models/__init__.py

@@ -0,0 +1,144 @@
+# 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.
+# --------------------------------------------------------------------------
+
+try:
+    from .resource_py3 import Resource
+    from .sub_resource_py3 import SubResource
+    from .encryption_identity_py3 import EncryptionIdentity
+    from .key_vault_meta_info_py3 import KeyVaultMetaInfo
+    from .encryption_config_py3 import EncryptionConfig
+    from .firewall_rule_py3 import FirewallRule
+    from .virtual_network_rule_py3 import VirtualNetworkRule
+    from .trusted_id_provider_py3 import TrustedIdProvider
+    from .data_lake_store_account_py3 import DataLakeStoreAccount
+    from .data_lake_store_account_basic_py3 import DataLakeStoreAccountBasic
+    from .operation_display_py3 import OperationDisplay
+    from .operation_py3 import Operation
+    from .operation_list_result_py3 import OperationListResult
+    from .capability_information_py3 import CapabilityInformation
+    from .name_availability_information_py3 import NameAvailabilityInformation
+    from .create_firewall_rule_with_account_parameters_py3 import CreateFirewallRuleWithAccountParameters
+    from .create_virtual_network_rule_with_account_parameters_py3 import CreateVirtualNetworkRuleWithAccountParameters
+    from .create_trusted_id_provider_with_account_parameters_py3 import CreateTrustedIdProviderWithAccountParameters
+    from .create_data_lake_store_account_parameters_py3 import CreateDataLakeStoreAccountParameters
+    from .update_key_vault_meta_info_py3 import UpdateKeyVaultMetaInfo
+    from .update_encryption_config_py3 import UpdateEncryptionConfig
+    from .update_firewall_rule_with_account_parameters_py3 import UpdateFirewallRuleWithAccountParameters
+    from .update_virtual_network_rule_with_account_parameters_py3 import UpdateVirtualNetworkRuleWithAccountParameters
+    from .update_trusted_id_provider_with_account_parameters_py3 import UpdateTrustedIdProviderWithAccountParameters
+    from .update_data_lake_store_account_parameters_py3 import UpdateDataLakeStoreAccountParameters
+    from .create_or_update_firewall_rule_parameters_py3 import CreateOrUpdateFirewallRuleParameters
+    from .update_firewall_rule_parameters_py3 import UpdateFirewallRuleParameters
+    from .create_or_update_virtual_network_rule_parameters_py3 import CreateOrUpdateVirtualNetworkRuleParameters
+    from .update_virtual_network_rule_parameters_py3 import UpdateVirtualNetworkRuleParameters
+    from .create_or_update_trusted_id_provider_parameters_py3 import CreateOrUpdateTrustedIdProviderParameters
+    from .update_trusted_id_provider_parameters_py3 import UpdateTrustedIdProviderParameters
+    from .check_name_availability_parameters_py3 import CheckNameAvailabilityParameters
+except (SyntaxError, ImportError):
+    from .resource import Resource
+    from .sub_resource import SubResource
+    from .encryption_identity import EncryptionIdentity
+    from .key_vault_meta_info import KeyVaultMetaInfo
+    from .encryption_config import EncryptionConfig
+    from .firewall_rule import FirewallRule
+    from .virtual_network_rule import VirtualNetworkRule
+    from .trusted_id_provider import TrustedIdProvider
+    from .data_lake_store_account import DataLakeStoreAccount
+    from .data_lake_store_account_basic import DataLakeStoreAccountBasic
+    from .operation_display import OperationDisplay
+    from .operation import Operation
+    from .operation_list_result import OperationListResult
+    from .capability_information import CapabilityInformation
+    from .name_availability_information import NameAvailabilityInformation
+    from .create_firewall_rule_with_account_parameters import CreateFirewallRuleWithAccountParameters
+    from .create_virtual_network_rule_with_account_parameters import CreateVirtualNetworkRuleWithAccountParameters
+    from .create_trusted_id_provider_with_account_parameters import CreateTrustedIdProviderWithAccountParameters
+    from .create_data_lake_store_account_parameters import CreateDataLakeStoreAccountParameters
+    from .update_key_vault_meta_info import UpdateKeyVaultMetaInfo
+    from .update_encryption_config import UpdateEncryptionConfig
+    from .update_firewall_rule_with_account_parameters import UpdateFirewallRuleWithAccountParameters
+    from .update_virtual_network_rule_with_account_parameters import UpdateVirtualNetworkRuleWithAccountParameters
+    from .update_trusted_id_provider_with_account_parameters import UpdateTrustedIdProviderWithAccountParameters
+    from .update_data_lake_store_account_parameters import UpdateDataLakeStoreAccountParameters
+    from .create_or_update_firewall_rule_parameters import CreateOrUpdateFirewallRuleParameters
+    from .update_firewall_rule_parameters import UpdateFirewallRuleParameters
+    from .create_or_update_virtual_network_rule_parameters import CreateOrUpdateVirtualNetworkRuleParameters
+    from .update_virtual_network_rule_parameters import UpdateVirtualNetworkRuleParameters
+    from .create_or_update_trusted_id_provider_parameters import CreateOrUpdateTrustedIdProviderParameters
+    from .update_trusted_id_provider_parameters import UpdateTrustedIdProviderParameters
+    from .check_name_availability_parameters import CheckNameAvailabilityParameters
+from .data_lake_store_account_basic_paged import DataLakeStoreAccountBasicPaged
+from .firewall_rule_paged import FirewallRulePaged
+from .virtual_network_rule_paged import VirtualNetworkRulePaged
+from .trusted_id_provider_paged import TrustedIdProviderPaged
+from .data_lake_store_account_management_client_enums import (
+    EncryptionConfigType,
+    EncryptionState,
+    EncryptionProvisioningState,
+    FirewallState,
+    FirewallAllowAzureIpsState,
+    TrustedIdProviderState,
+    TierType,
+    DataLakeStoreAccountStatus,
+    DataLakeStoreAccountState,
+    OperationOrigin,
+    SubscriptionState,
+)
+
+__all__ = [
+    'Resource',
+    'SubResource',
+    'EncryptionIdentity',
+    'KeyVaultMetaInfo',
+    'EncryptionConfig',
+    'FirewallRule',
+    'VirtualNetworkRule',
+    'TrustedIdProvider',
+    'DataLakeStoreAccount',
+    'DataLakeStoreAccountBasic',
+    'OperationDisplay',
+    'Operation',
+    'OperationListResult',
+    'CapabilityInformation',
+    'NameAvailabilityInformation',
+    'CreateFirewallRuleWithAccountParameters',
+    'CreateVirtualNetworkRuleWithAccountParameters',
+    'CreateTrustedIdProviderWithAccountParameters',
+    'CreateDataLakeStoreAccountParameters',
+    'UpdateKeyVaultMetaInfo',
+    'UpdateEncryptionConfig',
+    'UpdateFirewallRuleWithAccountParameters',
+    'UpdateVirtualNetworkRuleWithAccountParameters',
+    'UpdateTrustedIdProviderWithAccountParameters',
+    'UpdateDataLakeStoreAccountParameters',
+    'CreateOrUpdateFirewallRuleParameters',
+    'UpdateFirewallRuleParameters',
+    'CreateOrUpdateVirtualNetworkRuleParameters',
+    'UpdateVirtualNetworkRuleParameters',
+    'CreateOrUpdateTrustedIdProviderParameters',
+    'UpdateTrustedIdProviderParameters',
+    'CheckNameAvailabilityParameters',
+    'DataLakeStoreAccountBasicPaged',
+    'FirewallRulePaged',
+    'VirtualNetworkRulePaged',
+    'TrustedIdProviderPaged',
+    'EncryptionConfigType',
+    'EncryptionState',
+    'EncryptionProvisioningState',
+    'FirewallState',
+    'FirewallAllowAzureIpsState',
+    'TrustedIdProviderState',
+    'TierType',
+    'DataLakeStoreAccountStatus',
+    'DataLakeStoreAccountState',
+    'OperationOrigin',
+    'SubscriptionState',
+]