from datetime import datetime, timedelta import threading, time from owrx.config import PropertyManager from owrx.bands import Band import logging logger = logging.getLogger(__name__) class Location(object): def __dict__(self): return {} class Map(object): sharedInstance = None @staticmethod def getSharedInstance(): if Map.sharedInstance is None: Map.sharedInstance = Map() return Map.sharedInstance def __init__(self): self.clients = [] self.positions = {} def removeLoop(): while True: try: self.removeOldPositions() except Exception: logger.exception("error while removing old map positions") time.sleep(60) threading.Thread(target=removeLoop, daemon=True).start() super().__init__() def broadcast(self, update): for c in self.clients: c.write_update(update) def addClient(self, client): self.clients.append(client) client.write_update( [ { "callsign": callsign, "location": record["location"].__dict__(), "lastseen": record["updated"].timestamp() * 1000, "mode": record["mode"], "band": record["band"].getName() if record["band"] is not None else None, } for (callsign, record) in self.positions.items() ] ) def removeClient(self, client): try: self.clients.remove(client) except ValueError: pass def updateLocation(self, callsign, loc: Location, mode: str, band: Band = None): ts = datetime.now() self.positions[callsign] = {"location": loc, "updated": ts, "mode": mode, "band": band} self.broadcast( [ { "callsign": callsign, "location": loc.__dict__(), "lastseen": ts.timestamp() * 1000, "mode": mode, "band": band.getName() if band is not None else None, } ] ) def removeLocation(self, callsign): self.positions.pop(callsign, None) # TODO broadcast removal to clients def removeOldPositions(self): pm = PropertyManager.getSharedInstance() retention = timedelta(seconds=pm["map_position_retention_time"]) cutoff = datetime.now() - retention to_be_removed = [callsign for (callsign, pos) in self.positions.items() if pos["updated"] < cutoff] for callsign in to_be_removed: self.removeLocation(callsign) class LatLngLocation(Location): def __init__(self, lat: float, lon: float): self.lat = lat self.lon = lon def __dict__(self): return {"type": "latlon", "lat": self.lat, "lon": self.lon} class LocatorLocation(Location): def __init__(self, locator: str): self.locator = locator def __dict__(self): return {"type": "locator", "locator": self.locator}