X-Git-Url: http://average.org/gitweb/?a=blobdiff_plain;f=loctrkd%2Fcommon.py;h=941a93e008dbf793cad604d0df1fd86a81809b63;hb=77c191290dad765c5b2e09c60d2b7e38aff87565;hp=f112fbd53a51176e279b77c01cc05e6f32529bd2;hpb=be1de0bb68c5c4acf2edd6f04c5ea8e40d9f31fb;p=loctrkd.git diff --git a/loctrkd/common.py b/loctrkd/common.py index f112fbd..941a93e 100644 --- a/loctrkd/common.py +++ b/loctrkd/common.py @@ -3,11 +3,13 @@ from configparser import ConfigParser from importlib import import_module from getopt import getopt +from json import dumps from logging import Formatter, getLogger, Logger, StreamHandler, DEBUG, INFO from logging.handlers import SysLogHandler from pkg_resources import get_distribution, DistributionNotFound from sys import argv, stderr, stdout from typing import Any, cast, Dict, List, Optional, Tuple, Union +from types import SimpleNamespace from .protomodule import ProtoModule @@ -76,3 +78,75 @@ def parse_message(proto: str, packet: bytes, is_incoming: bool = True) -> Any: def exposed_protos() -> List[Tuple[str, bool]]: return [item for pmod in pmods for item in pmod.exposed_protos()] + + +class Report: + TYPE: str + + def __repr__(self) -> str: + return ( + self.__class__.__name__ + + "(" + + ", ".join( + [f"{k}={v.__repr__()}" for k, v in self.__dict__.items()] + ) + + ")" + ) + + @property + def json(self) -> str: + self.type = self.TYPE + return dumps(self.__dict__) + + +class CoordReport(Report): + TYPE = "location" + + def __init__( + self, + *, + devtime: str, + battery_percentage: Optional[int], + accuracy: Optional[float], + altitude: Optional[float], + speed: Optional[float], + direction: Optional[float], + latitude: float, + longitude: float, + ) -> None: + self.devtime = devtime + self.battery_percentage = battery_percentage + self.accuracy = accuracy + self.altitude = altitude + self.speed = speed + self.direction = direction + self.latitude = latitude + self.longitude = longitude + + +class HintReport(Report): + TYPE = "approximate_location" + + def __init__( + self, + *, + devtime: str, + battery_percentage: Optional[int], + mcc: int, + mnc: int, + gsm_cells: List[Tuple[int, int, int]], + wifi_aps: List[Tuple[str, str, int]], + ) -> None: + self.devtime = devtime + self.battery_percentage = battery_percentage + self.mcc = mcc + self.mnc = mnc + self.gsm_cells = gsm_cells + self.wifi_aps = wifi_aps + + +class StatusReport(Report): + TYPE = "status" + + def __init__(self, *, battery_percentage: int) -> None: + self.battery_percentage = battery_percentage