CodeGen from PR 13683 in Azure/azure-rest-api-specs

Merge 94192ed22e06616c1d57f07f3858833f82f9b726 into 8830e8d
AzureSDKAutomation · Mar 29, 2021 · 6fbadfd · 6fbadfd
1 parent 70bf0f0
commit 6fbadfd
Show file tree

Hide file tree

Showing 4 changed files with 82 additions and 1 deletion.
diff --git a/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_metadata.json b/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_metadata.json
@@ -0,0 +1,63 @@
+{
+    "chosen_version": "2018-04-01",
+    "total_api_version_list": ["2018-04-01"],
+    "client": {
+        "name": "DatabricksClient",
+        "filename": "_databricks_client",
+        "description": "ARM Databricks.",
+        "base_url": "\u0027https://management.azure.com\u0027",
+        "custom_base_url": null,
+        "azure_arm": true,
+        "has_lro_operations": true,
+        "client_side_validation": false
+    },
+    "global_parameters": {
+        "sync": {
+            "credential": {
+                "signature": "credential,  # type: \"TokenCredential\"",
+                "description": "Credential needed for the client to connect to Azure.",
+                "docstring_type": "~azure.core.credentials.TokenCredential",
+                "required": true
+            },
+            "subscription_id": {
+                "signature": "subscription_id,  # type: str",
+                "description": "The ID of the target subscription.",
+                "docstring_type": "str",
+                "required": true
+            }
+        },
+        "async": {
+            "credential": {
+                "signature": "credential,  # type: \"AsyncTokenCredential\"",
+                "description": "Credential needed for the client to connect to Azure.",
+                "docstring_type": "~azure.core.credentials_async.AsyncTokenCredential",
+                "required": true
+            },
+            "subscription_id": {
+                "signature": "subscription_id,  # type: str",
+                "description": "The ID of the target subscription.",
+                "docstring_type": "str",
+                "required": true
+            }
+        },
+        "constant": {
+        },
+        "call": "credential, subscription_id"
+    },
+    "config": {
+        "credential": true,
+        "credential_scopes": ["https://management.azure.com/.default"],
+        "credential_default_policy_type": "BearerTokenCredentialPolicy",
+        "credential_default_policy_type_has_async_version": true,
+        "credential_key_header_name": null
+    },
+    "operation_groups": {
+        "workspaces": "WorkspacesOperations",
+        "vnet_peering": "VNetPeeringOperations",
+        "operations": "Operations"
+    },
+    "operation_mixins": {
+    },
+    "sync_imports": "None",
+    "async_imports": "None"
+}
diff --git a/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_version.py b/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/_version.py
@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "1.0.0"
+VERSION = "0.1.0"
diff --git a/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/_models.py b/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/_models.py
@@ -767,6 +767,10 @@ class WorkspaceCustomParameters(msrest.serialization.Model):
      data at rest.
     :type require_infrastructure_encryption:
      ~azure.mgmt.databricks.models.WorkspaceCustomBooleanParameter
+    :param storage_account_name: Name of the default DBFS storage account name.
+    :type storage_account_name: ~azure.mgmt.databricks.models.WorkspaceCustomStringParameter
+    :param storage_account_sku_name: Storage account SKU name.
+    :type storage_account_sku_name: ~azure.mgmt.databricks.models.WorkspaceCustomStringParameter
     """
 
     _attribute_map = {
@@ -778,6 +782,8 @@ class WorkspaceCustomParameters(msrest.serialization.Model):
         'prepare_encryption': {'key': 'prepareEncryption', 'type': 'WorkspaceCustomBooleanParameter'},
         'encryption': {'key': 'encryption', 'type': 'WorkspaceEncryptionParameter'},
         'require_infrastructure_encryption': {'key': 'requireInfrastructureEncryption', 'type': 'WorkspaceCustomBooleanParameter'},
+        'storage_account_name': {'key': 'storageAccountName', 'type': 'WorkspaceCustomStringParameter'},
+        'storage_account_sku_name': {'key': 'storageAccountSkuName', 'type': 'WorkspaceCustomStringParameter'},
     }
 
     def __init__(
@@ -793,6 +799,8 @@ def __init__(
         self.prepare_encryption = kwargs.get('prepare_encryption', None)
         self.encryption = kwargs.get('encryption', None)
         self.require_infrastructure_encryption = kwargs.get('require_infrastructure_encryption', None)
+        self.storage_account_name = kwargs.get('storage_account_name', None)
+        self.storage_account_sku_name = kwargs.get('storage_account_sku_name', None)
 
 
 class WorkspaceCustomStringParameter(msrest.serialization.Model):

diff --git a/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/_models_py3.py b/sdk/databricks/azure-mgmt-databricks/azure/mgmt/databricks/models/_models_py3.py
@@ -836,6 +836,10 @@ class WorkspaceCustomParameters(msrest.serialization.Model):
      data at rest.
     :type require_infrastructure_encryption:
      ~azure.mgmt.databricks.models.WorkspaceCustomBooleanParameter
+    :param storage_account_name: Name of the default DBFS storage account name.
+    :type storage_account_name: ~azure.mgmt.databricks.models.WorkspaceCustomStringParameter
+    :param storage_account_sku_name: Storage account SKU name.
+    :type storage_account_sku_name: ~azure.mgmt.databricks.models.WorkspaceCustomStringParameter
     """
 
     _attribute_map = {
@@ -847,6 +851,8 @@ class WorkspaceCustomParameters(msrest.serialization.Model):
         'prepare_encryption': {'key': 'prepareEncryption', 'type': 'WorkspaceCustomBooleanParameter'},
         'encryption': {'key': 'encryption', 'type': 'WorkspaceEncryptionParameter'},
         'require_infrastructure_encryption': {'key': 'requireInfrastructureEncryption', 'type': 'WorkspaceCustomBooleanParameter'},
+        'storage_account_name': {'key': 'storageAccountName', 'type': 'WorkspaceCustomStringParameter'},
+        'storage_account_sku_name': {'key': 'storageAccountSkuName', 'type': 'WorkspaceCustomStringParameter'},
     }
 
     def __init__(
@@ -860,6 +866,8 @@ def __init__(
         prepare_encryption: Optional["WorkspaceCustomBooleanParameter"] = None,
         encryption: Optional["WorkspaceEncryptionParameter"] = None,
         require_infrastructure_encryption: Optional["WorkspaceCustomBooleanParameter"] = None,
+        storage_account_name: Optional["WorkspaceCustomStringParameter"] = None,
+        storage_account_sku_name: Optional["WorkspaceCustomStringParameter"] = None,
         **kwargs
     ):
         super(WorkspaceCustomParameters, self).__init__(**kwargs)
@@ -871,6 +879,8 @@ def __init__(
         self.prepare_encryption = prepare_encryption
         self.encryption = encryption
         self.require_infrastructure_encryption = require_infrastructure_encryption
+        self.storage_account_name = storage_account_name
+        self.storage_account_sku_name = storage_account_sku_name
 
 
 class WorkspaceCustomStringParameter(msrest.serialization.Model):