import os
from databricks_cli.workspace.api import WorkspaceApi
from databricks_cli.configure.provider import ProfileConfigProvider
from databricks_cli.sdk.api_client import ApiClient

def export_notebooks(source_path, target_dir, format=“SOURCE”):
“””
Exports all notebooks from a Databricks workspace directory to local files.

Args:
source_path: Path of the directory in the Databricks workspace
target_dir: Local directory to save the notebooks
format: Export format (SOURCE, HTML, JUPYTER)
“””
config = ProfileConfigProvider(“DEFAULT”).get_config()
api_client = ApiClient(host=config.host, token=config.token)
workspace_api = WorkspaceApi(api_client)

# list all objects in the directory
objects = workspace_api.list(source_path)

if not os.path.exists(target_dir):
os.makedirs(target_dir)

for obj in objects:
target_path = os.path.join(target_dir, obj.basename)

if obj.object_type == “DIRECTORY”:
# recursively export subdirectories
export_notebooks(obj.path, target_path, format)
elif obj.object_type == “NOTEBOOK”:
# export the notebook
with open(f”{target_path}.py”, “wb”) as f:
content = workspace_api.export(obj.path, format)
f.write(content)
print(f”Exported: {obj.path} -> {target_path}.py”)

# example usage
export_notebooks(“/Shared/Project”, “./backup_notebooks”)