Refactor of caching & data storage part 9 #118

core/utils.py

@@ -4,6 +4,7 @@ import simplejson
 from pyhamtools.frequency import freq_to_band
 
 from core.constants import UNKNOWN_BAND, BANDS, CW_MODES, PHONE_MODES, DATA_MODES, MODE_ALIASES, ALL_MODES
+from core.data_store import DATA_STORE
 
 
 def safe_json_dumps(obj):
@@ -68,4 +69,11 @@ def infer_mode_from_frequency(freq):
             mode = "FT4"
         return mode
     except KeyError:
-        return None
+        return None
+
+
+def get_flag_for_dxcc(dxcc):
+    """Get an emoji flag for a given DXCC entity ID"""
+
+    dxcc_data = DATA_STORE.dxcc_data[dxcc] if dxcc in DATA_STORE.dxcc_data else None
+    return dxcc_data["flag"] if dxcc_data else None