read_esc_ll.py

#!python2-32
from py9b.link.base import LinkOpenException, LinkTimeoutException
from py9b.link.tcp import TCPLink
from py9b.link.ble import BLELink
from py9b.link.serial import SerialLink
from py9b.transport.base import BaseTransport as BT
from py9b.transport.packet import BasePacket as PKT
from py9b.transport.xiaomi import XiaomiTransport

READ_CHUNK_SIZE = 0x40

# link = SerialLink()
# link = TCPLink()
link = BLELink()

with link:
    print("Scanning...")
    ports = link.scan()
    print(ports)

    tran = XiaomiTransport(link)

    # link.open(("127.0.0.1", 6000))
    link.open(ports[0][1])
    print("Connected")

    req = PKT(src=BT.HOST, dst=BT.ESC, cmd=0x01, arg=0, data=chr(READ_CHUNK_SIZE))

    hfo = open("EscRegs.bin", "wb")
    for i in xrange(0, 0x200, READ_CHUNK_SIZE):
        print(".")
        req.arg = i >> 1
        for retry in xrange(3):
            tran.send(req)
            try:
                rsp = tran.recv()
            except LinkTimeoutException:
                continue
            break
        hfo.write(rsp.data)

    hfo.close()
    link.close()