labnotes 0.0.95__tar.gz → 0.0.96__tar.gz

{labnotes-0.0.95 → labnotes-0.0.96}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.0.95
+Version: 0.0.96
 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.96/src/labnotes/tui/run_output.py

@@ -0,0 +1,61 @@
+import codecs
+from pathlib import Path
+
+from rich.text import Text
+from textual.geometry import Size
+from textual.widgets import RichLog
+
+STREAMS = ("stdout", "stderr")
+
+
+class RunOutput(RichLog):
+    def __init__(self, id: str | None = None) -> None:
+        super().__init__(id=id)
+        self.reset()
+
+    def reset(self) -> None:
+        self.offsets = dict.fromkeys(STREAMS, 0)
+        self.decoders = {
+            stream: codecs.getincrementaldecoder("utf-8")(errors="replace")
+            for stream in STREAMS
+        }
+        self.last_stream: str | None = None
+        self.unfinished = ""
+        self.clear()
+
+    @property
+    def mid_line(self) -> bool:
+        return bool(self.unfinished)
+
+    def follow(self, folder: Path) -> None:
+        for stream in STREAMS:
+            path = folder / f"{stream}.log"
+            if not path.exists():
+                continue
+            with path.open("rb") as log:
+                log.seek(self.offsets[stream])
+                data = log.read()
+            self.offsets[stream] += len(data)
+            text = self.decoders[stream].decode(data)
+            if not text:
+                continue
+            # A stream's unfinished line is redrawn as it grows, but the
+            # other stream's output starts below it.
+            if self.mid_line and stream == self.last_stream:
+                self.drop_last_line()
+                text = self.unfinished + text
+            *complete, self.unfinished = text.split("\n")
+            if complete:
+                self.write(Text.from_ansi("\n".join(complete)))
+            if self.unfinished:
+                self.write(Text.from_ansi(self.unfinished))
+            self.last_stream = stream
+
+    def drop_last_line(self) -> None:
+        if not self._size_known:
+            self._deferred_renders.pop()
+            return
+        del self.lines[-1]
+        self._line_cache.clear()
+        self.virtual_size = Size(self.virtual_size.width, len(self.lines))
+        self.refresh()

{labnotes-0.0.95 → labnotes-0.0.96}/testing-conventions.toml

@@ -7,8 +7,8 @@ reason = "LOCK_EX and LOCK_NB are disjoint bits, so | equals + and ^"
 [[python.exempt]]
 path = "src/labnotes/tui/run_output.py"
 rules = ["mutation"]
-lines = [37]
-reason = "last_stream is always the same str object taken from STREAMS, so is not equals !="
+lines = [44]
+reason = "last_stream is always the same str object taken from STREAMS, so is equals =="
 
 [[python.exempt]]
 path = "src/labnotes/load_experiment.py"

labnotes-0.0.95/src/labnotes/tui/run_output.py

@@ -1,41 +0,0 @@
-import codecs
-from pathlib import Path
-
-from textual.widgets import Log
-
-STREAMS = ("stdout", "stderr")
-
-
-class RunOutput(Log):
-    def __init__(self, id: str | None = None) -> None:
-        super().__init__(id=id)
-        self.reset()
-
-    def reset(self) -> None:
-        self.offsets = dict.fromkeys(STREAMS, 0)
-        self.decoders = {
-            stream: codecs.getincrementaldecoder("utf-8")(errors="replace")
-            for stream in STREAMS
-        }
-        self.last_stream: str | None = None
-        self.mid_line = False
-        self.clear()
-
-    def follow(self, folder: Path) -> None:
-        for stream in STREAMS:
-            path = folder / f"{stream}.log"
-            if not path.exists():
-                continue
-            with path.open("rb") as log:
-                log.seek(self.offsets[stream])
-                data = log.read()
-            self.offsets[stream] += len(data)
-            text = self.decoders[stream].decode(data)
-            if not text:
-                continue
-            # Don't glue one stream's unfinished line onto the other's output.
-            if self.mid_line and stream != self.last_stream:
-                text = "\n" + text
-            self.write(text)
-            self.last_stream = stream
-            self.mid_line = not text.endswith("\n")