[AutoPR sql/resource-manager] [DO NOT MERGE] Sensitivity labels: Disable/Enable recommended sensitivity labels

azure-mgmt-sql/azure/mgmt/sql/models/sensitivity_label.py

@@ -32,12 +32,17 @@ class SensitivityLabel(ProxyResource):
     :type information_type: str
     :param information_type_id: The information type ID.
     :type information_type_id: str
+    :ivar is_disabled: Is sensitivity recommendation disabled. Applicable for
+     recommended sensitivity label only. Specifies whether the sensitivity
+     recommendation on this column is disabled (dismissed) or not.
+    :vartype is_disabled: bool
     """
 
     _validation = {
         'id': {'readonly': True},
         'name': {'readonly': True},
         'type': {'readonly': True},
+        'is_disabled': {'readonly': True},
     }
 
     _attribute_map = {
@@ -48,6 +53,7 @@ class SensitivityLabel(ProxyResource):
         'label_id': {'key': 'properties.labelId', 'type': 'str'},
         'information_type': {'key': 'properties.informationType', 'type': 'str'},
         'information_type_id': {'key': 'properties.informationTypeId', 'type': 'str'},
+        'is_disabled': {'key': 'properties.isDisabled', 'type': 'bool'},
     }
 
     def __init__(self, **kwargs):
@@ -56,3 +62,4 @@ def __init__(self, **kwargs):
         self.label_id = kwargs.get('label_id', None)
         self.information_type = kwargs.get('information_type', None)
         self.information_type_id = kwargs.get('information_type_id', None)
+        self.is_disabled = None

azure-mgmt-sql/azure/mgmt/sql/models/sensitivity_label_py3.py

@@ -32,12 +32,17 @@ class SensitivityLabel(ProxyResource):
     :type information_type: str
     :param information_type_id: The information type ID.
     :type information_type_id: str
+    :ivar is_disabled: Is sensitivity recommendation disabled. Applicable for
+     recommended sensitivity label only. Specifies whether the sensitivity
+     recommendation on this column is disabled (dismissed) or not.
+    :vartype is_disabled: bool
     """
 
     _validation = {
         'id': {'readonly': True},
         'name': {'readonly': True},
         'type': {'readonly': True},
+        'is_disabled': {'readonly': True},
     }
 
     _attribute_map = {
@@ -48,6 +53,7 @@ class SensitivityLabel(ProxyResource):
         'label_id': {'key': 'properties.labelId', 'type': 'str'},
         'information_type': {'key': 'properties.informationType', 'type': 'str'},
         'information_type_id': {'key': 'properties.informationTypeId', 'type': 'str'},
+        'is_disabled': {'key': 'properties.isDisabled', 'type': 'bool'},
     }
 
     def __init__(self, *, label_name: str=None, label_id: str=None, information_type: str=None, information_type_id: str=None, **kwargs) -> None:
@@ -56,3 +62,4 @@ def __init__(self, *, label_name: str=None, label_id: str=None, information_type
         self.label_id = label_id
         self.information_type = information_type
         self.information_type_id = information_type_id
+        self.is_disabled = None

azure-mgmt-sql/azure/mgmt/sql/operations/managed_database_sensitivity_labels_operations.py

@@ -24,7 +24,6 @@ class ManagedDatabaseSensitivityLabelsOperations(object):
     :param serializer: An object model serializer.
     :param deserializer: An object model deserializer.
     :ivar api_version: The API version to use for the request. Constant value: "2018-06-01-preview".
-    :ivar sensitivity_label_source: The source of the sensitivity label. Constant value: "current".
     """
 
     models = models
@@ -35,7 +34,6 @@ def __init__(self, client, config, serializer, deserializer):
         self._serialize = serializer
         self._deserialize = deserializer
         self.api_version = "2018-06-01-preview"
-        self.sensitivity_label_source = "current"
 
         self.config = config
 
@@ -150,6 +148,8 @@ def create_or_update(
          ~msrest.pipeline.ClientRawResponse
         :raises: :class:`CloudError<msrestazure.azure_exceptions.CloudError>`
         """
+        sensitivity_label_source = "current"
+
         # Construct URL
         url = self.create_or_update.metadata['url']
         path_format_arguments = {
@@ -159,7 +159,7 @@ def create_or_update(
             'schemaName': self._serialize.url("schema_name", schema_name, 'str'),
             'tableName': self._serialize.url("table_name", table_name, 'str'),
             'columnName': self._serialize.url("column_name", column_name, 'str'),
-            'sensitivityLabelSource': self._serialize.url("self.sensitivity_label_source", self.sensitivity_label_source, 'str'),
+            'sensitivityLabelSource': self._serialize.url("sensitivity_label_source", sensitivity_label_source, 'str'),
             'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str')
         }
         url = self._client.format_url(url, **path_format_arguments)
@@ -232,6 +232,8 @@ def delete(
         :rtype: None or ~msrest.pipeline.ClientRawResponse
         :raises: :class:`CloudError<msrestazure.azure_exceptions.CloudError>`
         """
+        sensitivity_label_source = "current"
+
         # Construct URL
         url = self.delete.metadata['url']
         path_format_arguments = {
@@ -241,7 +243,7 @@ def delete(
             'schemaName': self._serialize.url("schema_name", schema_name, 'str'),
             'tableName': self._serialize.url("table_name", table_name, 'str'),
             'columnName': self._serialize.url("column_name", column_name, 'str'),
-            'sensitivityLabelSource': self._serialize.url("self.sensitivity_label_source", self.sensitivity_label_source, 'str'),
+            'sensitivityLabelSource': self._serialize.url("sensitivity_label_source", sensitivity_label_source, 'str'),
             'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str')
         }
         url = self._client.format_url(url, **path_format_arguments)
@@ -273,6 +275,147 @@ def delete(
             return client_raw_response
     delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/managedInstances/{managedInstanceName}/databases/{databaseName}/schemas/{schemaName}/tables/{tableName}/columns/{columnName}/sensitivityLabels/{sensitivityLabelSource}'}
 
+    def disable_recommendation(
+            self, resource_group_name, managed_instance_name, database_name, schema_name, table_name, column_name, custom_headers=None, raw=False, **operation_config):
+        """Disables sensitivity recommendations on a given column.
+
+        :param resource_group_name: The name of the resource group that
+         contains the resource. You can obtain this value from the Azure
+         Resource Manager API or the portal.
+        :type resource_group_name: str
+        :param managed_instance_name: The name of the managed instance.
+        :type managed_instance_name: str
+        :param database_name: The name of the database.
+        :type database_name: str
+        :param schema_name: The name of the schema.
+        :type schema_name: str
+        :param table_name: The name of the table.
+        :type table_name: str
+        :param column_name: The name of the column.
+        :type column_name: str
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: None or ClientRawResponse if raw=true
+        :rtype: None or ~msrest.pipeline.ClientRawResponse
+        :raises: :class:`CloudError<msrestazure.azure_exceptions.CloudError>`
+        """
+        sensitivity_label_source = "recommended"
+
+        # Construct URL
+        url = self.disable_recommendation.metadata['url']
+        path_format_arguments = {
+            'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
+            'managedInstanceName': self._serialize.url("managed_instance_name", managed_instance_name, 'str'),
+            'databaseName': self._serialize.url("database_name", database_name, 'str'),
+            'schemaName': self._serialize.url("schema_name", schema_name, 'str'),
+            'tableName': self._serialize.url("table_name", table_name, 'str'),
+            'columnName': self._serialize.url("column_name", column_name, 'str'),
+            'sensitivityLabelSource': self._serialize.url("sensitivity_label_source", sensitivity_label_source, 'str'),
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        if self.config.generate_client_request_id:
+            header_parameters['x-ms-client-request-id'] = str(uuid.uuid1())
+        if custom_headers:
+            header_parameters.update(custom_headers)
+        if self.config.accept_language is not None:
+            header_parameters['accept-language'] = self._serialize.header("self.config.accept_language", self.config.accept_language, 'str')
+
+        # Construct and send request
+        request = self._client.post(url, query_parameters, header_parameters)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200]:
+            exp = CloudError(response)
+            exp.request_id = response.headers.get('x-ms-request-id')
+            raise exp
+
+        if raw:
+            client_raw_response = ClientRawResponse(None, response)
+            return client_raw_response
+    disable_recommendation.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/managedInstances/{managedInstanceName}/databases/{databaseName}/schemas/{schemaName}/tables/{tableName}/columns/{columnName}/sensitivityLabels/{sensitivityLabelSource}/disable'}
+
+    def enable_recommendation(
+            self, resource_group_name, managed_instance_name, database_name, schema_name, table_name, column_name, custom_headers=None, raw=False, **operation_config):
+        """Enables sensitivity recommendations on a given column (recommendations
+        are enabled by default on all columns).
+
+        :param resource_group_name: The name of the resource group that
+         contains the resource. You can obtain this value from the Azure
+         Resource Manager API or the portal.
+        :type resource_group_name: str
+        :param managed_instance_name: The name of the managed instance.
+        :type managed_instance_name: str
+        :param database_name: The name of the database.
+        :type database_name: str
+        :param schema_name: The name of the schema.
+        :type schema_name: str
+        :param table_name: The name of the table.
+        :type table_name: str
+        :param column_name: The name of the column.
+        :type column_name: str
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: None or ClientRawResponse if raw=true
+        :rtype: None or ~msrest.pipeline.ClientRawResponse
+        :raises: :class:`CloudError<msrestazure.azure_exceptions.CloudError>`
+        """
+        sensitivity_label_source = "recommended"
+
+        # Construct URL
+        url = self.enable_recommendation.metadata['url']
+        path_format_arguments = {
+            'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
+            'managedInstanceName': self._serialize.url("managed_instance_name", managed_instance_name, 'str'),
+            'databaseName': self._serialize.url("database_name", database_name, 'str'),
+            'schemaName': self._serialize.url("schema_name", schema_name, 'str'),
+            'tableName': self._serialize.url("table_name", table_name, 'str'),
+            'columnName': self._serialize.url("column_name", column_name, 'str'),
+            'sensitivityLabelSource': self._serialize.url("sensitivity_label_source", sensitivity_label_source, 'str'),
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        if self.config.generate_client_request_id:
+            header_parameters['x-ms-client-request-id'] = str(uuid.uuid1())
+        if custom_headers:
+            header_parameters.update(custom_headers)
+        if self.config.accept_language is not None:
+            header_parameters['accept-language'] = self._serialize.header("self.config.accept_language", self.config.accept_language, 'str')
+
+        # Construct and send request
+        request = self._client.post(url, query_parameters, header_parameters)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200]:
+            exp = CloudError(response)
+            exp.request_id = response.headers.get('x-ms-request-id')
+            raise exp
+
+        if raw:
+            client_raw_response = ClientRawResponse(None, response)
+            return client_raw_response
+    enable_recommendation.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/managedInstances/{managedInstanceName}/databases/{databaseName}/schemas/{schemaName}/tables/{tableName}/columns/{columnName}/sensitivityLabels/{sensitivityLabelSource}/enable'}
+
     def list_current_by_database(
             self, resource_group_name, managed_instance_name, database_name, filter=None, custom_headers=None, raw=False, **operation_config):
         """Gets the sensitivity labels of a given database.
@@ -354,7 +497,7 @@ def internal_paging(next_link=None, raw=False):
     list_current_by_database.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/managedInstances/{managedInstanceName}/databases/{databaseName}/currentSensitivityLabels'}
 
     def list_recommended_by_database(
-            self, resource_group_name, managed_instance_name, database_name, skip_token=None, filter=None, custom_headers=None, raw=False, **operation_config):
+            self, resource_group_name, managed_instance_name, database_name, include_disabled_recommendations=None, skip_token=None, filter=None, custom_headers=None, raw=False, **operation_config):
         """Gets the sensitivity labels of a given database.
 
         :param resource_group_name: The name of the resource group that
@@ -365,6 +508,9 @@ def list_recommended_by_database(
         :type managed_instance_name: str
         :param database_name: The name of the database.
         :type database_name: str
+        :param include_disabled_recommendations: Specifies whether to include
+         disabled recommendations or not.
+        :type include_disabled_recommendations: bool
         :param skip_token:
         :type skip_token: str
         :param filter: An OData filter expression that filters elements in the
@@ -395,6 +541,8 @@ def internal_paging(next_link=None, raw=False):
 
                 # Construct parameters
                 query_parameters = {}
+                if include_disabled_recommendations is not None:
+                    query_parameters['includeDisabledRecommendations'] = self._serialize.query("include_disabled_recommendations", include_disabled_recommendations, 'bool')
                 if skip_token is not None:
                     query_parameters['$skipToken'] = self._serialize.query("skip_token", skip_token, 'str')
                 if filter is not None: