X-Git-Url: http://average.org/gitweb/?a=blobdiff_plain;f=loctrkd%2Fwatch.py;h=6d3dcd9fa8ed7eb1f4eba1cdacd266549e9b2519;hb=84861997657f7a8daab41aa13790981bd77749f8;hp=a8a53b0dd62f04938bef519e200796986a45a3a4;hpb=dbdf9d63af31770ad57302e16b17a2fdc526773f;p=loctrkd.git diff --git a/loctrkd/watch.py b/loctrkd/watch.py index a8a53b0..6d3dcd9 100644 --- a/loctrkd/watch.py +++ b/loctrkd/watch.py @@ -2,17 +2,27 @@ from configparser import ConfigParser from datetime import datetime, timezone +from importlib import import_module from logging import getLogger +from typing import Any, cast, List import zmq from . import common -from .zx303proto import parse_message +from .protomodule import ProtoModule from .zmsg import Bcast log = getLogger("loctrkd/watch") +pmods: List[ProtoModule] = [] + + def runserver(conf: ConfigParser) -> None: + global pmods + pmods = [ + cast(ProtoModule, import_module("." + modnm, __package__)) + for modnm in conf.get("common", "protocols").split(",") + ] # Is this https://github.com/zeromq/pyzmq/issues/1627 still not fixed?! zctx = zmq.Context() # type: ignore zsub = zctx.socket(zmq.SUB) # type: ignore @@ -22,8 +32,13 @@ def runserver(conf: ConfigParser) -> None: try: while True: zmsg = Bcast(zsub.recv()) - msg = parse_message(zmsg.packet, zmsg.is_incoming) - print("I" if zmsg.is_incoming else "O", zmsg.imei, msg) + print("I" if zmsg.is_incoming else "O", zmsg.proto, zmsg.imei) + for pmod in pmods: + if pmod.proto_handled(zmsg.proto): + msg = pmod.parse_message(zmsg.packet, zmsg.is_incoming) + print(msg) + if zmsg.is_incoming and hasattr(msg, "rectified"): + print(msg.rectified()) except KeyboardInterrupt: pass