labnotes 0.0.87__tar.gz → 0.0.88__tar.gz

{labnotes-0.0.87 → labnotes-0.0.88}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.0.87
+Version: 0.0.88
 Summary: Browse experiment designs, notes, and run data from a folder.
 Project-URL: Repository, https://github.com/thekevinscott/labnotes
 Author-email: Kevin Scott <me@thekevinscott.com>

labnotes-0.0.88/src/labnotes/file_changes.py

@@ -0,0 +1,11 @@
+import json
+from collections.abc import AsyncIterator
+
+from dirsql import DirSQL
+
+from labnotes.watch_rows import watch_rows
+
+
+async def file_changes(tables: DirSQL, experiment: str) -> AsyncIterator[str]:
+    async for row in watch_rows(tables, {"files"}, experiment):
+        yield f"data: {json.dumps(row['filename'])}\n\n"

{labnotes-0.0.87 → labnotes-0.0.88}/src/labnotes/tui/experiment_screen.py

@@ -10,9 +10,14 @@ from textual.worker import Worker, WorkerState
 
 from labnotes.load_experiment import load_experiment
 from labnotes.models import LabnotesConfig
+from labnotes.tui.fill_table import fill_table
 from labnotes.tui.invocation_columns import invocation_columns
 from labnotes.tui.invocation_rows import invocation_rows
+from labnotes.tui.reload_on_change import reload_on_change
 from labnotes.tui.runs_table import RunsTable
+from labnotes.watch_rows import watch_rows
+
+WATCHED = {"runs", "experiment_configs"}
 
 
 class ExperimentScreen(Screen[None]):
@@ -36,7 +41,12 @@ class ExperimentScreen(Screen[None]):
 
     def on_mount(self) -> None:
         self.sub_title = self.experiment_name
-        self.run_worker(self.load(), exit_on_error=False)
+        self.run_worker(
+            reload_on_change(
+                watch_rows(self.tables, WATCHED, self.experiment_name), self.load
+            ),
+            exit_on_error=False,
+        )
 
     def on_worker_state_changed(self, event: Worker.StateChanged) -> None:
         if event.state == WorkerState.ERROR:
@@ -50,20 +60,22 @@ class ExperimentScreen(Screen[None]):
             self.experiment_name,
             QueryParams(),
         )
-        if loaded is None:
+        if loaded is None and not self.experiment:
             raise ValueError(f"No experiment named {self.experiment_name}.")
+        if loaded is None:
+            self.sub_title = f"{self.experiment_name} was removed."
+            fill_table(self.query_one(DataTable), [], [])
+            return
         self.experiment = loaded
         self.sub_title = str(loaded["display_name"])
-        self.query_one(DataTable).add_columns(*invocation_columns(loaded))
         self.redraw()
 
     def redraw(self) -> None:
-        table = self.query_one(DataTable)
-        row = table.cursor_row
-        table.clear()
-        for key, cells in invocation_rows(self.experiment, self.opened):
-            table.add_row(*cells, key=key)
-        table.move_cursor(row=row)
+        fill_table(
+            self.query_one(DataTable),
+            invocation_columns(self.experiment),
+            invocation_rows(self.experiment, self.opened),
+        )
 
     def action_toggle_invocation(self) -> None:
         table = self.query_one(DataTable)

{labnotes-0.0.87 → labnotes-0.0.88}/src/labnotes/tui/experiments_screen.py

@@ -7,7 +7,12 @@ from textual.worker import Worker, WorkerState
 from labnotes.models import LabnotesConfig
 from labnotes.tables.summarize_experiments import summarize_experiments
 from labnotes.tui.experiment_screen import ExperimentScreen
+from labnotes.tui.fill_table import fill_table
+from labnotes.tui.reload_on_change import reload_on_change
 from labnotes.tui.runs_table import RunsTable
+from labnotes.watch_rows import watch_rows
+
+WATCHED = {"files", "runs", "experiment_configs"}
 
 
 class ExperimentsScreen(Screen[None]):
@@ -22,7 +27,10 @@ class ExperimentsScreen(Screen[None]):
         yield Footer()
 
     def on_mount(self) -> None:
-        self.run_worker(self.load(), exit_on_error=False)
+        self.run_worker(
+            reload_on_change(watch_rows(self.tables, WATCHED), self.load),
+            exit_on_error=False,
+        )
 
     def on_worker_state_changed(self, event: Worker.StateChanged) -> None:
         if event.state == WorkerState.ERROR:
@@ -30,16 +38,18 @@ class ExperimentsScreen(Screen[None]):
 
     async def load(self) -> None:
         rows = await summarize_experiments(self.tables)
-        if not rows:
-            self.sub_title = "No experiments."
-            return
-        table = self.query_one(DataTable)
-        table.add_columns(*rows[0])
-        for row in rows:
-            table.add_row(
-                *("" if value is None else str(value) for value in row.values()),
-                key=str(row["name"]),
-            )
+        self.sub_title = "" if rows else "No experiments."
+        fill_table(
+            self.query_one(DataTable),
+            list(next(iter(rows), ())),
+            [
+                (
+                    str(row["name"]),
+                    ["" if value is None else str(value) for value in row.values()],
+                )
+                for row in rows
+            ],
+        )
 
     def on_data_table_row_selected(self, event: DataTable.RowSelected) -> None:
         self.app.push_screen(

labnotes-0.0.88/src/labnotes/tui/fill_table.py

@@ -0,0 +1,26 @@
+from collections.abc import Sequence
+
+from textual.widgets import DataTable
+
+
+def fill_table(
+    table: DataTable[object],
+    columns: Sequence[str],
+    rows: Sequence[tuple[str, Sequence[object]]],
+) -> None:
+    # The cursor follows its row by key, so rows arriving above it don't move it.
+    row = table.cursor_row
+    key = (
+        table.coordinate_to_cell_key(table.cursor_coordinate).row_key.value
+        if table.row_count
+        else None
+    )
+    if [str(column.label) for column in table.columns.values()] == list(columns):
+        table.clear()
+    else:
+        table.clear(columns=True)
+        table.add_columns(*columns)
+    keys = [each for each, _ in rows]
+    for each, cells in rows:
+        table.add_row(*cells, key=each)
+    table.move_cursor(row=keys.index(key) if key in keys else row)

labnotes-0.0.88/src/labnotes/tui/reload_on_change.py

@@ -0,0 +1,22 @@
+import asyncio
+from collections.abc import AsyncIterable, Awaitable, Callable
+
+
+async def reload_on_change(
+    changes: AsyncIterable[object], reload: Callable[[], Awaitable[None]]
+) -> None:
+    # Changes that land during a reload collapse into one more reload.
+    stale = asyncio.Event()
+
+    async def mark() -> None:
+        async for _ in changes:
+            stale.set()
+
+    marker = asyncio.create_task(mark())
+    try:
+        while True:
+            await reload()
+            await stale.wait()
+            stale.clear()
+    finally:
+        marker.cancel()

labnotes-0.0.88/src/labnotes/watch_rows.py

@@ -0,0 +1,13 @@
+from collections.abc import AsyncIterator, Collection
+from typing import Any
+
+from dirsql import DirSQL
+
+
+async def watch_rows(
+    tables: DirSQL, names: Collection[str], experiment: str | None = None
+) -> AsyncIterator[dict[str, Any]]:
+    async for event in tables.watch():
+        row = event.row or event.old_row
+        if event.table in names and row and experiment in (None, row["experiment"]):
+            yield row

labnotes-0.0.87/src/labnotes/file_changes.py

@@ -1,11 +0,0 @@
-import json
-from collections.abc import AsyncIterator
-
-from dirsql import DirSQL
-
-
-async def file_changes(tables: DirSQL, experiment: str) -> AsyncIterator[str]:
-    async for event in tables.watch():
-        row = event.row or event.old_row
-        if event.table == "files" and row and row["experiment"] == experiment:
-            yield f"data: {json.dumps(row['filename'])}\n\n"