71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Pull state, config, or info from a WLED device.
|
||
|
|
Uses only standard library (no pip install).
|
||
|
|
"""
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
import sys
|
||
|
|
from typing import Optional
|
||
|
|
from urllib.request import Request, urlopen
|
||
|
|
from urllib.error import URLError, HTTPError
|
||
|
|
|
||
|
|
DEFAULT_HOST = "192.168.240.30"
|
||
|
|
ENDPOINTS = ("state", "info", "cfg", "eff", "pal", "full")
|
||
|
|
|
||
|
|
|
||
|
|
def pull(host: str, endpoint: str, save_path: Optional[str], indent: Optional[int]) -> dict:
|
||
|
|
base = host if host.startswith("http") else f"http://{host}"
|
||
|
|
url = f"{base}/json" if endpoint == "full" else f"{base}/json/{endpoint}"
|
||
|
|
|
||
|
|
req = Request(url, headers={"Accept": "application/json"})
|
||
|
|
try:
|
||
|
|
with urlopen(req, timeout=10) as r:
|
||
|
|
data = json.loads(r.read().decode())
|
||
|
|
except HTTPError as e:
|
||
|
|
print(f"HTTP error: {e.code} {e.reason}", file=sys.stderr)
|
||
|
|
if e.fp:
|
||
|
|
print(e.fp.read().decode(), file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
except URLError as e:
|
||
|
|
print(f"Request failed: {e.reason}", file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
except json.JSONDecodeError as e:
|
||
|
|
print(f"Invalid JSON: {e}", file=sys.stderr)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if save_path:
|
||
|
|
with open(save_path, "w") as f:
|
||
|
|
json.dump(data, f, indent=indent or 2)
|
||
|
|
print(f"Saved to {save_path}")
|
||
|
|
else:
|
||
|
|
print(json.dumps(data, indent=indent or 2))
|
||
|
|
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
p = argparse.ArgumentParser(description="Pull state/config from WLED")
|
||
|
|
p.add_argument("--host", "-H", default=DEFAULT_HOST, help=f"WLED host (default: {DEFAULT_HOST})")
|
||
|
|
p.add_argument(
|
||
|
|
"endpoint",
|
||
|
|
nargs="?",
|
||
|
|
default="full",
|
||
|
|
choices=ENDPOINTS,
|
||
|
|
help="state, info, cfg, eff, pal, or full (default: full)",
|
||
|
|
)
|
||
|
|
p.add_argument("--save", "-o", metavar="FILE", help="Write JSON to file")
|
||
|
|
p.add_argument("--no-indent", action="store_true", help="Compact JSON output")
|
||
|
|
args = p.parse_args()
|
||
|
|
|
||
|
|
pull(
|
||
|
|
args.host,
|
||
|
|
args.endpoint,
|
||
|
|
args.save,
|
||
|
|
None if args.no_indent else 2,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|