guardian_ctl.py

"""Command-line status and lifecycle control for the singleton Guardian."""
from __future__ import annotations

import argparse
import json

from guardian_client import ensure_broker, job_status, ping, stop_broker


def main() -> int:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("action", choices=("status", "start", "stop", "job"))
    parser.add_argument("job_id", nargs="?")
    options = parser.parse_args()
    if options.action == "status":
        result = ping() or {"ok": False, "status": "stopped"}
    elif options.action == "start":
        result = ensure_broker()
    elif options.action == "stop":
        result = stop_broker() if ping() else {"ok": True, "status": "already stopped"}
    else:
        if not options.job_id:
            parser.error("job requires a job_id")
        result = job_status(options.job_id)
    print(json.dumps(result, indent=2))
    return 0 if result.get("ok") else 1


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