78 lines
2.9 KiB
Python
78 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Host probe v3 — socket-id variants via the library business API.
|
|
|
|
Historical sweep that established the socket-transport command space
|
|
(``1101/1307/1705/1707`` instead of the ``off_8`` ids). Superseded by
|
|
``host_probe6.py`` for routine use; kept as a variant-list driver over
|
|
``LocalDeviceClient.send_probe``.
|
|
|
|
Run from the repo root on the SoftAP-joined host:
|
|
python3 scripts/host_probe3.py --sn A03X2516013210060005 --ports 9000
|
|
"""
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import socket
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(
|
|
os.path.abspath(__file__))))
|
|
|
|
from franklinwh_agate_client.battery import extract_soc # noqa: E402
|
|
from franklinwh_agate_client.device_local import ( # noqa: E402
|
|
LocalDeviceClient,
|
|
)
|
|
from franklinwh_agate_client.protocol import ( # noqa: E402
|
|
CommunicationCmd,
|
|
generate_secret_key,
|
|
)
|
|
|
|
VARIANTS = (
|
|
# (label, cmd, data, equip_no override or None for the client SN)
|
|
("queryInfo/1101+SN", CommunicationCmd.QUERY_DEVICE_INFO,
|
|
{"minProtocolVer": "V1.00.00"}, None),
|
|
("queryInfo/1101+empty", CommunicationCmd.QUERY_DEVICE_INFO,
|
|
{"minProtocolVer": "V1.00.00"}, ""),
|
|
("apower/1307+update2", CommunicationCmd.DEVICE_APOWER_INFO,
|
|
{"update": 2}, None),
|
|
("apower/1307+none", CommunicationCmd.DEVICE_APOWER_INFO, None, None),
|
|
("bms/1705", CommunicationCmd.BMS, None, None),
|
|
("ems/1707", CommunicationCmd.EMS, None, None),
|
|
)
|
|
|
|
|
|
def main():
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("--sn", required=True)
|
|
ap.add_argument("--host", default="10.100.1.1")
|
|
ap.add_argument("--ports", default="9000")
|
|
ap.add_argument("--timeout", type=float, default=5.0)
|
|
args = ap.parse_args()
|
|
print("secret key for %s: %s (radix 16)"
|
|
% (args.sn, generate_secret_key(args.sn)))
|
|
for port in [int(p) for p in args.ports.split(",") if p.strip()]:
|
|
client = LocalDeviceClient(args.sn, host=args.host,
|
|
socket_port=port, timeout=args.timeout)
|
|
for label, cmd, data, equip in VARIANTS:
|
|
try:
|
|
out = client.send_probe(cmd, data=data, timeout=args.timeout,
|
|
equip_no=equip)
|
|
except (ConnectionError, socket.error, TimeoutError) as exc:
|
|
print("--- %s: error: %s: %s"
|
|
% (label, type(exc).__name__, exc))
|
|
continue
|
|
rx = out.get("reply_raw") or b""
|
|
print("--- %s tx=%dB rx=%dB"
|
|
% (label, len(out["tx"]), len(rx)))
|
|
if out.get("reply") is not None:
|
|
print(" reply: "
|
|
+ json.dumps(out["reply"], default=str)[:1500])
|
|
soc = extract_soc(out["reply"])
|
|
print(" soc: %s (source=%s)" % (soc["value"],
|
|
soc["source"]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|