main.py

import argparse


import dcp
import rpc

from util import *
from protocol import *


parser = argparse.ArgumentParser()
parser.add_argument("-i", required=True,
                    help="use INTERFACE", metavar="INTERFACE")
parser.add_argument("action", choices=("discover", "get-param", "set-param", "read", "read-inm0-filter", "read-inm0", "read-inm1", "write-inm1"))
parser.add_argument("target", nargs='?', help="MAC address of the device")
parser.add_argument("param",  nargs='?', help="parameter to read/write")
parser.add_argument("value",  nargs='?', help="value to write")
parser.add_argument("additional1",  nargs='?', help="additional parameters")
parser.add_argument("additional2",  nargs='?', help="additional parameters")

args = parser.parse_args()


s = ethernet_socket(args.i, 3)
src = get_mac(args.i)

if args.action == "discover":
    dcp.send_discover(s, src)
    dcp.read_response(s, src, debug=True)
elif args.action == "get-param":
    dcp.get_param(s, src, args.target, args.param)
elif args.action == "set-param":
    dcp.set_param(s, src, args.target, args.param, args.value)
elif args.action.startswith("read") or args.action.startswith("write"):
    print("Getting station info ...")
    info = rpc.get_station_info(s, src, args.target)
    con = rpc.RPCCon(info)

    print("Connecting to device ...")
    con.connect(src)
    
    if args.action == "read":
        print(con.read(api=int(args.param), slot=int(args.value), subslot=int(args.additional1), idx=int(args.additional2, 16)).payload)
        
    if args.action[5:] == "inm0-filter":

        data = con.read_inm0filter()
        for api in data.keys():
            for slot_number, (module_ident_number, subslots) in data[api].items():
                print("Slot %d has module 0x%04X" % (slot_number, module_ident_number))
                for subslot_number, submodule_ident_number in subslots.items():
                    print("  Subslot %d has submodule 0x%04X" % (subslot_number, submodule_ident_number))

    elif args.action[5:] == "inm0":
        inm0 = PNInM0(con.read(api=int(args.param), slot=int(args.value), subslot=int(args.additional1), idx=PNInM0.IDX).payload)
        print(inm0)

    elif args.action[5:] == "inm1":
        inm1 = PNInM1(con.read(api=int(args.param), slot=int(args.value), subslot=int(args.additional1), idx=PNInM1.IDX).payload)
        print(inm1)

    elif args.action[6:] == "inm1":
        api = int(args.param)
        slot = int(args.value)
        subslot = int(args.additional1)
        inm1 = PNInM1(con.read(api, slot, subslot, PNInM1.IDX).payload)
        inm1 = PNInM1(inm1.block_header, bytes(args.additional2, "utf-8"), inm1.im_tag_location)
        con.write(api, slot, subslot, PNInM1.IDX, inm1)