A Python, object-oriented wrapper for the Azure Databricks REST API 2.0

This package is pip installable.

pip install azure-databricks-api

As of October 17th, 2019 there are 10 different services available in the Azure Databricks API. Currently, the following services are supported by the Azure Databricks API Wrapper.

• Clusters
• Groups
• Token
• Workspace
• DBFS
• Libraries
• Jobs
• Instance Pools
• Secrets
• SCIM (Preview)

To create the client object, you pass the Azure region your workspace is located in and the generated Personal Access Token

from azure_databricks_api import AzureDatabricksRESTClient

azure_region = 'centralus'
token = '[INSERT YOUR PERSONAL ACCESS TOKEN]' 

client = AzureDatabricksRESTClient(region=azure_region, token=token)

The services above are implements as children objects of the client. For example, to pin a cluster, you can either pass the cluster_name or cluster_id:

client.clusters.pin('test_cluster_name')

The other services are implemented similarly. (e.g. client.tokens or client.groups)