typing: make zmsg.py typecheck

[loctrkd.git] / gps303 / storage.py

diff --git a/gps303/storage.py b/gps303/storage.py
index 9237871d862a2ca6b4dabc7e466114303fbd9960..14afc49e1fde494f5aebcf96bb2582f572fe1de7 100644 (file)
--- a/gps303/storage.py
+++ b/gps303/storage.py
@@ -1,16 +1,12 @@
 """ Store zmq broadcasts to sqlite """
 
 from datetime import datetime, timezone
-from getopt import getopt
 from logging import getLogger
-from logging.handlers import SysLogHandler
-import sys
-from time import time
 import zmq
 
 from . import common
 from .evstore import initdb, stow
-from .gps303proto import parse_message
+from .gps303proto import proto_of_message
 from .zmsg import Bcast
 
 log = getLogger("gps303/storage")
@@ -28,20 +24,21 @@ def runserver(conf):
     try:
         while True:
             zmsg = Bcast(zsub.recv())
-            msg = parse_message(zmsg.packet)
-            log.debug("IMEI %s from %s at %s: %s", zmsg.imei, zmsg.peeraddr, datetime.fromtimestamp(zmsg.when).astimezone(tz=timezone.utc), msg)
-            if zmsg.peeraddr is not None:
-                addr, port = zmsg.peeraddr
-                peeraddr = str(addr), port
-            else:
-                peeraddr = None
-            stow(
-                peeraddr,
-                zmsg.when,
+            log.debug(
+                "%s IMEI %s from %s at %s: %s",
+                "I" if zmsg.is_incoming else "O",
                 zmsg.imei,
-                msg.length,
-                msg.PROTO,
-                msg.payload,
+                zmsg.peeraddr,
+                datetime.fromtimestamp(zmsg.when).astimezone(tz=timezone.utc),
+                zmsg.packet.hex(),
+            )
+            stow(
+                is_incoming=zmsg.is_incoming,
+                peeraddr=str(zmsg.peeraddr),
+                when=zmsg.when,
+                imei=zmsg.imei,
+                proto=proto_of_message(zmsg.packet),
+                packet=zmsg.packet,
             )
     except KeyboardInterrupt:
         pass