-import sys
-
-from .GT06mod import *
-
-db = connect(sys.argv[1])
-ldb = connect(sys.argv[2])
-lc = ldb.cursor()
-c = db.cursor()
-c.execute(
- """select timestamp, imei, clntaddr, length, proto, payload from events
- where proto in (?, ?)""",
- (WIFI_POSITIONING.PROTO, WIFI_OFFLINE_POSITIONING.PROTO),
-)
-for timestamp, imei, clntaddr, length, proto, payload in c:
- obj = make_object(length, proto, payload)
- qry = """select lat, lon from cells
- where mcc = {} and (area, cell) in ({})""".format(
- obj.mcc,
- ", ".join(
- [
- "({}, {})".format(locac, cellid)
- for locac, cellid, _ in obj.gsm_cells
- ]
- ),
+
+__all__ = "init", "lookup"
+
+ldb = None
+
+
+def init(conf):
+ global ldb
+ ldb = connect(conf["opencellid"]["dbfn"])
+
+
+def lookup(mcc, mnc, gsm_cells, __):
+ lc = ldb.cursor()
+ lc.execute("""attach database ":memory:" as mem""")
+ lc.execute("create table mem.seen (locac int, cellid int, signal int)")
+ lc.executemany(
+ """insert into mem.seen (locac, cellid, signal)
+ values (?, ?, ?)""",
+ gsm_cells,
+ )
+ ldb.commit()
+ lc.execute(
+ """select c.lat, c.lon, s.signal
+ from main.cells c, mem.seen s
+ where c.mcc = ?
+ and c.area = s.locac
+ and c.cell = s.cellid""",
+ (mcc,),
+ )
+ data = list(lc.fetchall())
+ if not data:
+ return None, None
+ sumsig = sum([1 / sig for _, _, sig in data])
+ nsigs = [1 / sig / sumsig for _, _, sig in data]
+ avlat = sum([lat * nsig for (lat, _, _), nsig in zip(data, nsigs)])
+ avlon = sum([lon * nsig for (_, lon, _), nsig in zip(data, nsigs)])
+ # lc.execute("drop table mem.seen")
+ lc.execute("""detach database mem""")
+ lc.close()
+ return avlat, avlon
+
+
+if __name__.endswith("__main__"):
+ from datetime import datetime, timezone
+ import sys
+ from .gps303proto import *
+
+ db = connect(sys.argv[1])
+ c = db.cursor()
+ c.execute(
+ """select tstamp, packet from events
+ where proto in (?, ?)""",
+ (WIFI_POSITIONING.PROTO, WIFI_OFFLINE_POSITIONING.PROTO),