Function main

ci/autoresearch/test_flexio_rx_objectfled.py:143–239

(argv: list[str] | None = None)

def main(argv: list[str] | None = None) -> int:
    # `argv` accepted so that unit tests can inject argument lists without
    # touching the real `sys.argv`. Defaults to `None`, which argparse
    # interprets as "use sys.argv[1:]".
    p = argparse.ArgumentParser()
    p.add_argument("--port", default=DEFAULT_PORT)
    p.add_argument("--baud", default=DEFAULT_BAUD, type=int)
    p.add_argument(
        "--tx-pin", default=3, type=int, help="TX pin driven by ObjectFLED (default: 3)"
    )
    p.add_argument(
        "--rx-pin", default=4, type=int, help="RX pin captured by FLEXIO1 (default: 4)"
    )
    p.add_argument(
        "--capture-ms",
        default=50,
        type=int,
        help="Capture window per test case in ms (default: 50)",
    )
    args = p.parse_args(argv)

    if serial is None:
        print(
            "ERROR: pyserial not installed. Run: uv pip install pyserial",
            file=sys.stderr,
        )
        return 2

    print(f"[flexio-objectfled] opening {args.port} @ {args.baud}")
    try:
        s = serial.Serial(args.port, args.baud, timeout=0.1)
    except KeyboardInterrupt as ki:
        # Catch Ctrl-C before the bare `Exception`/`SerialException` block so
        # interactive users can abort cleanly. `handle_keyboard_interrupt`
        # routes the signal through the project-wide interrupt handler
        # (required by the KBI002 lint).
        handle_keyboard_interrupt(ki)
        raise
    except serial.SerialException as e:
        print(f"ERROR: could not open {args.port}: {e}", file=sys.stderr)
        return 2
    with s:
        s.dtr = True
        s.rts = True
        time.sleep(1.0)
        s.reset_input_buffer()

        ping = send_rpc(s, "ping", request_id=1, timeout=5.0)
        if not ping or "result" not in ping:
            print(
                "ERROR: ping failed; is AutoResearch flashed and running?",
                file=sys.stderr,
            )
            return 2
        print(
            f"[flexio-objectfled] ping ok: uptime={ping['result'].get('uptimeMs')} ms"
        )