rustwright-core 0.1.1

Rust CDP core for a Python Playwright-compatible automation API
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

#!/usr/bin/env python3
from __future__ import annotations

import argparse
import subprocess
import sys
from pathlib import Path


ROOT = Path(__file__).resolve().parents[2]


def main() -> int:
    parser = argparse.ArgumentParser(description="Run JetStream through the local Crossbench wrapper.")
    parser.add_argument("--browser", default="chrome-stable")
    parser.add_argument("--repeat", type=int, default=20)
    parser.add_argument("--dry-run", action="store_true")
    parser.add_argument("--crossbench-arg", action="append", default=[])
    parser.add_argument("--timeout", type=int, default=None)
    args = parser.parse_args()

    command = [
        sys.executable,
        str(ROOT / "benchmarks" / "crossbench" / "run.py"),
        "--benchmark",
        "jetstream",
        "--browser",
        args.browser,
        "--repeat",
        str(args.repeat),
    ]
    if args.dry_run:
        command.append("--dry-run")
    for item in args.crossbench_arg:
        command.append(f"--crossbench-arg={item}")
    if args.timeout is not None:
        command.extend(["--timeout", str(args.timeout)])
    return subprocess.call(command, cwd=ROOT)


if __name__ == "__main__":
    raise SystemExit(main())