labnotes 0.0.90__tar.gz → 0.0.92__tar.gz

{labnotes-0.0.90 → labnotes-0.0.92}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.0.90
+Version: 0.0.92
 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.90 → labnotes-0.0.92}/src/labnotes/tui/experiment_screen.py

@@ -16,6 +16,7 @@ from labnotes.tui.invocation_rows import invocation_rows
 from labnotes.tui.reload_on_change import reload_on_change
 from labnotes.tui.run_screen import RunScreen
 from labnotes.tui.runs_table import RunsTable
+from labnotes.tui.stopped_title import stopped_title
 from labnotes.watch_rows import watch_rows
 
 WATCHED = {"runs", "experiment_configs"}
@@ -44,7 +45,9 @@ class ExperimentScreen(Screen[None]):
         self.sub_title = self.experiment_name
         self.run_worker(
             reload_on_change(
-                watch_rows(self.tables, WATCHED, self.experiment_name), self.load
+                watch_rows(self.tables, WATCHED, self.experiment_name),
+                self.load,
+                self.stopped,
             ),
             exit_on_error=False,
         )
@@ -53,6 +56,9 @@ class ExperimentScreen(Screen[None]):
         if event.state == WorkerState.ERROR:
             self.app.exit(event.worker.error)
 
+    def stopped(self, error: BaseException | None) -> None:
+        self.sub_title = stopped_title(self.sub_title, error)
+
     async def load(self) -> None:
         loaded = await load_experiment(
             self.tables,

{labnotes-0.0.90 → labnotes-0.0.92}/src/labnotes/tui/experiments_screen.py

@@ -10,6 +10,7 @@ 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.tui.stopped_title import stopped_title
 from labnotes.watch_rows import watch_rows
 
 WATCHED = {"files", "runs", "experiment_configs"}
@@ -28,7 +29,7 @@ class ExperimentsScreen(Screen[None]):
 
     def on_mount(self) -> None:
         self.run_worker(
-            reload_on_change(watch_rows(self.tables, WATCHED), self.load),
+            reload_on_change(watch_rows(self.tables, WATCHED), self.load, self.stopped),
             exit_on_error=False,
         )
 
@@ -36,6 +37,9 @@ class ExperimentsScreen(Screen[None]):
         if event.state == WorkerState.ERROR:
             self.app.exit(event.worker.error)
 
+    def stopped(self, error: BaseException | None) -> None:
+        self.sub_title = stopped_title(self.sub_title, error)
+
     async def load(self) -> None:
         rows = await summarize_experiments(self.tables)
         self.sub_title = "" if rows else "No experiments."

{labnotes-0.0.90 → labnotes-0.0.92}/src/labnotes/tui/reload_on_change.py

@@ -3,7 +3,9 @@ from collections.abc import AsyncIterable, Awaitable, Callable
 
 
 async def reload_on_change(
-    changes: AsyncIterable[object], reload: Callable[[], Awaitable[None]]
+    changes: AsyncIterable[object],
+    reload: Callable[[], Awaitable[None]],
+    stopped: Callable[[BaseException | None], None],
 ) -> None:
     # Changes that land during a reload collapse into one more reload.
     stale = asyncio.Event()
@@ -13,10 +15,13 @@ async def reload_on_change(
             stale.set()
 
     marker = asyncio.create_task(mark())
+    marker.add_done_callback(lambda _: stale.set())
     try:
-        while True:
+        while not marker.done():
             await reload()
             await stale.wait()
             stale.clear()
+        await reload()
+        stopped(marker.exception())
     finally:
         marker.cancel()

labnotes-0.0.92/src/labnotes/tui/stopped_title.py

@@ -0,0 +1,3 @@
+def stopped_title(title: str | None, error: BaseException | None) -> str:
+    reason = f"Live updates stopped: {error}" if error else "Live updates stopped."
+    return f"{title} · {reason}" if title else reason