Updated the Az Databricks cli to 2024-05-01 api version. (#7707)

* Updated the Az Databricks cli to 2024-05-01 api version. * updated the version * Updated the Databricks Az Cli cmdlets to 2024-05-01 api version. * Updated the version. Fixed linter errors. * Update the History file. * fixed PR check error --------- Co-authored-by: root <root@DESKTOP-ADSN5Q0>
Azure · Jul 2, 2024 · 1be7674 · 1be7674
1 parent 95ec6a8
commit 1be7674
Show file tree

Hide file tree

Showing 46 changed files with 1,045 additions and 244 deletions.
diff --git a/linter_exclusions.yml b/linter_exclusions.yml
@@ -438,6 +438,16 @@ databox job mark-devices-shipped:
     deliver_package_details:
       rule_exclusions:
         - option_length_too_long
+databricks access-connector create:
+  parameters:
+    user_assigned_identities:
+      rule_exclusions:
+      - option_length_too_long
+databricks access-connector update:
+  parameters:
+    user_assigned_identities:
+      rule_exclusions:
+      - option_length_too_long
 databricks workspace create:
   parameters:
     managed_resource_group:
@@ -446,6 +456,20 @@ databricks workspace create:
     require_infrastructure_encryption:
       rule_exclusions:
       - option_length_too_long
+    default_storage_firewall:
+      rule_exclusions:
+        - option_length_too_long
+    enhanced_security_compliance:
+      rule_exclusions:
+        - option_length_too_long
+databricks workspace update:
+  parameters:
+    default_storage_firewall:
+      rule_exclusions:
+        - option_length_too_long
+    enhanced_security_compliance:
+      rule_exclusions:
+        - option_length_too_long
 databricks workspace vnet-peering create:
   parameters:
     allow_forwarded_traffic:

diff --git a/src/databricks/HISTORY.rst b/src/databricks/HISTORY.rst
@@ -2,6 +2,13 @@
 
 Release History
 ===============
+1.0.0
++++++
+az databricks workspace create/update: Add --access-connector to associate an Access Connector Resource with workspace.
+az databricks workspace create/update: Add --default-storage-firewall to set default storage firewall configuration information on workspace.
+az databricks workspace create/update: Add --enhanced-security-compliance to enable the Enhanced Security and Compliance on workspace.
+az databricks workspace delete: Add --force-deletion to delete all data on Uc enabled workspace.
+
 0.10.2
 +++++
 * az databricks workspace create/update: Add --disk-key-auto-rotation to enable the latest key version should be automatically.

diff --git a/src/databricks/azext_databricks/aaz/latest/__init__.py b/src/databricks/azext_databricks/aaz/latest/__init__.py
@@ -4,3 +4,7 @@
 #
 # Code generated by aaz-dev-tools
 # --------------------------------------------------------------------------------------------
+
+# pylint: skip-file
+# flake8: noqa
+
diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_create.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_create.py
@@ -19,15 +19,12 @@ class Create(AAZCommand):
 
     :example: Create a databricks accessConnector
         az databricks access-connector create --resource-group MyResourceGroup --name MyAccessConnector --location westus --identity-type SystemAssigned
-
-    :example: Create a databricks accessConnector with identities
-        az databricks access-connector create --resource-group MyResourceGroup --name MyAccessConnector --location westus --identity-type UserAssigned --user-assigned-identities {"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}":{}}
     """
 
     _aaz_info = {
-        "version": "2022-10-01-preview",
+        "version": "2024-05-01",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2024-05-01"],
         ]
     }
 
@@ -176,7 +173,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -292,6 +289,13 @@ def _build_schema_on_200_201(cls):
                 serialized_name="provisioningState",
                 flags={"read_only": True},
             )
+            properties.refered_by = AAZListType(
+                serialized_name="referedBy",
+                flags={"read_only": True},
+            )
+
+            refered_by = cls._schema_on_200_201.properties.refered_by
+            refered_by.Element = AAZStrType()
 
             system_data = cls._schema_on_200_201.system_data
             system_data.created_at = AAZStrType(

diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_delete.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_delete.py
@@ -22,9 +22,9 @@ class Delete(AAZCommand):
     """
 
     _aaz_info = {
-        "version": "2022-10-01-preview",
+        "version": "2024-05-01",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2024-05-01"],
         ]
     }
 
@@ -146,7 +146,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }

diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_list.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_list.py
@@ -25,13 +25,15 @@ class List(AAZCommand):
     """
 
     _aaz_info = {
-        "version": "2022-10-01-preview",
+        "version": "2024-05-01",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/accessconnectors", "2022-10-01-preview"],
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/providers/microsoft.databricks/accessconnectors", "2024-05-01"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors", "2024-05-01"],
         ]
     }
 
+    AZ_SUPPORT_PAGINATION = True
+
     def _handler(self, command_args):
         super()._handler(command_args)
         return self.build_paging(self._execute_operations, self._output)
@@ -117,7 +119,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -213,6 +215,13 @@ def _build_schema_on_200(cls):
                 serialized_name="provisioningState",
                 flags={"read_only": True},
             )
+            properties.refered_by = AAZListType(
+                serialized_name="referedBy",
+                flags={"read_only": True},
+            )
+
+            refered_by = cls._schema_on_200.value.Element.properties.refered_by
+            refered_by.Element = AAZStrType()
 
             system_data = cls._schema_on_200.value.Element.system_data
             system_data.created_at = AAZStrType(
@@ -279,7 +288,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -375,6 +384,13 @@ def _build_schema_on_200(cls):
                 serialized_name="provisioningState",
                 flags={"read_only": True},
             )
+            properties.refered_by = AAZListType(
+                serialized_name="referedBy",
+                flags={"read_only": True},
+            )
+
+            refered_by = cls._schema_on_200.value.Element.properties.refered_by
+            refered_by.Element = AAZStrType()
 
             system_data = cls._schema_on_200.value.Element.system_data
             system_data.created_at = AAZStrType(

diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_show.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_show.py
@@ -22,9 +22,9 @@ class Show(AAZCommand):
     """
 
     _aaz_info = {
-        "version": "2022-10-01-preview",
+        "version": "2024-05-01",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2024-05-01"],
         ]
     }
 
@@ -124,7 +124,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -211,6 +211,13 @@ def _build_schema_on_200(cls):
                 serialized_name="provisioningState",
                 flags={"read_only": True},
             )
+            properties.refered_by = AAZListType(
+                serialized_name="referedBy",
+                flags={"read_only": True},
+            )
+
+            refered_by = cls._schema_on_200.properties.refered_by
+            refered_by.Element = AAZStrType()
 
             system_data = cls._schema_on_200.system_data
             system_data.created_at = AAZStrType(

diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_update.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_update.py
@@ -19,15 +19,12 @@ class Update(AAZCommand):
 
     :example: Update a databricks accessConnector
         az databricks access-connector update --resource-group MyResourceGroup --name MyAccessConnector --location westus --identity-type SystemAssigned
-
-    :example: Update a databricks accessConnector with identities
-        az databricks access-connector update --resource-group MyResourceGroup --name MyAccessConnector --identity-type UserAssigned --user-assigned-identities {"/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}":{}}
     """
 
     _aaz_info = {
-        "version": "2022-10-01-preview",
+        "version": "2024-05-01",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2024-05-01"],
         ]
     }
 
@@ -174,7 +171,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -273,7 +270,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -433,6 +430,13 @@ def _build_schema_access_connector_read(cls, _schema):
             serialized_name="provisioningState",
             flags={"read_only": True},
         )
+        properties.refered_by = AAZListType(
+            serialized_name="referedBy",
+            flags={"read_only": True},
+        )
+
+        refered_by = _schema_access_connector_read.properties.refered_by
+        refered_by.Element = AAZStrType()
 
         system_data = _schema_access_connector_read.system_data
         system_data.created_at = AAZStrType(

diff --git a/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_wait.py b/src/databricks/azext_databricks/aaz/latest/databricks/access_connector/_wait.py
@@ -20,7 +20,7 @@ class Wait(AAZWaitCommand):
 
     _aaz_info = {
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2022-10-01-preview"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.databricks/accessconnectors/{}", "2024-05-01"],
         ]
     }
 
@@ -120,7 +120,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2022-10-01-preview",
+                    "api-version", "2024-05-01",
                     required=True,
                 ),
             }
@@ -207,6 +207,13 @@ def _build_schema_on_200(cls):
                 serialized_name="provisioningState",
                 flags={"read_only": True},
             )
+            properties.refered_by = AAZListType(
+                serialized_name="referedBy",
+                flags={"read_only": True},
+            )
+
+            refered_by = cls._schema_on_200.properties.refered_by
+            refered_by.Element = AAZStrType()
 
             system_data = cls._schema_on_200.system_data
             system_data.created_at = AAZStrType(