Demographics and health endpoint and CSV updates (#496)

ua-snap · Dec 9, 2024 · eccd8c4 · eccd8c4
1 parent ea2a0e1
commit eccd8c4
Show file tree

Hide file tree

Showing 3 changed files with 220 additions and 58 deletions.
diff --git a/csv_functions.py b/csv_functions.py
@@ -1051,9 +1051,9 @@ def demographics_csv(data):
     value_cols = []
     for key in data.keys():
         for subkey in data[key].keys():
-            if subkey != "description":
+            if subkey != "description" and subkey != "source":
                 value_cols.append(subkey)
-    value_cols = list(set(value_cols)) + ["description"]
+    value_cols = list(set(value_cols)) + ["description", "source"]
 
     values = value_cols
     fieldnames = ["variable"] + values
@@ -1066,9 +1066,9 @@ def demographics_csv(data):
             if csv_dict["variable"] == key:
                 ordered_csv_dicts.append(csv_dict)
 
-    metadata = "# Demographic data for individual communities plus the state of Alaska and United States.\n"
+    metadata = "# Demographic and health data for individual communities plus the state of Alaska and United States.\n"
 
-    filename_data_name = "Demographic Data - "
+    filename_data_name = "Demographic and Health Data - "
 
     return {
         "csv_dicts": ordered_csv_dicts,