control section opening from url

Author: djudjuu

dlt/_workspace/helpers/dashboard/dlt_dashboard.py

@@ -1346,33 +1346,45 @@ def ui_controls(mo_cli_arg_with_test_identifiers: bool):
         label="<small>Refresh</small>",
     )
 
+    # read sections from query params to allow navigation via URL
+    # use #<cell_name> hash to scroll (e.g. #section_trace) — marimo scrolls to named cells
+    _open_sections = (cast(str, mo.query_params().get("sections")) or "").split(",")
+
     # page switches
     dlt_section_sync_switch: mo.ui.switch = mo.ui.switch(
         value=True, label="sync" if mo_cli_arg_with_test_identifiers else ""
     )
     dlt_section_info_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="overview" if mo_cli_arg_with_test_identifiers else ""
+        value="overview" in _open_sections,
+        label="overview" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_schema_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="schema" if mo_cli_arg_with_test_identifiers else ""
+        value="schema" in _open_sections,
+        label="schema" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_browse_data_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="data" if mo_cli_arg_with_test_identifiers else ""
+        value="data" in _open_sections,
+        label="data" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_state_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="state" if mo_cli_arg_with_test_identifiers else ""
+        value="state" in _open_sections,
+        label="state" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_trace_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="trace" if mo_cli_arg_with_test_identifiers else ""
+        value="trace" in _open_sections,
+        label="trace" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_loads_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="loads" if mo_cli_arg_with_test_identifiers else ""
+        value="loads" in _open_sections,
+        label="loads" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_ibis_browser_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="ibis" if mo_cli_arg_with_test_identifiers else ""
+        value="ibis" in _open_sections,
+        label="ibis" if mo_cli_arg_with_test_identifiers else "",
     )
     dlt_section_data_quality_switch: mo.ui.switch = mo.ui.switch(
-        value=False, label="data_quality" if mo_cli_arg_with_test_identifiers else ""
+        value="data_quality" in _open_sections,
+        label="data_quality" if mo_cli_arg_with_test_identifiers else "",
     )
 
     # other switches

tests/e2e/helpers/dashboard/test_e2e.py

@@ -412,3 +412,43 @@ def test_broken_trace_pipeline(page: Page, broken_trace_pipeline: Any, pipelines
     # should also render the trace section, but there should be an error message
     _open_section(page, "trace")
     expect(page.get_by_text("Error while building trace section:")).to_be_visible()
+
+
+def test_sections_query_param(page: Page, fruit_pipeline: Any):
+    """Sections specified in ?sections= query param should be pre-opened."""
+    # navigate with sections=trace,loads in the URL
+    page.goto("http://localhost:2718/?pipeline=fruit_pipeline&sections=trace,loads")
+
+    # wait for the pipeline to load
+    expect(page.get_by_role("switch", name="trace")).to_be_visible(timeout=20000)
+
+    # trace and loads switches should be checked
+    expect(page.get_by_role("switch", name="trace")).to_be_checked()
+    expect(page.get_by_role("switch", name="loads")).to_be_checked()
+
+    # other sections should NOT be checked
+    expect(page.get_by_role("switch", name="overview")).not_to_be_checked()
+    expect(page.get_by_role("switch", name="schema")).not_to_be_checked()
+    expect(page.get_by_role("switch", name="data")).not_to_be_checked()
+    expect(page.get_by_role("switch", name="state")).not_to_be_checked()
+
+    # verify the trace section content is actually visible
+    expect(page.get_by_text(app_strings.trace_subtitle)).to_be_visible()
+
+    # verify loads section content is visible
+    expect(page.get_by_role("row", name="fruitshop").nth(0)).to_be_visible()
+
+
+def test_sections_query_param_all(page: Page, fruit_pipeline: Any):
+    """All sections should open when all are specified in ?sections= query param."""
+    page.goto(
+        "http://localhost:2718/?pipeline=fruit_pipeline"
+        "&sections=overview,schema,data,state,trace,loads"
+    )
+
+    # wait for the pipeline to load
+    expect(page.get_by_role("switch", name="overview")).to_be_visible(timeout=20000)
+
+    # all specified switches should be checked
+    for section in ["overview", "schema", "data", "state", "trace", "loads"]:
+        expect(page.get_by_role("switch", name=section)).to_be_checked()