mkgpx: quick temp switch to beesure proto

diff --git a/loctrkd/mkgpx.py b/loctrkd/mkgpx.py
index 115dd806cf9da08a6292bdbc7685164e3eb85ed9..4ea74355ad5c12e4101845a35be41c83a42b7fcb 100644 (file)
--- a/loctrkd/mkgpx.py
+++ b/loctrkd/mkgpx.py
@@ -1,60 +1,96 @@
 """ Example that produces gpx from events in evstore """
 
 # run as:
-# python -m loctrkd.mkgpx <sqlite-file> <IMEI>
+# python -m loctrkd.mkgpx <IMEI>
 # Generated gpx is emitted to stdout
 
+from configparser import ConfigParser
 from datetime import datetime, timezone
+from getopt import getopt
+from importlib import import_module
+from logging import getLogger
 from sqlite3 import connect
-import sys
-
-from .zx303proto import *
-
-db = connect(sys.argv[1])
-c = db.cursor()
-c.execute(
-    """select tstamp, is_incoming, packet from events
-       where imei = ?
-       and ((is_incoming = false and proto = ?) 
-         or (is_incoming = true and proto = ?))
-       order by tstamp""",
-    (sys.argv[2], proto_name(WIFI_POSITIONING), proto_name(GPS_POSITIONING)),
-)
-
-print(
-    """<?xml version="1.0"?>
-<gpx version="1.1"
-creator="loctrkd"
-xmlns="http://www.topografix.com/GPX/1/1">
-  <name>Location Data</name>
-  <trk>
-    <name>Location Data</name>
-    <trkseg>
-"""
-)
-
-for tstamp, is_incoming, packet in c:
-    msg = parse_message(packet, is_incoming=is_incoming)
-    lat, lon = msg.latitude, msg.longitude
-    isotime = (
-        datetime.fromtimestamp(tstamp).astimezone(tz=timezone.utc).isoformat()
+from sys import argv
+from typing import Any, cast, List, Tuple
+
+from . import common
+
+log = getLogger("loctrkd/mkgpx")
+
+
+class ProtoModule:
+    @staticmethod
+    def proto_handled(proto: str) -> bool:
+        ...
+
+    @staticmethod
+    def parse_message(packet: bytes, is_incoming: bool = True) -> Any:
+        ...
+
+
+pmods: List[ProtoModule] = []
+
+
+def main(
+    conf: ConfigParser, opts: List[Tuple[str, str]], args: List[str]
+) -> None:
+    global pmods
+    pmods = [
+        cast(ProtoModule, import_module("." + modnm, __package__))
+        for modnm in conf.get("collector", "protocols").split(",")
+    ]
+    db = connect(conf.get("storage", "dbfn"))
+    c = db.cursor()
+    c.execute(
+        """select tstamp, is_incoming, proto, packet from events
+           where imei = ?  and is_incoming = true
+           and proto in (?, ?)
+           order by tstamp""",
+        (args[0], "BS:UD", "BS:UD2"),
     )
-    isotime = isotime[: isotime.rfind(".")] + "Z"
-    trkpt = """      <trkpt lat="{}" lon="{}">
-          <time>{}</time>
-      </trkpt>""".format(
-        lat, lon, isotime
+    print(
+        """<?xml version="1.0"?>
+    <gpx version="1.1"
+    creator="loctrkd"
+    xmlns="http://www.topografix.com/GPX/1/1">
+      <name>Location Data</name>
+      <trk>
+        <name>Location Data</name>
+        <trkseg>
+    """
     )
-    print(trkpt)
-    if False:
-        print(
+
+    for tstamp, is_incoming, proto, packet in c:
+        for pmod in pmods:
+            if pmod.proto_handled(proto):
+                msg = pmod.parse_message(packet, is_incoming=is_incoming)
+        lat, lon = msg.latitude, msg.longitude
+        isotime = (
             datetime.fromtimestamp(tstamp)
             .astimezone(tz=timezone.utc)
-            .isoformat(),
-            msg,
+            .isoformat()
+        )
+        isotime = isotime[: isotime.rfind(".")] + "Z"
+        trkpt = """      <trkpt lat="{}" lon="{}">
+              <time>{}</time>
+          </trkpt>""".format(
+            lat, lon, isotime
         )
-print(
-    """    </trkseg>
-  </trk>
-</gpx>"""
-)
+        print(trkpt)
+        if False:
+            print(
+                datetime.fromtimestamp(tstamp)
+                .astimezone(tz=timezone.utc)
+                .isoformat(),
+                msg,
+            )
+    print(
+        """    </trkseg>
+      </trk>
+    </gpx>"""
+    )
+
+
+if __name__.endswith("__main__"):
+    opts, args = getopt(argv[1:], "o:c:d")
+    main(common.init(log, opts=opts), opts, args)