Refactor of caching & data storage part 7 #118

core/sig_lookup_helper.py

@@ -0,0 +1,93 @@
+import logging
+
+from pyhamtools.locator import locator_to_latlong, latlong_to_locator
+
+from core.data_store import DATA_STORE
+from core.geo_utils import wab_wai_square_to_lat_lon
+
+
+def populate_sig_ref_info(sig_ref):
+    """Look up details of a SIG reference (e.g. POTA park) such as name, lat/lon, and grid. Takes in a sig_ref object
+    which must at minimum have a "sig" and an "id". The rest of the object will be populated and returned. This makes
+    use of SIG ref data in the data store, live lookups from the web, or just automatic calculation depending on which
+    SIG we are getting data for.
+    Note there is currently no support for KRMNPA location lookup, see issue #61."""
+
+    if sig_ref.sig is None or sig_ref.sig == "" or sig_ref.id is None or sig_ref.id == "":
+        logging.debug("Failed to look up sig_ref info, sig or id were not set.")
+        return sig_ref
+
+    sig = sig_ref.sig
+    ref_id = sig_ref.id
+
+    try:
+        ### FUDGES ###
+        #
+        # DME fudge. Our database has leading zeros padding to 5 digits which is the expected format, but not all
+        # activators add leading zeros.
+        if sig.upper() == "DME":
+            ref_id = ref_id.zfill(5)
+
+        # KRMNPA & SANPCPA fudge. These don't have their own reference system, they just use VKFF references, so pretend
+        # the sig is WWFF and carry on
+        if sig.upper() == "KRMNPA" or sig.upper() == "SANPCPA":
+            sig = "WWFF"
+
+        ### SKIPS ###
+        #
+        # If the SIG is HEMA, we have no current lookup for this so just skip the lookup here.
+        if sig.upper() == "HEMA":
+            return sig_ref
+
+        ### PROGRAMMATIC DATA GENERATION INSTEAD OF LOOKUPS ###
+        #
+        # If the SIG is Tiles, WAB, WAI or BOTA (Beaches), we don't have anything to look up from the data store, we can
+        # calculate all the information we are going to get directly.
+        if sig.upper() == "TILES":
+            # Tiles on the Air just uses Maidenhead 6-digit squares, so ID, Name and Grid are all the same
+            if not sig_ref.name:
+                sig_ref.name = sig_ref.id
+            if not sig_ref.grid:
+                sig_ref.grid = sig_ref.id
+            if sig_ref.grid and not sig_ref.latitude:
+                ll = locator_to_latlong(str(sig_ref.grid))
+                sig_ref.latitude = ll[0]
+                sig_ref.longitude = ll[1]
+            return sig_ref
+
+        elif sig.upper() == "WAB" or sig.upper() == "WAI":
+            ll = wab_wai_square_to_lat_lon(ref_id)
+            if ll:
+                sig_ref.name = ref_id
+                try:
+                    sig_ref.grid = latlong_to_locator(ll[0], ll[1], 6)
+                    sig_ref.latitude = ll[0]
+                    sig_ref.longitude = ll[1]
+                except:
+                    logging.warning("Invalid lat/lon received for WAB/WAI reference")
+            return sig_ref
+
+        elif sig.upper() == "BOTA":
+            # For BOTA all we can ever generate is the URL, there is no data file or lookup for lat/longs
+            if not sig_ref.name:
+                sig_ref.name = sig_ref.id
+            sig_ref.url = "https://www.beachesontheair.com/beaches/" + sig_ref.name.lower().replace(" ", "-")
+            return sig_ref
+
+        ### ACTUAL LOOKUP ###
+        #
+        # OK, this is something we have to look up. Now check to see if our data store contains SIG ref information for
+        # this SIG. If so, check for the reference data and use that.
+        key = sig + ":" + ref_id
+        lookup_data = DATA_STORE.sigrefs.get(key) if key in DATA_STORE.sigrefs else None
+        if lookup_data:
+            # Copy new sig ref data into existing object where data was previously missing
+            for key, value in lookup_data.__dict__.items():
+                if value is not None and sig_ref.__dict__.get(key) is None:
+                    sig_ref.__dict__[key] = value
+        else:
+            logging.warning("%s database did not contain data for ref %s", sig, ref_id)
+
+    except Exception:
+        logging.error("Exception when looking up sig_ref info for " + sig + " ref " + ref_id, exc_info=True)
+    return sig_ref