[AutoPR machinelearningservices/resource-manager] [Do not merge] Adding workspace level quota management APIs

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/azure_machine_learning_workspaces.py

@@ -16,6 +16,7 @@
 from .operations.operations import Operations
 from .operations.workspaces_operations import WorkspacesOperations
 from .operations.machine_learning_compute_operations import MachineLearningComputeOperations
+from .operations.quotas_operations import QuotasOperations
 from . import models
 
 
@@ -63,6 +64,8 @@ class AzureMachineLearningWorkspaces(SDKClient):
     :vartype workspaces: azure.mgmt.machinelearningservices.operations.WorkspacesOperations
     :ivar machine_learning_compute: MachineLearningCompute operations
     :vartype machine_learning_compute: azure.mgmt.machinelearningservices.operations.MachineLearningComputeOperations
+    :ivar quotas: Quotas operations
+    :vartype quotas: azure.mgmt.machinelearningservices.operations.QuotasOperations
 
     :param credentials: Credentials needed for the client to connect to Azure.
     :type credentials: :mod:`A msrestazure Credentials
@@ -89,3 +92,5 @@ def __init__(
             self._client, self.config, self._serialize, self._deserialize)
         self.machine_learning_compute = MachineLearningComputeOperations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.quotas = QuotasOperations(
+            self._client, self.config, self._serialize, self._deserialize)

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

@@ -14,6 +14,12 @@
     from .operation_py3 import Operation
     from .workspace_py3 import Workspace
     from .workspace_update_parameters_py3 import WorkspaceUpdateParameters
+    from .quota_base_properties_py3 import QuotaBaseProperties
+    from .quota_update_parameters_py3 import QuotaUpdateParameters
+    from .update_workspace_quotas_py3 import UpdateWorkspaceQuotas
+    from .update_workspace_quotas_result_py3 import UpdateWorkspaceQuotasResult
+    from .resource_name_py3 import ResourceName
+    from .resource_quota_py3 import ResourceQuota
     from .identity_py3 import Identity
     from .resource_py3 import Resource
     from .password_py3 import Password
@@ -46,6 +52,12 @@
     from .operation import Operation
     from .workspace import Workspace
     from .workspace_update_parameters import WorkspaceUpdateParameters
+    from .quota_base_properties import QuotaBaseProperties
+    from .quota_update_parameters import QuotaUpdateParameters
+    from .update_workspace_quotas import UpdateWorkspaceQuotas
+    from .update_workspace_quotas_result import UpdateWorkspaceQuotasResult
+    from .resource_name import ResourceName
+    from .resource_quota import ResourceQuota
     from .identity import Identity
     from .resource import Resource
     from .password import Password
@@ -76,8 +88,10 @@
 from .operation_paged import OperationPaged
 from .workspace_paged import WorkspacePaged
 from .compute_resource_paged import ComputeResourcePaged
+from .resource_quota_paged import ResourceQuotaPaged
 from .azure_machine_learning_workspaces_enums import (
     ProvisioningState,
+    Status,
     ResourceIdentityType,
     ComputeType,
 )
@@ -87,6 +101,12 @@
     'Operation',
     'Workspace',
     'WorkspaceUpdateParameters',
+    'QuotaBaseProperties',
+    'QuotaUpdateParameters',
+    'UpdateWorkspaceQuotas',
+    'UpdateWorkspaceQuotasResult',
+    'ResourceName',
+    'ResourceQuota',
     'Identity',
     'Resource',
     'Password',
@@ -117,7 +137,9 @@
     'OperationPaged',
     'WorkspacePaged',
     'ComputeResourcePaged',
+    'ResourceQuotaPaged',
     'ProvisioningState',
+    'Status',
     'ResourceIdentityType',
     'ComputeType',
 ]

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/azure_machine_learning_workspaces_enums.py

@@ -23,6 +23,13 @@ class ProvisioningState(str, Enum):
     canceled = "Canceled"
 
 
+class Status(str, Enum):
+
+    undefined = "Undefined"
+    success = "Success"
+    failure = "Failure"
+
+
 class ResourceIdentityType(str, Enum):
 
     system_assigned = "SystemAssigned"

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/quota_base_properties.py

@@ -0,0 +1,38 @@
+# 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 QuotaBaseProperties(Model):
+    """The properties for Quota update or retrieval.
+
+    :param vm_family: Virtual Machine family name. The family name of the
+     virtual machine size.
+    :type vm_family: str
+    :param workspace_id: AML workspace Id. Fully qualified resource identifier
+     of the workspace.
+    :type workspace_id: str
+    :param quota: Quota. The workspace level quota.
+    :type quota: int
+    """
+
+    _attribute_map = {
+        'vm_family': {'key': 'vmFamily', 'type': 'str'},
+        'workspace_id': {'key': 'workspaceId', 'type': 'str'},
+        'quota': {'key': 'quota', 'type': 'int'},
+    }
+
+    def __init__(self, **kwargs):
+        super(QuotaBaseProperties, self).__init__(**kwargs)
+        self.vm_family = kwargs.get('vm_family', None)
+        self.workspace_id = kwargs.get('workspace_id', None)
+        self.quota = kwargs.get('quota', None)

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/quota_base_properties_py3.py

@@ -0,0 +1,38 @@
+# 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 QuotaBaseProperties(Model):
+    """The properties for Quota update or retrieval.
+
+    :param vm_family: Virtual Machine family name. The family name of the
+     virtual machine size.
+    :type vm_family: str
+    :param workspace_id: AML workspace Id. Fully qualified resource identifier
+     of the workspace.
+    :type workspace_id: str
+    :param quota: Quota. The workspace level quota.
+    :type quota: int
+    """
+
+    _attribute_map = {
+        'vm_family': {'key': 'vmFamily', 'type': 'str'},
+        'workspace_id': {'key': 'workspaceId', 'type': 'str'},
+        'quota': {'key': 'quota', 'type': 'int'},
+    }
+
+    def __init__(self, *, vm_family: str=None, workspace_id: str=None, quota: int=None, **kwargs) -> None:
+        super(QuotaBaseProperties, self).__init__(**kwargs)
+        self.vm_family = vm_family
+        self.workspace_id = workspace_id
+        self.quota = quota

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/quota_update_parameters.py

@@ -0,0 +1,36 @@
+# 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 QuotaUpdateParameters(Model):
+    """Quota update parameters.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar value: The list for update quota.
+    :vartype value:
+     list[~azure.mgmt.machinelearningservices.models.QuotaBaseProperties]
+    """
+
+    _validation = {
+        'value': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'value': {'key': 'value', 'type': '[QuotaBaseProperties]'},
+    }
+
+    def __init__(self, **kwargs):
+        super(QuotaUpdateParameters, self).__init__(**kwargs)
+        self.value = None

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/quota_update_parameters_py3.py

@@ -0,0 +1,36 @@
+# 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 QuotaUpdateParameters(Model):
+    """Quota update parameters.
+
+    Variables are only populated by the server, and will be ignored when
+    sending a request.
+
+    :ivar value: The list for update quota.
+    :vartype value:
+     list[~azure.mgmt.machinelearningservices.models.QuotaBaseProperties]
+    """
+
+    _validation = {
+        'value': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'value': {'key': 'value', 'type': '[QuotaBaseProperties]'},
+    }
+
+    def __init__(self, **kwargs) -> None:
+        super(QuotaUpdateParameters, self).__init__(**kwargs)
+        self.value = None

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/resource_name.py

@@ -0,0 +1,40 @@
+# 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 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

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/resource_name_py3.py

@@ -0,0 +1,40 @@
+# 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 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) -> None:
+        super(ResourceName, self).__init__(**kwargs)
+        self.value = None
+        self.localized_value = None

sdk/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/resource_quota.py

@@ -0,0 +1,55 @@
+# 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 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 quota: Quota. Quota assigned to the resource.
+    :vartype quota: int
+    :ivar unit: The units used to represent the quota.
+    :vartype unit: str
+    """
+
+    _validation = {
+        'id': {'readonly': True},
+        'type': {'readonly': True},
+        'name': {'readonly': True},
+        'quota': {'readonly': True},
+        'unit': {'readonly': True},
+    }
+
+    _attribute_map = {
+        'id': {'key': 'id', 'type': 'str'},
+        'type': {'key': 'type', 'type': 'str'},
+        'name': {'key': 'name', 'type': 'ResourceName'},
+        'quota': {'key': 'quota', 'type': 'int'},
+        'unit': {'key': 'unit', 'type': 'str'},
+    }
+
+    def __init__(self, **kwargs):
+        super(ResourceQuota, self).__init__(**kwargs)
+        self.id = None
+        self.type = None
+        self.name = None
+        self.quota = None
+        self.unit = None