example.py

"""
Example of using the wlmodem library
"""
from __future__ import division, print_function
import logging
import sys
import time
from wlmodem import WlModem


def main():
    """ Demo code """
    logging.basicConfig(level=logging.DEBUG)
    import argparse
    parser = argparse.ArgumentParser(description="Water Linked Modem example")
    parser.add_argument('-D', '--device', action="store", required=True, type=str, help="Serial port.")
    parser.add_argument('-r', '--role', action="store", type=str, default="a", help="Role: a or b.")
    parser.add_argument('-c', '--channel', action="store", type=int, default=4, help="Channel: 1-7.")
    parser.add_argument('--baudrate', action="store", type=int, default=115200, help="Serial baudrate.")
    parser.add_argument('-v' , '--verbose', action="store_true", help="Verbose.")
    args = parser.parse_args()

    ch = args.channel
    if ch < 1 or ch > 7:
        print("Error: invalid channel: {}".format(ch))
        sys.exit(1)

    role = args.role
    if role not in ["a", "b"]:
        print("Error: invalid role: {}".format(role))
        sys.exit(1)

    modem = WlModem(args.device, debug=args.verbose)
    if not modem.connect():
        print("Error connecting to modem")
        sys.exit(1)

    print("Set modem role ({}) and channel ({}): ".format(role, ch), end="")
    success = modem.cmd_configure(role, ch)
    if success:
        print("success")
    else:
        print("failed")
        sys.exit(1)

    success = modem.cmd_flush_queue()
    print("Flush      :", "success" if success else "failed")

    data = b"HelloSea"
    print("Sending '{}': ".format(data), end="")
    success = modem.cmd_queue_packet(data)
    print("success" if success else "failed")

    import struct
    data = struct.pack(">bbbbbbbb", 8, 7, 6, 5, 4, 3, 2, 1)
    print("Sending encoded numbers {}: ".format(data), end="")
    success = modem.cmd_queue_packet(data)
    print("success" if success else "failed")

    print("Queue length: ", modem.cmd_get_queue_length())
    print("Diagnostic  : ", modem.cmd_get_diagnostic())

    timeout = 10
    while timeout > 0:
        link_up = modem.cmd_get_diagnostic().get("link_up")
        print("Link is " + ("UP" if link_up else "down"))
        if link_up:
            break
        time.sleep(1.0)
        timeout -= 1

    if timeout < 1:
        print("Link cannot be established. Did you start the other modem?")
        return

    print("Wait for packet from other modem. Ctrl-C to abort")
    t0 = time.time()
    try:
        while True:
            pkt = modem.get_data_packet()
            if pkt:
                print("Got:", pkt)
            if t0 + 5 > time.time():
                # Queue another packet after some time
                modem.cmd_queue_packet(data)
                t0 = time.time()

            time.sleep(0.1)
    except KeyboardInterrupt:
        print("Stopping")

    print("Diagnostic  : ", modem.cmd_get_diagnostic())


if __name__ == "__main__":
    main()