databricks_labs_blueprint_cli_app_with_workspace_client.py

1from databricks.sdk import WorkspaceClient
2from databricks.labs.blueprint.tui import Terminal
3from databricks.labs.blueprint.app import App
4
5class MyTool(App):
6    def __init__(self):
7        super().__init__(name="my-tool")
8
9    def run(self, workspace: WorkspaceClient, terminal: Terminal):
10        terminal.print(f"Connected to {workspace.config.host}")
11        me = workspace.current_user.me()
12        terminal.print(f"Current user: {me.display_name} ({me.user_name})")
13
14if __name__ == "__main__":
15    tool = MyTool()
16    tool.launch()