loctrkd/googlemaps.py

   1 import googlemaps as gmaps
   2 from typing import Any, Dict, List, Tuple
   3
   4 gclient = None
   5
   6
   7 def init(conf: Dict[str, Any]) -> None:
   8     global gclient
   9     with open(conf["googlemaps"]["accesstoken"], encoding="ascii") as fl:
  10         token = fl.read().rstrip()
  11     gclient = gmaps.Client(key=token)
  12
  13
  14 def shut() -> None:
  15     return
  16
  17
  18 def lookup(
  19     mcc: int,
  20     mnc: int,
  21     gsm_cells: List[Tuple[int, int, int]],
  22     wifi_aps: List[Tuple[str, int]],
  23 ) -> Tuple[float, float]:
  24     assert gclient is not None
  25     kwargs = {
  26         "home_mobile_country_code": mcc,
  27         "home_mobile_network_code": mnc,
  28         "radio_type": "gsm",
  29         "carrier": "O2",
  30         "consider_ip": False,
  31         "cell_towers": [
  32             {
  33                 "locationAreaCode": loc,
  34                 "cellId": cellid,
  35                 "signalStrength": sig,
  36             }
  37             for loc, cellid, sig in gsm_cells
  38         ],
  39         "wifi_access_points": [
  40             {"macAddress": mac, "signalStrength": sig} for mac, sig in wifi_aps
  41         ],
  42     }
  43     result = gclient.geolocate(**kwargs)
  44     if "location" in result:
  45         return result["location"]["lat"], result["location"]["lng"]
  46     else:
  47         raise ValueError("google geolocation: " + str(result))