From 246d15466d1cef7ed54d81729619acc454f2e9bb Mon Sep 17 00:00:00 2001
From: milinda-abay <43531318+milinda-abay@users.noreply.github.com>
Date: Wed, 1 Sep 2021 15:06:43 +1000
Subject: [PATCH] Include incidence columns for nepal data download.

---
 scripts/csv_data_output.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/scripts/csv_data_output.py b/scripts/csv_data_output.py
index e8f591030b..5e98fd5322 100644
--- a/scripts/csv_data_output.py
+++ b/scripts/csv_data_output.py
@@ -16,6 +16,7 @@
     "infection_deaths",
 ]
 
+AGE_GROUPS = list(range(0, 80, 5))
 
 phl = {
     "region": [
@@ -34,10 +35,11 @@
 }
 
 lka = {"region": ["sri_lanka"], "columns": STANDARD_COL}
-npl = {"region": ["nepal"], "columns": STANDARD_COL + ["notificationsXagegroup_50","notificationsXagegroup_55"]}
 
-os.chdir(DATA_PATH)
+npl_incidence_col = [f"incidenceXagegroup_{each_age}" for each_age in AGE_GROUPS]
+npl = {"region": ["nepal"], "columns": STANDARD_COL + npl_incidence_col}
 
+os.chdir(DATA_PATH)
 
 list_of_files = os.listdir(DATA_PATH)
 
@@ -56,7 +58,7 @@ def get_files(country):
 lka["region"] = get_files(lka)
 npl["region"] = get_files(npl)
 
-country = {"lka": lka,"phl": phl, "mys": mys, "npl": npl }
+country = {"lka": lka, "phl": phl, "mys": mys, "npl": npl}
 
 for ctry in country: