[AutoPR machinelearningservices/resource-manager] [READY TO MERGE] Adding workspace level quota management APIs in version 2019-06-01

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/_azure_machine_learning_workspaces.py

@@ -17,6 +17,7 @@
 from .operations import WorkspacesOperations
 from .operations import UsagesOperations
 from .operations import VirtualMachineSizesOperations
+from .operations import QuotasOperations
 from .operations import MachineLearningComputeOperations
 from . import models
 
@@ -35,6 +36,8 @@ class AzureMachineLearningWorkspaces(SDKClient):
     :vartype usages: azure.mgmt.machinelearningservices.operations.UsagesOperations
     :ivar virtual_machine_sizes: VirtualMachineSizes operations
     :vartype virtual_machine_sizes: azure.mgmt.machinelearningservices.operations.VirtualMachineSizesOperations
+    :ivar quotas: Quotas operations
+    :vartype quotas: azure.mgmt.machinelearningservices.operations.QuotasOperations
     :ivar machine_learning_compute: MachineLearningCompute operations
     :vartype machine_learning_compute: azure.mgmt.machinelearningservices.operations.MachineLearningComputeOperations
 
@@ -65,5 +68,7 @@ def __init__(
             self._client, self.config, self._serialize, self._deserialize)
         self.virtual_machine_sizes = VirtualMachineSizesOperations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.quotas = QuotasOperations(
+            self._client, self.config, self._serialize, self._deserialize)
         self.machine_learning_compute = MachineLearningComputeOperations(
             self._client, self.config, self._serialize, self._deserialize)

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/__init__.py

@@ -40,13 +40,19 @@
     from ._models_py3 import Operation
     from ._models_py3 import OperationDisplay
     from ._models_py3 import Password
+    from ._models_py3 import QuotaBaseProperties
+    from ._models_py3 import QuotaUpdateParameters
     from ._models_py3 import RegistryListCredentialsResult
     from ._models_py3 import Resource
     from ._models_py3 import ResourceId
+    from ._models_py3 import ResourceName
+    from ._models_py3 import ResourceQuota
     from ._models_py3 import ScaleSettings
     from ._models_py3 import ServicePrincipalCredentials
     from ._models_py3 import SslConfiguration
     from ._models_py3 import SystemService
+    from ._models_py3 import UpdateWorkspaceQuotas
+    from ._models_py3 import UpdateWorkspaceQuotasResult
     from ._models_py3 import Usage
     from ._models_py3 import UsageName
     from ._models_py3 import UserAccountCredentials
@@ -89,13 +95,19 @@
     from ._models import Operation
     from ._models import OperationDisplay
     from ._models import Password
+    from ._models import QuotaBaseProperties
+    from ._models import QuotaUpdateParameters
     from ._models import RegistryListCredentialsResult
     from ._models import Resource
     from ._models import ResourceId
+    from ._models import ResourceName
+    from ._models import ResourceQuota
     from ._models import ScaleSettings
     from ._models import ServicePrincipalCredentials
     from ._models import SslConfiguration
     from ._models import SystemService
+    from ._models import UpdateWorkspaceQuotas
+    from ._models import UpdateWorkspaceQuotasResult
     from ._models import Usage
     from ._models import UsageName
     from ._models import UserAccountCredentials
@@ -109,11 +121,14 @@
     from ._models import WorkspaceUpdateParameters
 from ._paged_models import ComputeResourcePaged
 from ._paged_models import OperationPaged
+from ._paged_models import ResourceQuotaPaged
 from ._paged_models import UsagePaged
 from ._paged_models import WorkspacePaged
 from ._azure_machine_learning_workspaces_enums import (
     ProvisioningState,
     UsageUnit,
+    QuotaUnit,
+    Status,
     ResourceIdentityType,
     VmPriority,
     RemoteLoginPortPublicAccess,
@@ -154,13 +169,19 @@
     'Operation',
     'OperationDisplay',
     'Password',
+    'QuotaBaseProperties',
+    'QuotaUpdateParameters',
     'RegistryListCredentialsResult',
     'Resource',
     'ResourceId',
+    'ResourceName',
+    'ResourceQuota',
     'ScaleSettings',
     'ServicePrincipalCredentials',
     'SslConfiguration',
     'SystemService',
+    'UpdateWorkspaceQuotas',
+    'UpdateWorkspaceQuotasResult',
     'Usage',
     'UsageName',
     'UserAccountCredentials',
@@ -175,9 +196,12 @@
     'OperationPaged',
     'WorkspacePaged',
     'UsagePaged',
+    'ResourceQuotaPaged',
     'ComputeResourcePaged',
     'ProvisioningState',
     'UsageUnit',
+    'QuotaUnit',
+    'Status',
     'ResourceIdentityType',
     'VmPriority',
     'RemoteLoginPortPublicAccess',

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/_azure_machine_learning_workspaces_enums.py

@@ -28,6 +28,21 @@ class UsageUnit(str, Enum):
     count = "Count"
 
 
+class QuotaUnit(str, Enum):
+
+    count = "Count"
+
+
+class Status(str, Enum):
+
+    undefined = "Undefined"
+    success = "Success"
+    failure = "Failure"
+    invalid_quota_below_cluster_minimum = "InvalidQuotaBelowClusterMinimum"
+    invalid_quota_exceeds_subscription_limit = "InvalidQuotaExceedsSubscriptionLimit"
+    invalid_vm_family_name = "InvalidVMFamilyName"
+
+
 class ResourceIdentityType(str, Enum):
 
     system_assigned = "SystemAssigned"

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/_models.py

@@ -1366,6 +1366,52 @@ def __init__(self, **kwargs):
         self.value = None
 
 
+class QuotaBaseProperties(Model):
+    """The properties for Quota update or retrieval.
+
+    :param id: Specifies the resource ID.
+    :type id: str
+    :param type: Specifies the resource type.
+    :type type: str
+    :param limit: Limit. The maximum permitted quota of the resource.
+    :type limit: long
+    :param unit: An enum describing the unit of quota measurement. Possible
+     values include: 'Count'
+    :type unit: str or ~azure.mgmt.machinelearningservices.models.QuotaUnit
+    """
+
+    _attribute_map = {
+        'id': {'key': 'id', 'type': 'str'},
+        'type': {'key': 'type', 'type': 'str'},
+        'limit': {'key': 'limit', 'type': 'long'},
+        'unit': {'key': 'unit', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(QuotaBaseProperties, self).__init__(**kwargs)
+        self.id = kwargs.get('id', None)
+        self.type = kwargs.get('type', None)
+        self.limit = kwargs.get('limit', None)
+        self.unit = kwargs.get('unit', None)
+
+
+class QuotaUpdateParameters(Model):
+    """Quota update parameters.
+
+    :param value: The list for update quota.
+    :type value:
+     list[~azure.mgmt.machinelearningservices.models.QuotaBaseProperties]
+    """
+
+    _attribute_map = {
+        'value': {'key': 'value', 'type': '[QuotaBaseProperties]'},
+    }
+
+    def __init__(self, **kwargs):
+        super(QuotaUpdateParameters, self).__init__(**kwargs)
+        self.value = kwargs.get('value', None)
+
+
 class RegistryListCredentialsResult(Model):
     """RegistryListCredentialsResult.
 
@@ -1421,6 +1467,78 @@ def __init__(self, **kwargs):
         self.id = kwargs.get('id', None)
 
 
+class ResourceName(Model):
+    """The Resource Name.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar value: The name of the resource.
+    :vartype value: str
+    :ivar localized_value: The localized name of the resource.
+    :vartype localized_value: str
+    """
+
+    _validation = {
+        'value': {'readonly': True},
+        'localized_value': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'value': {'key': 'value', 'type': 'str'},
+        'localized_value': {'key': 'localizedValue', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(ResourceName, self).__init__(**kwargs)
+        self.value = None
+        self.localized_value = None
+
+
+class ResourceQuota(Model):
+    """The quota assigned to a resource.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar id: Specifies the resource ID.
+    :vartype id: str
+    :ivar type: Specifies the resource type.
+    :vartype type: str
+    :ivar name: Name of the resource.
+    :vartype name: ~azure.mgmt.machinelearningservices.models.ResourceName
+    :ivar limit: Limit. The maximum permitted quota of the resource.
+    :vartype limit: long
+    :ivar unit: An enum describing the unit of quota measurement. Possible
+     values include: 'Count'
+    :vartype unit: str or ~azure.mgmt.machinelearningservices.models.QuotaUnit
+    """
+
+    _validation = {
+        'id': {'readonly': True},
+        'type': {'readonly': True},
+        'name': {'readonly': True},
+        'limit': {'readonly': True},
+        'unit': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'id': {'key': 'id', 'type': 'str'},
+        'type': {'key': 'type', 'type': 'str'},
+        'name': {'key': 'name', 'type': 'ResourceName'},
+        'limit': {'key': 'limit', 'type': 'long'},
+        'unit': {'key': 'unit', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(ResourceQuota, self).__init__(**kwargs)
+        self.id = None
+        self.type = None
+        self.name = None
+        self.limit = None
+        self.unit = None
+
+
 class ScaleSettings(Model):
     """scale settings for AML Compute.
 
@@ -1541,6 +1659,82 @@ def __init__(self, **kwargs):
         self.version = None
 
 
+class UpdateWorkspaceQuotas(Model):
+    """The properties for update Quota response.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar id: Specifies the resource ID.
+    :vartype id: str
+    :ivar type: Specifies the resource type.
+    :vartype type: str
+    :param limit: Limit. The maximum permitted quota of the resource.
+    :type limit: long
+    :ivar unit: An enum describing the unit of quota measurement. Possible
+     values include: 'Count'
+    :vartype unit: str or ~azure.mgmt.machinelearningservices.models.QuotaUnit
+    :param status: Update Workspace Quota Status. Status of update workspace
+     quota. Possible values include: 'Undefined', 'Success', 'Failure',
+     'InvalidQuotaBelowClusterMinimum', 'InvalidQuotaExceedsSubscriptionLimit',
+     'InvalidVMFamilyName'
+    :type status: str or ~azure.mgmt.machinelearningservices.models.Status
+    """
+
+    _validation = {
+        'id': {'readonly': True},
+        'type': {'readonly': True},
+        'unit': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'id': {'key': 'id', 'type': 'str'},
+        'type': {'key': 'type', 'type': 'str'},
+        'limit': {'key': 'limit', 'type': 'long'},
+        'unit': {'key': 'unit', 'type': 'str'},
+        'status': {'key': 'status', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(UpdateWorkspaceQuotas, self).__init__(**kwargs)
+        self.id = None
+        self.type = None
+        self.limit = kwargs.get('limit', None)
+        self.unit = None
+        self.status = kwargs.get('status', None)
+
+
+class UpdateWorkspaceQuotasResult(Model):
+    """The result of update workspace quota.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar value: The list of workspace quota update result.
+    :vartype value:
+     list[~azure.mgmt.machinelearningservices.models.UpdateWorkspaceQuotas]
+    :ivar next_link: The URI to fetch the next page of workspace quota update
+     result. Call ListNext() with this to fetch the next page of Workspace
+     Quota update result.
+    :vartype next_link: str
+    """
+
+    _validation = {
+        'value': {'readonly': True},
+        'next_link': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'value': {'key': 'value', 'type': '[UpdateWorkspaceQuotas]'},
+        'next_link': {'key': 'nextLink', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(UpdateWorkspaceQuotasResult, self).__init__(**kwargs)
+        self.value = None
+        self.next_link = None
+
+
 class Usage(Model):
     """Describes AML Resource Usage.