readregs.py

#!/usr/bin/python3
# information gathering for saj inverters through data_transmission mqtt topic
# usage example: ./readregs.py 192.168.16.1 H1S2602J2119E01121 0x4000 0x100 2>/dev/null | python3 saj.py -p

from sajmqtt import SajMqtt
from sys import argv, stdout, stderr
import logging

FORMAT = '%(asctime)s %(message)s'
logging.basicConfig(format=FORMAT)
logging.getLogger().setLevel(logging.DEBUG)


def normalize_hex(argument: str) -> int:

    base = 10
    start = 0

    if argument[:2] == "0x":
        base = 16
        start = 2

    return int(argument[start:], base)


if len(argv) < 5:
    print("Usage: %s <broker_ip> <serial> <register_start> <register_count>" % (argv[0],))
    print("Example: %s 192.168.1.30 H1S267K2429B029410 0x3200 0x80 > data.bin" % (argv[0],))
    exit()

broker_ip = argv[1]
serial = argv[2]
register_start = normalize_hex(argv[3])
register_count = normalize_hex(argv[4])

mqtt = SajMqtt(broker_ip, "empty_user", "empty_pass", serial)
mqtt.listen()

try:
    data = mqtt.query(register_start, register_count)
    stderr.write("registers size: %d\n" % len(data,), )
    stdout.buffer.write(data)
except (Exception, KeyboardInterrupt) as e:
    logging.error("an exception occurred: %s" % (e,))

mqtt.shutdown()