Azure Databricks API Wrapper

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

Installation

This package is pip installable.

pip install azure-databricks-api

Implemented APIs

As of September 19th, 2018 there are 9 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
• Jobs
• Libraries
• Secrets
• SCIM (Preview)

Client Instantiation

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)

Clusters Client Usage

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)