[AutoPR datalake-store/resource-manager] [ADLS] add Get Usages API

sdk/azure-mgmt-datalake-store/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
+

sdk/azure-mgmt-datalake-store/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)

sdk/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/__init__.py

@@ -0,0 +1,154 @@
+# 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 .usage_name_py3 import UsageName
+    from .usage_py3 import Usage
+    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 .usage_name import UsageName
+    from .usage import Usage
+    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 .usage_paged import UsagePaged
+from .data_lake_store_account_management_client_enums import (
+    EncryptionConfigType,
+    EncryptionState,
+    EncryptionProvisioningState,
+    FirewallState,
+    FirewallAllowAzureIpsState,
+    TrustedIdProviderState,
+    TierType,
+    DataLakeStoreAccountStatus,
+    DataLakeStoreAccountState,
+    OperationOrigin,
+    SubscriptionState,
+    UsageUnit,
+)
+
+__all__ = [
+    'Resource',
+    'SubResource',
+    'EncryptionIdentity',
+    'KeyVaultMetaInfo',
+    'EncryptionConfig',
+    'FirewallRule',
+    'VirtualNetworkRule',
+    'TrustedIdProvider',
+    'DataLakeStoreAccount',
+    'DataLakeStoreAccountBasic',
+    'OperationDisplay',
+    'Operation',
+    'OperationListResult',
+    'CapabilityInformation',
+    'UsageName',
+    'Usage',
+    'NameAvailabilityInformation',
+    'CreateFirewallRuleWithAccountParameters',
+    'CreateVirtualNetworkRuleWithAccountParameters',
+    'CreateTrustedIdProviderWithAccountParameters',
+    'CreateDataLakeStoreAccountParameters',
+    'UpdateKeyVaultMetaInfo',
+    'UpdateEncryptionConfig',
+    'UpdateFirewallRuleWithAccountParameters',
+    'UpdateVirtualNetworkRuleWithAccountParameters',
+    'UpdateTrustedIdProviderWithAccountParameters',
+    'UpdateDataLakeStoreAccountParameters',
+    'CreateOrUpdateFirewallRuleParameters',
+    'UpdateFirewallRuleParameters',
+    'CreateOrUpdateVirtualNetworkRuleParameters',
+    'UpdateVirtualNetworkRuleParameters',
+    'CreateOrUpdateTrustedIdProviderParameters',
+    'UpdateTrustedIdProviderParameters',
+    'CheckNameAvailabilityParameters',
+    'DataLakeStoreAccountBasicPaged',
+    'FirewallRulePaged',
+    'VirtualNetworkRulePaged',
+    'TrustedIdProviderPaged',
+    'UsagePaged',
+    'EncryptionConfigType',
+    'EncryptionState',
+    'EncryptionProvisioningState',
+    'FirewallState',
+    'FirewallAllowAzureIpsState',
+    'TrustedIdProviderState',
+    'TierType',
+    'DataLakeStoreAccountStatus',
+    'DataLakeStoreAccountState',
+    'OperationOrigin',
+    'SubscriptionState',
+    'UsageUnit',
+]

sdk/azure-mgmt-datalake-store/azure/mgmt/datalake/store/models/capability_information.py

@@ -0,0 +1,60 @@
+# 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.serialization import Model
+
+
+class CapabilityInformation(Model):
+    """Subscription-level properties and limits for Data Lake Store.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar subscription_id: The subscription credentials that uniquely
+     identifies the subscription.
+    :vartype subscription_id: str
+    :ivar state: The subscription state. Possible values include:
+     'Registered', 'Suspended', 'Deleted', 'Unregistered', 'Warned'
+    :vartype state: str or ~azure.mgmt.datalake.store.models.SubscriptionState
+    :ivar max_account_count: The maximum supported number of accounts under
+     this subscription.
+    :vartype max_account_count: int
+    :ivar account_count: The current number of accounts under this
+     subscription.
+    :vartype account_count: int
+    :ivar migration_state: The Boolean value of true or false to indicate the
+     maintenance state.
+    :vartype migration_state: bool
+    """
+
+    _validation = {
+        'subscription_id': {'readonly': True},
+        'state': {'readonly': True},
+        'max_account_count': {'readonly': True},
+        'account_count': {'readonly': True},
+        'migration_state': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'subscription_id': {'key': 'subscriptionId', 'type': 'str'},
+        'state': {'key': 'state', 'type': 'str'},
+        'max_account_count': {'key': 'maxAccountCount', 'type': 'int'},
+        'account_count': {'key': 'accountCount', 'type': 'int'},
+        'migration_state': {'key': 'migrationState', 'type': 'bool'},
+    }
+
+    def __init__(self, **kwargs):
+        super(CapabilityInformation, self).__init__(**kwargs)
+        self.subscription_id = None
+        self.state = None
+        self.max_account_count = None
+        self.account_count = None
+        self.migration_state = None