1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# 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 azure.identity import DefaultAzureCredential
from azure.mgmt.databricks import AzureDatabricksManagementClient
"""
# PREREQUISITES
pip install azure-identity
pip install azure-mgmt-databricks
# USAGE
python workspace_managed_disk_encryption_update.py
Before run the sample, please set the values of the client ID, tenant ID and client secret
of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID,
AZURE_CLIENT_SECRET. For more info about how to get the value, please see:
https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal
"""
def main():
client = AzureDatabricksManagementClient(
credential=DefaultAzureCredential(),
subscription_id="subid",
)
response = client.workspaces.begin_create_or_update(
resource_group_name="rg",
workspace_name="myWorkspace",
parameters={
"location": "westus",
"properties": {
"encryption": {
"entities": {
"managedDisk": {
"keySource": "Microsoft.Keyvault",
"keyVaultProperties": {
"keyName": "test-cmk-key",
"keyVaultUri": "https://test-vault-name.vault.azure.net/",
"keyVersion": "00000000000000000000000000000000",
},
"rotationToLatestKeyVersionEnabled": True,
}
}
},
"managedResourceGroupId": "/subscriptions/subid/resourceGroups/myManagedRG",
},
"tags": {"mytag1": "myvalue1"},
},
).result()
print(response)
# x-ms-original-file: specification/databricks/resource-manager/Microsoft.Databricks/preview/2022-04-01-preview/examples/WorkspaceManagedDiskEncryptionUpdate.json
if __name__ == "__main__":
main()
|
