Added ability to run create-table-mapping command as collection (#2602)

<!-- REMOVE IRRELEVANT COMMENTS BEFORE CREATING A PULL REQUEST -->
## Changes
This PR enables create-table-mapping command to run as a collection.


### Linked issues
<!-- DOC: Link issue with a keyword: close, closes, closed, fix, fixes,
fixed, resolve, resolves, resolved. See
https://docs.github.com/en/issues/tracking-your-work-with-issues/linking-a-pull-request-to-an-issue#linking-a-pull-request-to-an-issue-using-a-keyword
-->
#1782

### Functionality

- [ ] modified existing command: `databricks labs ucx ...`

### Tests
<!-- How is this tested? Please see the checklist below and also
describe any other relevant tests -->

- [ ] added unit tests
- [ ] verified on staging environment (screenshot attached)



https://github.com/user-attachments/assets/a6d5cfe2-05ea-4fd9-b035-bde87aace32a

Author: HariGS-DB

labs.yml

@@ -65,6 +65,9 @@ commands:
 
   - name: create-table-mapping
     description: create initial table mapping for review
+    flags:
+      - name: run-as-collection
+        description: (Optional) boolean flag to indicate to run the cmd as a collection. Default is False.
 
   - name: ensure-assessment-run
     description: ensure the assessment job was run on a workspace

src/databricks/labs/ucx/cli.py

@@ -153,11 +153,21 @@ def manual_workspace_info(w: WorkspaceClient, prompts: Prompts):
 
 
 @ucx.command
-def create_table_mapping(w: WorkspaceClient):
+def create_table_mapping(
+    w: WorkspaceClient,
+    ctx: WorkspaceContext | None = None,
+    run_as_collection: bool = False,
+    a: AccountClient | None = None,
+):
     """create initial table mapping for review"""
-    ctx = WorkspaceContext(w)
-    path = ctx.table_mapping.save(ctx.tables_crawler, ctx.workspace_info)
-    webbrowser.open(f"{w.config.host}/#workspace{path}")
+    workspace_contexts = _get_workspace_contexts(w, a, run_as_collection)
+    if ctx:
+        workspace_contexts = [ctx]
+    for workspace_ctx in workspace_contexts:
+        logger.info(f"Running cmd for workspace {workspace_ctx.workspace_client.get_workspace_id()}")
+        path = workspace_ctx.table_mapping.save(workspace_ctx.tables_crawler, workspace_ctx.workspace_info)
+        if len(workspace_contexts) == 1:
+            webbrowser.open(f"{w.config.host}/#workspace{path}")
 
 
 @ucx.command

tests/unit/test_cli.py

@@ -265,9 +265,10 @@ def test_manual_workspace_info(ws):
     manual_workspace_info(ws, prompts)
 
 
-def test_create_table_mapping(ws):
+def test_create_table_mapping(ws, acc_client):
+    ctx = WorkspaceContext(ws)
     with pytest.raises(ValueError, match='databricks labs ucx sync-workspace-info'):
-        create_table_mapping(ws)
+        create_table_mapping(ws, ctx, False, acc_client)
 
 
 def test_validate_external_locations(ws):