X-Git-Url: http://average.org/gitweb/?a=blobdiff_plain;f=gps303%2Ftermconfig.py;h=4599dc2ed181c1d4a8d463135c295a56a60f53be;hb=bf48ccad4b4b91e7d7e09d1087f5953bc2db97d7;hp=1ab65ce937004f8bfeb5c2c99feef2934b190d76;hpb=602261422a9c34a1ac5ecde75ad87dde7fea6d8c;p=loctrkd.git diff --git a/gps303/termconfig.py b/gps303/termconfig.py index 1ab65ce..4599dc2 100644 --- a/gps303/termconfig.py +++ b/gps303/termconfig.py @@ -1,34 +1,30 @@ """ For when responding to the terminal is not trivial """ +from configparser import ConfigParser from datetime import datetime, timezone from logging import getLogger from struct import pack import zmq from . import common -from .gps303proto import * -from .zmsg import Bcast, Resp +from .zx303proto import * +from .zmsg import Bcast, Resp, topic log = getLogger("gps303/termconfig") -def runserver(conf): - termconfig = common.normconf(conf["termconfig"]) - zctx = zmq.Context() - zsub = zctx.socket(zmq.SUB) +def runserver(conf: ConfigParser) -> None: + # Is this https://github.com/zeromq/pyzmq/issues/1627 still not fixed?! + zctx = zmq.Context() # type: ignore + zsub = zctx.socket(zmq.SUB) # type: ignore zsub.connect(conf.get("collector", "publishurl")) - for protoname in ( - "SUPERVISION", - "STATUS", - "RESET", - "WHITELIST_TOTAL", - "PROHIBIT_LBS", - "SETUP", - "POSITION_UPLOAD_INTERVAL", + for proto in ( + proto_name(STATUS), + proto_name(SETUP), + proto_name(POSITION_UPLOAD_INTERVAL), ): - topic = pack("B", proto_by_name(protoname)) - zsub.setsockopt(zmq.SUBSCRIBE, topic) - zpush = zctx.socket(zmq.PUSH) + zsub.setsockopt(zmq.SUBSCRIBE, topic(proto)) + zpush = zctx.socket(zmq.PUSH) # type: ignore zpush.connect(conf.get("collector", "listenurl")) try: @@ -42,6 +38,16 @@ def runserver(conf): datetime.fromtimestamp(zmsg.when).astimezone(tz=timezone.utc), msg, ) + if msg.RESPOND is not Respond.EXT: + log.error( + "%s does not expect externally provided response", msg + ) + if zmsg.imei is not None and conf.has_section(zmsg.imei): + termconfig = common.normconf(conf[zmsg.imei]) + elif conf.has_section("termconfig"): + termconfig = common.normconf(conf["termconfig"]) + else: + termconfig = {} kwargs = {} if isinstance(msg, STATUS): kwargs = { @@ -63,12 +69,16 @@ def runserver(conf): ): if key in termconfig: kwargs[key] = termconfig[key] - resp = Resp(imei=zmsg.imei, packet=msg.response(**kwargs)) + resp = Resp( + imei=zmsg.imei, when=zmsg.when, packet=msg.Out(**kwargs).packed + ) log.debug("Response: %s", resp) zpush.send(resp.packed) except KeyboardInterrupt: - pass + zsub.close() + zpush.close() + zctx.destroy() # type: ignore if __name__.endswith("__main__"):