databricks_labs_blueprint_quickstart_with_tui_and_cluster_listing.py

1from databricks.sdk import WorkspaceClient
2from databricks.labs.blueprint.tui import TUI
3from databricks.labs.blueprint.installation import Installation
4
5# 1. Initialize the Workspace Client (automatically uses environment variables/config)
6workspace = WorkspaceClient()
7
8# 2. Use Blueprint's TUI (Terminal User Interface) helper for logging and interaction
9tui = TUI("my-awesome-tool")
10
11# 3. Access information about the current installation/environment
12install = Installation(workspace)
13
14tui.info(f"Connected to: {workspace.config.host}")
15tui.info(f"Current User: {workspace.current_user.me().display_name}")
16
17# Example: List clusters in the workspace
18clusters = workspace.clusters.list()
19for cluster in clusters:
20    tui.info(f"Found cluster: {cluster.cluster_name} ({cluster.cluster_id})")