64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
WeConnect push-server test client.
|
|
|
|
Connects to the collect.py TCP push server and prints each incoming
|
|
snapshot to stdout as pretty-printed JSON.
|
|
|
|
Usage
|
|
-----
|
|
python client.py
|
|
python client.py --host 192.168.1.10 --port 9999
|
|
"""
|
|
|
|
import argparse
|
|
import json
|
|
import socket
|
|
import sys
|
|
|
|
|
|
def main() -> None:
|
|
p = argparse.ArgumentParser(
|
|
description=__doc__,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
p.add_argument("--host", default="127.0.0.1", help="Server host (default: 127.0.0.1)")
|
|
p.add_argument("--port", type=int, default=9999, help="Server port (default: 9999)")
|
|
args = p.parse_args()
|
|
|
|
print(f"Connecting to {args.host}:{args.port} …", flush=True)
|
|
try:
|
|
sock = socket.create_connection((args.host, args.port))
|
|
except OSError as e:
|
|
print(f"Connection failed: {e}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
print("Connected. Waiting for data (Ctrl-C to quit).\n", flush=True)
|
|
buf = ""
|
|
count = 0
|
|
try:
|
|
with sock:
|
|
while True:
|
|
chunk = sock.recv(4096).decode()
|
|
if not chunk:
|
|
print("Server closed the connection.")
|
|
break
|
|
buf += chunk
|
|
while "\n" in buf:
|
|
line, buf = buf.split("\n", 1)
|
|
line = line.strip()
|
|
if not line:
|
|
continue
|
|
try:
|
|
data = json.loads(line)
|
|
count += 1
|
|
print(json.dumps(data, indent=2))
|
|
print(f"--- #{count} ", "-" * 50, flush=True)
|
|
except json.JSONDecodeError as e:
|
|
print(f"[invalid JSON] {e}: {line!r}", file=sys.stderr)
|
|
except KeyboardInterrupt:
|
|
print("\nStopped by user.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |