Added labels mappings

Author: Dmytro Trotsko

src/alternative_interface/helper.py

@@ -1,4 +1,4 @@
-covidcast_fluview_locations_mapping = {
+COVIDCAST_FLUVIEW_LOCATIONS_MAPPING = {
     "nation:US": "nat",
     "hhs:1": "hhs1",
     "hhs:2": "hhs2",
@@ -61,4 +61,24 @@
     "state:WI": "WI",
     "state:WV": "WV",
     "state:WY": "WY",
-}
+}
+
+EXPRESS_VIEW_LABELS_MAPPING = {
+    "smoothed_pct_ed_visits_influenza": "Influenza emergency department visits (NSSP)",
+    "confirmed_admissions_flu_ew": "Influenza hospital admissions (NHSN)",
+    "percent_positive": "Test positivity for influenza (ILINet)",
+    "percent_a": "Test positivity for influenza A (ILINet)",
+    "percent_b": "Test positivity for influenza B (ILINet)",
+    "smoothed_flu_from_claims": "Influenza hospital admissions (from claims)",
+    "deaths_flu_incidence_num": "Influenza deaths (NCHS)",
+    "smoothed_cli": "Covid related doctor visits (from claims)",
+    "smoothed_pct_ed_visits_covid": "Covid emergency department visits (NSSP)",
+    "confirmed_admissions_covid_ew": "Covid hospital admissions (NHNS)",
+    "smoothed_covid19_from_claims": "Covid hospital admissions (from claims)",
+    "deaths_covid_incidence_num": "Covid deaths (NCHS)",
+    "pct_ed_visits_rsv": "RSV emergency department visits (NSSP)",
+    "confirmed_admissions_rsv_ew": "RSV hospital admissions (NHSN)",
+    "pct_ed_visits_combined": "Combined emergency department visits (NSSP)",
+    "wili": "ILI related doctor visits (ILINet)",
+    "deaths_pneumonia_or_flu_or_covid_incidence_num": "Deaths from flu, Covid or pneumonia (NCHS)",
+}

src/alternative_interface/utils.py

@@ -12,7 +12,7 @@
     get_epiweek,
     group_by_property,
 )
-from alternative_interface.helper import covidcast_fluview_locations_mapping
+from alternative_interface.helper import COVIDCAST_FLUVIEW_LOCATIONS_MAPPING, EXPRESS_VIEW_LABELS_MAPPING
 
 
 def epiweeks_in_date_range(start_date_str: str, end_date_str: str):
@@ -161,7 +161,7 @@ def get_covidcast_data(indicator, start_date, end_date, geo, api_key):
 def get_fluview_data(indicator, geo, start_date, end_date, api_key):
     region = None
     try:
-        region = covidcast_fluview_locations_mapping[geo]
+        region = COVIDCAST_FLUVIEW_LOCATIONS_MAPPING[geo]
     except KeyError:
         region = geo.split(":")[1]
     time_values = f"{start_date}--{end_date}"
@@ -173,7 +173,6 @@ def get_fluview_data(indicator, geo, start_date, end_date, api_key):
         "epiweeks": time_values,
         "api_key": api_key if api_key else settings.EPIDATA_API_KEY,
     }
-    print(indicator)
     response = requests.get(f"{settings.EPIDATA_URL}{indicator['data_source']}", params=params)
     if response.status_code == 200:
         data = response.json()
@@ -437,7 +436,7 @@ def get_chart_data(indicators, geography):
     data_end_date = today.strftime("%Y-%m-%d")
 
     for indicator in indicators:
-        title = generate_epivis_custom_title(indicator, geo_display_name)
+        title = EXPRESS_VIEW_LABELS_MAPPING.get(indicator["name"], generate_epivis_custom_title(indicator, geo_display_name))
         color = generate_random_color()
         indicator_time_type = indicator.get("time_type", "week")
         data = None
@@ -468,7 +467,7 @@ def get_chart_data(indicators, geography):
             )
             # Apply readable label, color, and normalize data for each dataset
             for ds in series["datasets"]:
-                ds["label"] = f"{title} - {ds['label']}"
+                ds["label"] = title
                 ds["borderColor"] = color
                 ds["backgroundColor"] = f"{color}33"
                 # Normalize data to 0-100% range