Databricks Tools Toolkit

This repository contains a collection of tools and utilities to facilitate working with Databricks. The goal is to centralize scripts and packages that automate common tasks on the platform.

Repository Structure

The project is organized in a modular way to support different tools:

• genie_tools/: Utilities for managing and automating Databricks Genie Spaces.
  • creator.py: Logic for creating, updating, and sorting Genie metadata.
  • utils.py: Helper functions for payload manipulation, Black formatting, and space retrieval.

Installation

To use the tools, install the required dependencies:

pip install -r requirements.txt

How to Use

Genie Tools

The GenieCreator allows for bulk creation and updating of Genie spaces.

from databricks_tools.genie_tools import GenieCreator, GenieMetadata

# Metadata configuration
metadata = GenieMetadata(
    title="My Genie",
    description="Space description",
    raw_serialized_space={...}
)

# Creator initialization
creator = GenieCreator(warehouse_id="your_warehouse_id")
creator.add_metadata(metadata)

# Execution
creator.create_or_update_all_genies()

Contribution

This is an evolving project. Feel free to add new tools to the folder structure.

---

Note: This project is intended for generic use and does not contain proprietary or sensitive information. Make sure to configure your Databricks credentials via environment variables or a .databrickscfg file.