batonsh 0.1.0__py3-none-any.whl

baton/__init__.py

@@ -0,0 +1,54 @@
+"""Python SDK for Baton — drive the multi-agent daemon (batond) from scripts.
+
+    import baton
+
+    with baton.connect() as client:          # starts the project daemon if needed
+        run = client.run_state()
+        seq = run.last_seq
+        client.submit("fix the flaky test in tests/api")
+        run.wait_idle(timeout=600, since_seq=seq)
+        for a in run.agents.values():
+            print(a.name, a.status, a.diff, "$%.2f" % a.cost)
+"""
+
+from ._discovery import discover_socket, find_project_root, project_paths
+from .client import (
+    CONTROL_PROTOCOL_VERSION,
+    BatonError,
+    BatonTimeoutError,
+    Client,
+    Conflict,
+    ConnectionClosed,
+    InvalidParams,
+    NotFound,
+    Subscription,
+    Unavailable,
+    VersionMismatch,
+    connect,
+)
+from .run import BUSY_STATUSES, Agent, Approval, RunState, Workspace
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "CONTROL_PROTOCOL_VERSION",
+    "BUSY_STATUSES",
+    "Agent",
+    "Approval",
+    "BatonError",
+    "BatonTimeoutError",
+    "Client",
+    "ConnectionClosed",
+    "Conflict",
+    "InvalidParams",
+    "NotFound",
+    "Unavailable",
+    "VersionMismatch",
+    "RunState",
+    "Subscription",
+    "Workspace",
+    "connect",
+    "discover_socket",
+    "find_project_root",
+    "project_paths",
+]

baton/_discovery.py

@@ -0,0 +1,50 @@
+"""Project root and socket discovery (mirrors @batonsh/kernel findProjectRoot / projectPaths)."""
+
+import os
+from typing import Dict, Optional
+
+
+def find_project_root(start: Optional[str] = None) -> str:
+    """Walk up from ``start`` looking for ``.baton/config.toml``, then ``.git``; falls back to ``start``."""
+    begin = os.path.abspath(start if start is not None else os.getcwd())
+    d = begin
+    git_root = None
+    while True:
+        if os.path.exists(os.path.join(d, ".baton", "config.toml")):
+            return d
+        if git_root is None and os.path.exists(os.path.join(d, ".git")):
+            git_root = d
+        parent = os.path.dirname(d)
+        if parent == d:
+            break
+        d = parent
+    return git_root if git_root is not None else begin
+
+
+def project_paths(root: str) -> Dict[str, str]:
+    d = os.path.join(root, ".baton")
+    run = os.path.join(d, "run")
+    return {
+        "dir": d,
+        "config": os.path.join(d, "config.toml"),
+        "state": os.path.join(d, "state.db"),
+        "run": run,
+        "socket": os.path.join(run, "baton.sock"),
+        "pid_file": os.path.join(run, "batond.pid"),
+        "log": os.path.join(run, "batond.log"),
+        "worktrees": os.path.join(d, "worktrees"),
+    }
+
+
+def discover_socket(project_root: str) -> str:
+    """The daemon's socket: ``.baton/run/socket-path`` (long project paths) or ``.baton/run/baton.sock``."""
+    paths = project_paths(project_root)
+    alt = os.path.join(paths["run"], "socket-path")
+    try:
+        with open(alt, "r", encoding="utf-8") as f:
+            p = f.read().strip()
+        if p:
+            return p
+    except OSError:
+        pass
+    return paths["socket"]