databricks-labs-blueprint

import os from databricks.sdk import WorkspaceClient from databricks.labs.blueprint.paths import WorkspacePath # Ensure DATABRICKS_HOST and DATABRICKS_TOKEN environment variables are set, # or a Databricks CLI profile is configured. # For local testing, you might run: # export DATABRICKS_HOST='https://<your-databricks-instance>.cloud.databricks.com' # export DATABRICKS_TOKEN='dapi...' if not os.environ.get('DATABRICKS_HOST') or not os.environ.get('DATABRICKS_TOKEN'): print("Please set DATABRICKS_HOST and DATABRICKS_TOKEN environment variables or configure Databricks CLI.") # In a real quickstart, you might exit or raise an error here. # For demonstration, we'll use placeholder values that will likely fail. ws = WorkspaceClient(host=os.environ.get('DATABRICKS_HOST', 'https://example.cloud.databricks.com'), token=os.environ.get('DATABRICKS_TOKEN', 'dapi-fake-token')) else: ws = WorkspaceClient() print(f"Initialized WorkspaceClient for host: {ws.host}") try: user_name = ws.current_user.me().user_name print(f"Current user: {user_name}") # Example: Working with user home folders folder_name = 'blueprint-test-folder' wsp = WorkspacePath(ws, f"~/{{folder_name}}/sub/dir") # Expand the user path and create the directory with_user = wsp.expanduser() print(f"Expanded path: {with_user}") with_user.mkdir() print(f"Directory '{with_user}' created.") # Verify existence assert with_user.is_dir() print(f"Directory '{with_user}' exists.") # Clean up (recursive rmdir) with_user.parent.parent.rmdir(recursive=True) assert not with_user.parent.parent.exists() print(f"Directory '{with_user.parent.parent}' and its contents removed.") except Exception as e: print(f"An error occurred: {e}") print("Please ensure your Databricks environment is correctly configured (host, token, permissions).")