labnotes 0.0.78__tar.gz → 0.0.79__tar.gz

{labnotes-0.0.78 → labnotes-0.0.79}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.0.78
+Version: 0.0.79
 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.78 → labnotes-0.0.79}/src/labnotes/context.md

@@ -126,6 +126,9 @@ going. Elsewhere, it prints a line per event instead.
   far plus the latest line each running run logged. `-f` keeps following, and
   waits for a daemon when none is up. `--json` prints one JSON object per line.
   Following doesn't keep the daemon up.
+- `labnotes run control NAME ACTION [ARG]` sends one live-view control to a
+  running experiment: `kill`, `retry`, `run_next` or `toggle` a run,
+  `set_concurrency` to a number, `interrupt` or `finish`.
 - A second `run` of the same experiment fails while the first is going.
 - It exits 0 even if runs fail; the summary line counts failures.
 

labnotes-0.0.79/src/labnotes/daemon/control_run.py

@@ -0,0 +1,16 @@
+from pathlib import Path
+
+from labnotes.daemon.daemon_client import daemon_client
+from labnotes.daemon.daemon_running import daemon_running
+
+
+async def control_run(
+    root: Path, name: str, action: str, args: list[object]
+) -> str | None:
+    if not await daemon_running(root):
+        return "Nothing is running."
+    async with daemon_client(root) as client:
+        reply = await client.post(f"/experiments/{name}/{action}", json={"args": args})
+    if reply.is_success:
+        return None
+    return str(reply.json()["error"])

{labnotes-0.0.78 → labnotes-0.0.79}/src/labnotes/run_command.py

@@ -6,6 +6,7 @@ from labnotes.daemon.stop_runs import stop_runs
 from labnotes.load_config import load_config
 from labnotes.read_config import CONFIG_FILE
 from labnotes.run_status import run_status
+from labnotes.run_control import run_control
 
 
 @click.group(name="run", help="Control the runs going in this project.")
@@ -28,3 +29,4 @@ def stop(ctx: click.Context) -> None:
 
 
 run_command.add_command(run_status)
+run_command.add_command(run_control)

labnotes-0.0.79/src/labnotes/run_control.py

@@ -0,0 +1,46 @@
+import asyncio
+import typing
+
+import click
+
+from labnotes.daemon.control_run import control_run
+from labnotes.load_config import load_config
+from labnotes.models import Controls
+from labnotes.read_config import CONFIG_FILE
+
+PARAMS = {
+    action: typing.get_args(hint)[0]
+    for action, hint in typing.get_type_hints(Controls).items()
+}
+
+
+@click.command(
+    name="control",
+    help=(
+        "Send one control to an experiment's runs, as the live view's keys do. "
+        "kill, retry, run_next and toggle take a run; set_concurrency a number; "
+        "interrupt and finish nothing."
+    ),
+)
+@click.argument("name")
+@click.argument("action", type=click.Choice(list(PARAMS)))
+@click.argument("args", nargs=-1)
+@click.pass_context
+def run_control(
+    ctx: click.Context, name: str, action: str, args: tuple[str, ...]
+) -> None:
+    config = load_config(ctx)
+    if config is None:
+        raise click.ClickException(
+            f"No {CONFIG_FILE} here. Run labnotes init, or pass -c PATH."
+        )
+    kinds = PARAMS[action]
+    if len(args) != len(kinds):
+        raise click.UsageError(f"{action} takes {len(kinds)} argument(s).")
+    try:
+        values = [kind(args[i]) for i, kind in enumerate(kinds)]
+    except ValueError:
+        raise click.UsageError(f"{action} takes a number.") from None
+    refused = asyncio.run(control_run(config.root, name, action, values))
+    if refused is not None:
+        raise click.ClickException(refused)

{labnotes-0.0.78 → labnotes-0.0.79}/testing-conventions.toml

@@ -39,3 +39,9 @@ path = "src/labnotes/daemon/remote_start.py"
 rules = ["mutation"]
 lines = [39]
 reason = "the daemon answers a run with 202, 400, 409 or 500, so every status but 202 is above it: != equals >"
+
+[[python.exempt]]
+path = "src/labnotes/run_control.py"
+rules = ["mutation"]
+lines = [38]
+reason = "argument counts are small ints, which CPython caches, so is not equals !="