|
| 1 | +"""Vessim CLI – ``vessim view <results-dir>``.""" |
| 2 | + |
| 3 | +from __future__ import annotations |
| 4 | + |
| 5 | +import argparse |
| 6 | +import functools |
| 7 | +import os |
| 8 | +import sys |
| 9 | +import webbrowser |
| 10 | +from http.server import HTTPServer, SimpleHTTPRequestHandler |
| 11 | +from pathlib import Path |
| 12 | + |
| 13 | + |
| 14 | +def _viewer_dist_dir() -> Path: |
| 15 | + return Path(__file__).parent / "_viewer_dist" |
| 16 | + |
| 17 | + |
| 18 | +class _ViewerHandler(SimpleHTTPRequestHandler): |
| 19 | + """Serves the built viewer app and exposes the results directory under /results/.""" |
| 20 | + |
| 21 | + def __init__(self, *args, viewer_dir: Path, results_dir: Path, **kwargs): |
| 22 | + self.viewer_dir = viewer_dir |
| 23 | + self.results_dir = results_dir |
| 24 | + super().__init__(*args, **kwargs) |
| 25 | + |
| 26 | + def translate_path(self, path: str) -> str: |
| 27 | + # Strip query string and fragment |
| 28 | + path = path.split("?", 1)[0].split("#", 1)[0] |
| 29 | + |
| 30 | + if path.startswith("/results/"): |
| 31 | + rel = path[len("/results/"):] |
| 32 | + return str(self.results_dir / rel) |
| 33 | + if path == "/results": |
| 34 | + return str(self.results_dir) |
| 35 | + |
| 36 | + # Serve from viewer dist |
| 37 | + rel = path.lstrip("/") |
| 38 | + full = self.viewer_dir / rel |
| 39 | + # SPA fallback: if file doesn't exist, serve index.html |
| 40 | + if not full.exists() or full.is_dir(): |
| 41 | + if rel and not (full.exists() and full.is_dir()): |
| 42 | + return str(self.viewer_dir / "index.html") |
| 43 | + return str(full) |
| 44 | + |
| 45 | + def log_message(self, format, *args): |
| 46 | + # Silence request logs |
| 47 | + pass |
| 48 | + |
| 49 | + |
| 50 | +def _cmd_view(args: argparse.Namespace) -> None: |
| 51 | + results_dir = Path(args.directory).resolve() |
| 52 | + if not results_dir.is_dir(): |
| 53 | + print(f"Error: '{results_dir}' is not a directory", file=sys.stderr) |
| 54 | + sys.exit(1) |
| 55 | + |
| 56 | + viewer_dir = _viewer_dist_dir() |
| 57 | + if not (viewer_dir / "index.html").exists(): |
| 58 | + print("Error: viewer assets not found. Rebuild with `npm run build` in viewer/.", |
| 59 | + file=sys.stderr) |
| 60 | + sys.exit(1) |
| 61 | + |
| 62 | + port = args.port |
| 63 | + handler = functools.partial( |
| 64 | + _ViewerHandler, |
| 65 | + viewer_dir=viewer_dir, |
| 66 | + results_dir=results_dir, |
| 67 | + ) |
| 68 | + server = HTTPServer(("localhost", port), handler) |
| 69 | + |
| 70 | + url = f"http://localhost:{port}" |
| 71 | + print(f"Serving experiment viewer at {url}") |
| 72 | + print(f"Results directory: {results_dir}") |
| 73 | + print("Press Ctrl+C to stop.\n") |
| 74 | + |
| 75 | + if not args.no_browser: |
| 76 | + webbrowser.open(url) |
| 77 | + |
| 78 | + try: |
| 79 | + server.serve_forever() |
| 80 | + except KeyboardInterrupt: |
| 81 | + print("\nStopped.") |
| 82 | + server.server_close() |
| 83 | + |
| 84 | + |
| 85 | +def main(): |
| 86 | + parser = argparse.ArgumentParser(prog="vessim", description="Vessim CLI") |
| 87 | + sub = parser.add_subparsers(dest="command") |
| 88 | + |
| 89 | + view = sub.add_parser("view", help="View experiment results in the browser") |
| 90 | + view.add_argument("directory", help="Path to a results directory (with config.yaml + timeseries.csv)") |
| 91 | + view.add_argument("-p", "--port", type=int, default=8710, help="Port (default: 8710)") |
| 92 | + view.add_argument("--no-browser", action="store_true", help="Don't open browser automatically") |
| 93 | + |
| 94 | + args = parser.parse_args() |
| 95 | + if args.command == "view": |
| 96 | + _cmd_view(args) |
| 97 | + else: |
| 98 | + parser.print_help() |
| 99 | + |
| 100 | + |
| 101 | +if __name__ == "__main__": |
| 102 | + main() |
0 commit comments